package com.example.netty;


import com.example.netty.udpserver.UdpNettyServer;
import com.example.netty.websocketserver.WebSocketNettyServer;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Scope;
import org.springframework.stereotype.Component;

import javax.annotation.PostConstruct;
import javax.annotation.PreDestroy;
import java.util.ArrayList;
import java.util.List;

@Component
@Scope("singleton")
public class AppContext {

    private final Logger logger = LoggerFactory.getLogger(AppContext.class);

    @Autowired
    private WebSocketNettyServer websocketNettyServer;
    @Autowired
    private UdpNettyServer udpNettyServer;


    private List<Thread> nettyThreads = new ArrayList<>();

    /**
     * 描述：Tomcat加载完ApplicationContext-main和netty文件后：
     *      1. 启动Netty WebSocket服务器；
     *      2. 加载用户数据；
     *      3. 加载用户交流群数据。
     */
    @PostConstruct
    public void init() {
        Thread websocketNettyThread = new Thread(websocketNettyServer);
        logger.info("开启独立线程，启动Netty WebSocket服务器...");
        websocketNettyThread.start();
        nettyThreads.add(websocketNettyThread);

        Thread uodNettyThread = new Thread(udpNettyServer);
        logger.info("开启独立线程，启动Netty UDP服务器...");
        uodNettyThread.start();
        nettyThreads.add(uodNettyThread);

    }

    /**
     * 描述：Tomcat服务器关闭前需要手动关闭Netty Websocket相关资源，否则会造成内存泄漏。
     *      1. 释放Netty Websocket相关连接；
     *      2. 关闭Netty Websocket服务器线程。（强行关闭，是否有必要？）
     */
    @SuppressWarnings("deprecation")
    @PreDestroy
    public void close() {
        logger.info("正在释放Netty Websocket相关连接...");
        websocketNettyServer.close();
        logger.info("正在关闭Netty UDP Websocket服务器线程...");
        for (Thread nettyThread :this.nettyThreads){
            nettyThread.stop();
        }
        logger.info("系统成功关闭！");
    }
}