io.github.shanqiang.network.server.Server Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of jstream Show documentation
There is a newer version: 2.0.6
package io.github.shanqiang.network.server;

import com.google.common.util.concurrent.ThreadFactoryBuilder;
import io.netty.bootstrap.ServerBootstrap;
import io.netty.channel.ChannelInitializer;
import io.netty.channel.ChannelOption;
import io.netty.channel.ChannelPipeline;
import io.netty.channel.EventLoopGroup;
import io.netty.channel.nio.NioEventLoopGroup;
import io.netty.channel.socket.SocketChannel;
import io.netty.channel.socket.nio.NioServerSocketChannel;
import io.netty.handler.logging.LogLevel;
import io.netty.handler.logging.LoggingHandler;
import io.netty.handler.ssl.SslContext;
import io.netty.handler.ssl.SslContextBuilder;
import io.netty.handler.ssl.util.SelfSignedCertificate;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import javax.net.ssl.SSLException;

import java.security.cert.CertificateException;

import static java.util.Objects.requireNonNull;

public class Server {
    private static final Logger logger = LoggerFactory.getLogger(Server.class);
    private final boolean isSSL;
    private final String host;
    private final int port;
    private final EventLoopGroup bossGroup;
    private final EventLoopGroup workerGroup;

    public Server(boolean isSSL, String host, int port, int bossThreads, int workerThreads) {
        this.isSSL = isSSL;
        this.host = requireNonNull(host);
        this.port = port;
        bossGroup = new NioEventLoopGroup(bossThreads
                , new ThreadFactoryBuilder().setNameFormat("server-boss").build());
        workerGroup = new NioEventLoopGroup(workerThreads
                , new ThreadFactoryBuilder()
                .setNameFormat("server-worker")
                .build());
    }

    public void start() throws CertificateException, InterruptedException, SSLException {
        final SslContext sslCtx;
        if (isSSL) {
            SelfSignedCertificate ssc = new SelfSignedCertificate();
            sslCtx = SslContextBuilder.forServer(ssc.certificate(), ssc.privateKey()).build();
        } else {
            sslCtx = null;
        }

        try {
            ServerBootstrap b = new ServerBootstrap();
            b.group(bossGroup, workerGroup)
                    .channel(NioServerSocketChannel.class)
                    .handler(new LoggingHandler(LogLevel.INFO))
                    .childHandler(new ChannelInitializer() {
                        @Override
                        public void initChannel(SocketChannel ch) throws Exception {
                            ChannelPipeline p = ch.pipeline();
                            if (sslCtx != null) {
                                p.addLast(sslCtx.newHandler(ch.alloc()));
                            }
                            p.addLast(
                                    new ResponseEncoder(),
                                    new RequestDecoder(),
                                    new ServerHandler());
                        }
                    })
                    .option(ChannelOption.SO_BACKLOG, 12800)
                    .childOption(ChannelOption.SO_KEEPALIVE, true);

            logger.info("listening to " + host + ":" + port);
            b.bind(host, port).sync().channel().closeFuture().sync();
        } finally {
            close();
        }
    }

    public void close() throws InterruptedException {
        if (!bossGroup.isShuttingDown() && !bossGroup.isShutdown() && !bossGroup.isTerminated()) {
            bossGroup.shutdownGracefully().sync();
        }
        if (!workerGroup.isShuttingDown() && !workerGroup.isShutdown() && !workerGroup.isTerminated()) {
            workerGroup.shutdownGracefully().sync();
        }
    }
}