Netty网络编程实战4，使用Netty实现心跳检测机制

Posted 2022-12-23 哪吒

tags:

篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了Netty网络编程实战4，使用Netty实现心跳检测机制相关的知识，希望对你有一定的参考价值。

一、使用Netty实现心跳检测机制

通过心跳机制检测长连接的状态。Netty是通过IdleStateHandler来实现心跳机制的，每隔一段时间，就向远端发出一次心跳，用于检测远端是否处于活动状态。

二、服务端

1、主程序类

package com.guor.demo.heartbeat;

import io.netty.bootstrap.ServerBootstrap;
import io.netty.channel.ChannelFuture;
import io.netty.channel.EventLoopGroup;
import io.netty.channel.nio.NioEventLoopGroup;
import io.netty.channel.socket.nio.NioserverSocketChannel;

/**
 * 主程序类
 */
public class HeartNettyServerTest 
    public static void main(String[] args) throws InterruptedException 
        /**
         * EventLoopGroup：事件循环组，是一个线程池，也是一个死循环，用于不断地接收用户请求；
         * serverGroup：用户监听及建立连接，并把每一个连接抽象为一个channel，最后再将连接交给clientGroup处理；
         * clientGroup：真正的处理连接
         */
        EventLoopGroup serverGroup = new NioEventLoopGroup();
        EventLoopGroup clientGroup = new NioEventLoopGroup();
        try 
            // 服务端启动时的初始化操作
            ServerBootstrap serverBootstrap = new ServerBootstrap();
            // 1、将serverGroup和clientGroup注册到服务端的Channel上；
            // 2、注册一个服务端的初始化器MyNettyServerInitializer；
            // 3、该初始化器中的initChannel()方法会在连接被注册到Channel后立刻执行；
            // 5、最后将端口号绑定到8080；
            ChannelFuture channelFuture = serverBootstrap.group(serverGroup, clientGroup)
                    .channel(NioServerSocketChannel.class)
                    .childHandler(new HeartNettyServerInitializer()).bind(8080).sync();
            channelFuture.channel().closeFuture().sync();
        catch (Exception e)
            System.out.println(e);
        finally 
            serverGroup.shutdownGracefully();
            clientGroup.shutdownGracefully();

2、自定义初始化器

package com.guor.demo.heartbeat;

import io.netty.channel.ChannelInitializer;
import io.netty.channel.ChannelPipeline;
import io.netty.channel.socket.SocketChannel;
import io.netty.handler.timeout.IdleStateHandler;

/**
 * 自定义初始化器
 */
public class HeartNettyServerInitializer extends ChannelInitializer<SocketChannel> 

    @Override
    protected void initChannel(SocketChannel socketChannel) throws Exception 
        ChannelPipeline pipeline = socketChannel.pipeline();
        /**
         * IdleStateHandler：心跳机制处理器，主要用来检测远端是否读写超时，
         * 如果超时则将超时事件传入userEventTriggered(ctx,evt)方法的evt参数中
         */
        pipeline.addLast("IdleStateHandler",new IdleStateHandler(3,5,7));
        // 自定义处理器
        pipeline.addLast("HeartNettyServerHandler",new HeartNettyServerHandler());

3、自定义处理器

package com.guor.demo.heartbeat;

import io.netty.channel.ChannelHandlerContext;
import io.netty.channel.SimpleChannelInboundHandler;
import io.netty.handler.timeout.IdleState;
import io.netty.handler.timeout.IdleStateEvent;

/**
 * 自定义处理器
 */
public class HeartNettyServerHandler extends SimpleChannelInboundHandler<Object> 

    @Override
    protected void channelRead0(ChannelHandlerContext ctx, Object msg) throws Exception 

    

    /**
     * 如果IdleStateHandler检测到了超时事件，则会触发userEventTriggered(ctx,evt)方法，
     * 将超时事件传入evt参数中
     */
    @Override
    public void userEventTriggered(ChannelHandlerContext ctx, Object evt) throws Exception 
        if(evt instanceof IdleStateEvent)
            IdleStateEvent event = (IdleStateEvent)evt;
            String eventType = null;
            switch (event.state())
                case READER_IDLE:
                    eventType = "读空闲";
                    break;
                case WRITER_IDLE:
                    eventType = "写空闲";
                    break;
                case ALL_IDLE:
                    eventType = "读写空闲";
                    break;
            
            System.out.println(ctx.channel().remoteAddress()+"，超时事件："+eventType);
            ctx.channel().close();