com.gateway.connector.tcp.server.HttpServer Maven / Gradle / Ivy

Go to download
 
package com.gateway.connector.tcp.server;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.gateway.connector.tcp.config.ServerTransportConfig;
import com.gateway.utils.ThreadNameFactory;
import com.gateway.exception.InitErrorException;

import io.netty.bootstrap.ServerBootstrap;
import io.netty.buffer.PooledByteBufAllocator;
import io.netty.channel.ChannelFuture;
import io.netty.channel.ChannelFutureListener;
import io.netty.channel.ChannelOption;
import io.netty.channel.EventLoopGroup;
import io.netty.channel.nio.NioEventLoopGroup;
import io.netty.channel.socket.nio.NioServerSocketChannel;

 
public class HttpServer {

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

    private ServerTransportConfig serverConfig;

    public ServerTransportConfig getServerConfig() {
		return serverConfig;
	}

	private int port;

    public int getPort() {
		return port;
	}

	private static final int BIZ_GROUP_SIZE = Runtime.getRuntime().availableProcessors() * 2;
    private static final int BIZ_THREAD_SIZE = Runtime.getRuntime().availableProcessors() *4;

    private final EventLoopGroup bossGroup = new NioEventLoopGroup(BIZ_GROUP_SIZE,new ThreadNameFactory(this.getClass().getSimpleName()+" Boss"));
    private final EventLoopGroup workerGroup = new NioEventLoopGroup(BIZ_THREAD_SIZE,new ThreadNameFactory(this.getClass().getSimpleName()+" Worker"));

    public void init() throws Exception {
        logger.info("start http server ...");

        Class clazz = NioServerSocketChannel.class;
        // Server 
        ServerBootstrap bootstrap = new ServerBootstrap();

        bootstrap.group(bossGroup, workerGroup);
        bootstrap.channel(clazz);
        bootstrap.childHandler(new HttpServerChannelInitializer(serverConfig));
        
        bootstrap  
        .option(ChannelOption.SO_REUSEADDR, Boolean.TRUE)        
        .childOption(ChannelOption.TCP_NODELAY, true)       
        .childOption(ChannelOption.ALLOCATOR, new PooledByteBufAllocator(false))  // heap buf 's better
        .childOption(ChannelOption.CONNECT_TIMEOUT_MILLIS, Integer.valueOf(1000));
 
     
        // ，
        logger.info("start http server at port[" + port + "].");
        ChannelFuture future = bootstrap.bind(port).sync();
        ChannelFuture channelFuture = future.addListener(new ChannelFutureListener() {
            public void operationComplete(ChannelFuture future) throws Exception {
                if (future.isSuccess()) {
                    logger.info("http server have success bind to " + port);
                } else {
                    logger.error("http server fail bind to " + port);
                    throw new InitErrorException("http server start fail !", future.cause());
                }
            }
        });
    }

    public void shutdown() {
        logger.info("shutdown http server ...");
        // 
        bossGroup.shutdownGracefully();
        workerGroup.shutdownGracefully();
        logger.info("shutdown http server end.");
    }

    //------------------ set && get --------------------


    public void setServerConfig(ServerTransportConfig serverConfig) {
        this.serverConfig = serverConfig;
    }

    public void setPort(int port) {
        this.port = port;
    }
}