package com.xcong.excoin.netty.server;

import com.xcong.excoin.netty.ChatServer;
import com.xcong.excoin.netty.initalizer.WebSocketServerInitializer;
import io.netty.bootstrap.ServerBootstrap;
import io.netty.channel.ChannelFuture;
import io.netty.channel.EventLoopGroup;
import io.netty.channel.nio.NioEventLoopGroup;
import io.netty.channel.socket.nio.NioServerSocketChannel;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

/**
 * @author wzy
 * @date 2019-05-06
 */
@Slf4j
@Component("webSocketServer")
public class WebSocketServer implements ChatServer {


    private EventLoopGroup boss = new NioEventLoopGroup();
    private EventLoopGroup work = new NioEventLoopGroup();

    private ChannelFuture channelFuture;

    @Autowired
    private WebSocketServerInitializer webSocketServerInitializer;

    @Override
    public void start() throws Exception {
        log.info("[websocket服务器启动]");
        try {
            ServerBootstrap b = new ServerBootstrap();
            b.group(boss, work)
                    .channel(NioServerSocketChannel.class)
                    .childHandler(webSocketServerInitializer);

            channelFuture = b.bind(9999).sync();

            log.info("[websocket服务器启动完成]-->{}", channelFuture.channel().localAddress());
        } finally {
            Runtime.getRuntime().addShutdownHook(new Thread() {
                @Override
                public void run() {
                    shutdown();
                }
            });
        }
    }

    @Override
    public void shutdown() {
        if (channelFuture != null) {
            channelFuture.channel().close().syncUninterruptibly();
        }

        if (boss != null) {
            boss.shutdownGracefully();
        }

        if (work != null) {
            work.shutdownGracefully();
        }
    }

}