【Netty4 简单项目实践】十二、客户端连接池-压测利器

【前言】

之前想写压测工具来着，但是没有很好的分析工具，所以没什么动力写。最近学会了用阿尔萨斯赶紧把压测工具补上了，疯狂的压榨服务器性能。用Netty已有的连接池模型，可以省去自己维护连接保活的工作。本来写了5个类，整理了一下代码发现非常简单，基础类就一个。

【连接池】

忘记自己借鉴了谁的代码，客户端连接池采用Netty的ChannelPoolMap接口，用网络连接地址做Key，用FixedChannelPool实例化value，即不同的连接服务地址对应不同的连接池。FixedChannelPool的理论连接数上限是Integer.MAX_VALUE，并且使用ChannelHealthChecker接口来判断channel被取出池的时候是否是活的，如果是活的才向应用层吐出去。这样一来保活问题就不用自己操心了。

public class TcpClientPool {
    final EventLoopGroup group = new NioEventLoopGroup();
    final Bootstrap bootstrap = new Bootstrap();
    private static final int thread_num = Runtime.getRuntime().availableProcessors();
    // key 是地址， value是pool，即一个地址一个pool
    private AbstractChannelPoolMap<InetSocketAddress, SimpleChannelPool> poolMap;

    public void build(ChannelPoolHandler poolHandler) throws Exception {
        bootstrap.group(group)
            .channel(NioSocketChannel.class)
            .option(ChannelOption.TCP_NODELAY, true)
            .option(ChannelOption.SO_KEEPALIVE, true);

        poolMap = new AbstractChannelPoolMap<InetSocketAddress, SimpleChannelPool>() {
            @Override
            protected SimpleChannelPool newPool(InetSocketAddress key) {
                return new FixedChannelPool(bootstrap.remoteAddress(key), poolHandler, thread_num);
            }
        };
    }
    /* 下面的代码省略 */
}