Netty-心跳检测的作用和实现方法

81 阅读 0 评论 54 点赞

我是靠谱客的博主苹果白开水，最近开发中收集的这篇文章主要介绍Netty-心跳检测的作用和实现方法，觉得挺不错的，现在分享给大家，希望可以做个参考。

概述

欢迎微信搜索并关注“小猴子的技术笔记”公众号私信我领取丰富的视频学习资料！
关于心跳我们在长链接的开发中一般都是会遇到的，因为是长链接所以需要定时发送心跳保持连接的活跃。当服务端检测不到客户端的心跳之后就会释放资源，这个操作是一个很重要的操作。

如果你处理过原生socket的心跳检测机制，你会发现那是一个比较麻烦的处理。你需要起一个线程或者定时任务来不停的检测连接是否有心跳上送，如果没有心跳你就需要释放资源，关闭socket或者尝试重连机制。

Netty为我们提供了一个“IdleStateHandler”闲置状态处理器。如果在一定的时间内没有读、写或者两者都没有的操作将会被触发这个处理器的事件。

关于“IdleStateHandler”我们需要设置一些参数，一般的设置为：


public IdleStateHandler(
    long readerIdleTime, 
    long writerIdleTime,
    long allIdleTime,
    TimeUnit unit) {
    this(false, readerIdleTime, writerIdleTime, allIdleTime, unit);
}

“readerIdleTime”：读取数据的超时时间，也就是说连接建立之后在这个readerIdleTime时间内没有读取到数据就会被触发。

“writerIdleTime”：写超时事件，也就是说在连接建立之后在writerIdleTime时间内没有往外面写任何数据。

“allIdleTime”：读写超时事件，也就是说在连接建立之后在allIdleTime时间内既没有发生读的事件，也没有发生写的事件。

“unit”：前面参数的时间单位，是以分钟啊，还是秒啊，或者是小时为单位进行一次检测数据。

如果参数设置成0的话，那么是“无效”的，也就是说netty会忽略到这个配置。接下来看看心跳检测的示例。

public class HeartBeatServer {
    public static void main(String[] args) {
        EventLoopGroup boss = new NioEventLoopGroup();
        EventLoopGroup worker = new NioEventLoopGroup();
        try {
            ServerBootstrap server = new ServerBootstrap();
            server.group(boss, worker).channel(NioServerSocketChannel.class).childHandler(new HeartBeatServerInitializer());
            ChannelFuture channelFuture = server.bind(9999).sync();
            channelFuture.channel().closeFuture().sync();
        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            boss.shutdownGracefully();
            worker.shutdownGracefully();
        }
    }
}

public class HeartBeatServerInitializer extends ChannelInitializer<SocketChannel> {
    @Override
    protected void initChannel(SocketChannel ch) throws Exception {
        ChannelPipeline pipeline = ch.pipeline();
        pipeline.addLast(new LineBasedFrameDecoder(1024));
        pipeline.addLast(new StringDecoder());
        pipeline.addLast(new StringEncoder());
        pipeline.addLast(new IdleStateHandler(5, 0, 0, TimeUnit.SECONDS));
        pipeline.addLast(new HeartBeatServerHandler());
    }
}


public class HeartBeatServerHandler extends SimpleChannelInboundHandler<String> {
    @Override
    protected void channelRead0(ChannelHandlerContext ctx, String msg) throws Exception {
        System.out.println("接收到信息: " + msg);
    }
    @Override
    public void userEventTriggered(ChannelHandlerContext ctx, Object evt) throws Exception {
        IdleStateEvent event = (IdleStateEvent) evt;
        switch (event.state()) {
            case READER_IDLE:
                System.out.println("读空闲");
                break;
            case WRITER_IDLE:
                System.out.println("写空闲");
                break;
            case ALL_IDLE:
                System.out.println("读写空闲");
                break;
            default:
                throw new IllegalStateException("非法状态！");
        }
    }
}

假设我们写一个客户端不进行数据的发送，看看会触发什么效果：

public class HeartBeatClient {
    public static void main(String[] args) {
        EventLoopGroup worker = new NioEventLoopGroup();
        try {
            Bootstrap client = new Bootstrap();
            client.group(worker).channel(NioSocketChannel.class).handler(new HeartBeatClientInitializer());
            Channel channel = client.connect("localhost", 9999).sync().channel();
            channel.closeFuture().sync();
        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            worker.shutdownGracefully();
        }
    }
}

public class HeartBeatClientInitializer extends ChannelInitializer<SocketChannel> {
    @Override
    protected void initChannel(SocketChannel ch) throws Exception {
        ChannelPipeline pipeline = ch.pipeline();
        pipeline.addLast(new LineBasedFrameDecoder(1024));
        pipeline.addLast(new StringEncoder());
        pipeline.addLast(new StringDecoder());
        pipeline.addLast(new HeartBeatClientHandler());
    }
}

public class HeartBeatClientHandler extends SimpleChannelInboundHandler<String> {
    @Override
    protected void channelRead0(ChannelHandlerContext ctx, String msg) throws Exception {

    }
}

启动服务端和客户端，等过一段时间你会发现：


读空闲
读空闲
读空闲

也许你会疑问，被触发了之后该怎么办呢？其实，想怎么办就怎办！也许你已经发现了，在事件被触发的方法中有一个比较重要的“ChannelHandlerContext”。这个“ChannelHandlerContext”是连接ChannelHandler和ChannelPipeline的重要上下文，有了它你就可以对channel进行操作了！
在这里插入图片描述
那么将之前的代码进行一下修改，没有读取到数据就断开连接。

public class HeartBeatServerHandler extends SimpleChannelInboundHandler<String> {
    @Override
    protected void channelRead0(ChannelHandlerContext ctx, String msg) throws Exception {
        System.out.println("接收到信息: " + msg);
    }
    @Override
    public void userEventTriggered(ChannelHandlerContext ctx, Object evt) throws Exception {
        IdleStateEvent event = (IdleStateEvent) evt;
        switch (event.state()) {
            case READER_IDLE:
                System.out.println("读空闲");
                ctx.channel().close();
                break;
            case WRITER_IDLE:
                System.out.println("写空闲");
                break;
            case ALL_IDLE:
                System.out.println("读写空闲");
                break;
            default:
                throw new IllegalStateException("非法状态！");
        }
    }
}