美文网首页
Netty统计连接数

Netty统计连接数

作者: destiny_m | 来源:发表于2018-08-22 11:46 被阅读0次

    Netty统计连接数

    思路

    netty如何统计当前的连接数?
    当有连接接入netty server的时候,ChannelInboundHandlerAdapter中就会调用regiser和active方法。我们只需要在这里对计数器递增即可。
    同时当有连接断开(客户端程序手动断开的时候,客户端异常断开不会完成四次挥手,服务端没法立刻判断客户端是否离开), ChannelInboundHandlerAdapter中会调用unregisert和inactive方法,我们需要在这里对计数器进行递减即可。

    注意点

    1 例子中的NettyConnectServerHandler不是sharedable的,每次都需要new来创建。
    所以统计的变量应该是传入的参数。
    2 保证多线程的时候统计变量不会出现问题,使用AtomicInteger来保证。

    核心代码

    public class NettyConnectServerHandler extends ChannelInboundHandlerAdapter {
        private AtomicInteger connectNum;
    
        public NettyConnectServerHandler(AtomicInteger connectNum) {
            this.connectNum = connectNum;
        }
    
        @Override
        public void channelRead(ChannelHandlerContext ctx, Object msg) {
            ByteBuf in = (ByteBuf) msg;
            try {
                while (in.isReadable()) { // (1)
                    System.out.print((char) in.readByte());
                    System.out.flush();
                }
            } finally {
                ReferenceCountUtil.release(msg); // (2)
            }
        }
    
        @Override
        public void channelRegistered(ChannelHandlerContext ctx) throws Exception {
            super.channelRegistered(ctx);
            if (connectNum.incrementAndGet() % 100 == 0) {
                System.out.println("current connected" + connectNum.get());
            }
        }
    
        @Override
        public void channelUnregistered(ChannelHandlerContext ctx) throws Exception {
            super.channelUnregistered(ctx);
            if (connectNum.decrementAndGet() % 100 == 0) {
                System.out.println("current connected" + connectNum.get());
            }
        }
    
    }
    

    git地址

    https://github.com/destinym/nettyconnect.git
    count模块

    相关文章

      网友评论

          本文标题:Netty统计连接数

          本文链接:https://www.haomeiwen.com/subject/wqieiftx.html