Netty（29）如何实现基于Netty的长连接和推送通知？

爬山算法2025-12-25 8:46

实现基于Netty的长连接和推送通知涉及到多个步骤，包括服务器端和客户端的实现、心跳检测、消息推送等。以下是详细的实现过程和代码示例。

1. 实现服务器端

服务器端需要处理客户端的连接、心跳检测、消息推送等功能。

1.1 定义服务器端

首先，我们定义一个简单的服务器，它将接收客户端的连接并处理消息。

java 复制代码 
import io.netty.bootstrap.ServerBootstrap;
import io.netty.channel.ChannelInitializer;
import io.netty.channel.ChannelPipeline;
import io.netty.channel.EventLoopGroup;
import io.netty.channel.nio.NioEventLoopGroup;
import io.netty.channel.socket.SocketChannel;
import io.netty.channel.socket.nio.NioServerSocketChannel;
import io.netty.handler.codec.string.StringDecoder;
import io.netty.handler.codec.string.StringEncoder;
import io.netty.handler.timeout.IdleStateHandler;

public class PushNotificationServer {
    public static void main(String[] args) throws Exception {
        EventLoopGroup bossGroup = new NioEventLoopGroup(1);
        EventLoopGroup workerGroup = new NioEventLoopGroup();
        try {
            ServerBootstrap b = new ServerBootstrap();
            b.group(bossGroup, workerGroup)
             .channel(NioServerSocketChannel.class)
             .childHandler(new ChannelInitializer<SocketChannel>() {
                 @Override
                 protected void initChannel(SocketChannel ch) throws Exception {
                     ChannelPipeline p = ch.pipeline();
                     p.addLast(new IdleStateHandler(60, 30, 0)); // 心跳检测
                     p.addLast(new StringDecoder());
                     p.addLast(new StringEncoder());
                     p.addLast(new HeartbeatHandler()); // 心跳处理器
                     p.addLast(new PushNotificationServerHandler()); // 主要处理器
                 }
             });

            b.bind(8080).sync().channel().closeFuture().sync();
        } finally {
            bossGroup.shutdownGracefully();
            workerGroup.shutdownGracefully();
        }
    }
}
1.2 实现心跳处理器

HeartbeatHandler用于处理心跳检测，确保客户端和服务器端的长连接保持活跃。

java 复制代码 
import io.netty.channel.ChannelHandlerContext;
import io.netty.channel.ChannelInboundHandlerAdapter;
import io.netty.handler.timeout.IdleState;
import io.netty.handler.timeout.IdleStateEvent;

public class HeartbeatHandler extends ChannelInboundHandlerAdapter {
    @Override
    public void userEventTriggered(ChannelHandlerContext ctx, Object evt) throws Exception {
        if (evt instanceof IdleStateEvent) {
            IdleStateEvent event = (IdleStateEvent) evt;
            if (event.state() == IdleState.READER_IDLE) {
                System.out.println("Reader idle, closing connection");
                ctx.close();
            } else if (event.state() == IdleState.WRITER_IDLE) {
                System.out.println("Writer idle, sending heartbeat");
                ctx.writeAndFlush("HEARTBEAT");
            }
        } else {
            super.userEventTriggered(ctx, evt);
        }
    }

    @Override
    public void exceptionCaught(ChannelHandlerContext ctx, Throwable cause) {
        cause.printStackTrace();
        ctx.close();
    }
}
1.3 实现主要处理器

PushNotificationServerHandler用于处理客户端消息和推送通知。

java 复制代码 
import io.netty.channel.ChannelHandlerContext;
import io.netty.channel.SimpleChannelInboundHandler;

public class PushNotificationServerHandler extends SimpleChannelInboundHandler<String> {
    @Override
    protected void channelRead0(ChannelHandlerContext ctx, String msg) throws Exception {
        System.out.println("Received message: " + msg);
        // 处理接收到的消息
    }

    @Override
    public void channelActive(ChannelHandlerContext ctx) throws Exception {
        System.out.println("Client connected: " + ctx.channel().remoteAddress());
        // 可以在这里向客户端发送欢迎消息或初始化数据
    }

    @Override
    public void exceptionCaught(ChannelHandlerContext ctx, Throwable cause) {
        cause.printStackTrace();
        ctx.close();
    }

    // 用于向所有连接的客户端推送消息
    public void pushNotification(ChannelHandlerContext ctx, String message) {
        ctx.writeAndFlush(message);
    }
}

2. 实现客户端

客户端需要连接服务器并处理心跳响应和推送通知。

2.1 定义客户端

首先，我们定义一个简单的客户端，它将连接服务器并处理消息。

java 复制代码 
import io.netty.bootstrap.Bootstrap;
import io.netty.channel.ChannelInitializer;
import io.netty.channel.ChannelPipeline;
import io.netty.channel.EventLoopGroup;
import io.netty.channel.nio.NioEventLoopGroup;
import io.netty.channel.socket.SocketChannel;
import io.netty.channel.socket.nio.NioSocketChannel;
import io.netty.handler.codec.string.StringDecoder;
import io.netty.handler.codec.string.StringEncoder;
import io.netty.handler.timeout.IdleStateHandler;

public class PushNotificationClient {
    public static void main(String[] args) throws Exception {
        EventLoopGroup group = new NioEventLoopGroup();
        try {
            Bootstrap b = new Bootstrap();
            b.group(group)
             .channel(NioSocketChannel.class)
             .handler(new ChannelInitializer<SocketChannel>() {
                 @Override
                 protected void initChannel(SocketChannel ch) throws Exception {
                     ChannelPipeline p = ch.pipeline();
                     p.addLast(new IdleStateHandler(0, 30, 0)); // 心跳检测
                     p.addLast(new StringDecoder());
                     p.addLast(new StringEncoder());
                     p.addLast(new HeartbeatHandler()); // 心跳处理器
                     p.addLast(new PushNotificationClientHandler()); // 主要处理器
                 }
             });

            b.connect("localhost", 8080).sync().channel().closeFuture().sync();
        } finally {
            group.shutdownGracefully();
        }
    }
}
2.2 实现主要处理器

PushNotificationClientHandler用于处理服务器的消息和心跳响应。

java 复制代码 
import io.netty.channel.ChannelHandlerContext;
import io.netty.channel.SimpleChannelInboundHandler;

public class PushNotificationClientHandler extends SimpleChannelInboundHandler<String> {
    @Override
    protected void channelRead0(ChannelHandlerContext ctx, String msg) throws Exception {
        if ("HEARTBEAT".equals(msg)) {
            System.out.println("Received heartbeat from server");
        } else {
            System.out.println("Received notification: " + msg);
        }
    }

    @Override
    public void channelActive(ChannelHandlerContext ctx) throws Exception {
        System.out.println("Connected to server");
        // 可以在这里发送初始消息
        ctx.writeAndFlush("Client connected");
    }

    @Override
    public void exceptionCaught(ChannelHandlerContext ctx, Throwable cause) {
        cause.printStackTrace();
        ctx.close();
    }
}

3. 发送推送通知

服务器端可以通过ChannelGroup来管理所有的客户端连接，并向所有连接的客户端发送推送通知。

3.1 使用ChannelGroup管理客户端连接

修改PushNotificationServerHandler以使用ChannelGroup管理客户端连接。

java 复制代码 
import io.netty.channel.ChannelHandlerContext;
import io.netty.channel.SimpleChannelInboundHandler;
import io.netty.channel.group.ChannelGroup;
import io.netty.channel.group.DefaultChannelGroup;
import io.netty.util.concurrent.GlobalEventExecutor;

public class PushNotificationServerHandler extends SimpleChannelInboundHandler<String> {
    private static final ChannelGroup channels = new DefaultChannelGroup(GlobalEventExecutor.INSTANCE);

    @Override
    public void handlerAdded(ChannelHandlerContext ctx) throws Exception {
        channels.add(ctx.channel());
    }

    @Override
    public void handlerRemoved(ChannelHandlerContext ctx) throws Exception {
        channels.remove(ctx.channel());
    }

    @Override
    protected void channelRead0(ChannelHandlerContext ctx, String msg) throws Exception {
        System.out.println("Received message: " + msg);
        // 处理接收到的消息
    }

    @Override
    public void channelActive(ChannelHandlerContext ctx) throws Exception {
        System.out.println("Client connected: " + ctx.channel().remoteAddress());
    }

    @Override
    public void exceptionCaught(ChannelHandlerContext ctx, Throwable cause) {
        cause.printStackTrace();
        ctx.close();
    }

    // 用于向所有连接的客户端推送消息
    public void pushNotification(String message) {
        channels.writeAndFlush(message);
    }
}
3.2 发送推送通知

在服务器端的某个地方调用pushNotification方法向所有客户端发送推送通知。

java 复制代码 
public class PushNotificationServerMain {
    public static void main(String[] args) throws Exception {
        PushNotificationServer server = new PushNotificationServer();
        server.start();

        // 模拟推送通知
        while (true) {
            Thread.sleep(10000); // 每10秒推送一次通知
            server.pushNotification("This is a push notification");
        }
    }
}

4. 总结

通过以上步骤，我们实现了一个基于Netty的长连接和推送通知系统。服务器端通过ChannelGroup管理所有客户端连接，并定期发送心跳消息以保持连接活跃。客户端处理心跳响应和推送通知，确保连接的稳定性和消息的实时性。

相关推荐
m0_748245927 分钟前
Docker 容器基本操作
运维·docker·容器
23zhgjx-zgx16 分钟前
SQL注入攻击分析报告
网络·sql·ctf
咋吃都不胖lyh33 分钟前
Docker 是什么？全面解析容器化技术
运维·docker·容器
Xの哲學42 分钟前
Linux SKB: 深入解析网络包的灵魂
linux·服务器·网络·算法·边缘计算
阿杰 AJie1 小时前
Docker 常用镜像启动参数对照表
运维·docker·容器
我想发发发1 小时前
Linux实现虚拟串口通信-socat
linux·运维·服务器
济6171 小时前
linux 系统移植（第五期）--Uboot移植（4）--在U-Boot 中添加自己的开发板(4) -其他需要修改的地方-- Ubuntu20.04
linux·运维·服务器
老顾聊技术1 小时前
“Anthropic 最新发布的 AI Skills：赋能任务自动化与跨领域应用“
运维·人工智能·自动化
令狐少侠20111 小时前
Linux 系统部署夜莺 nightingale 监控公司的watchdog
linux·运维·服务器
2502_911679142 小时前
精准与稳定的基石：Agilent 66311B，为移动通信测试量身定制的核心供电单元
大数据·网络·5g·信息与通信·信号处理
热门推荐
01GitHub 镜像站点02Linux下V2Ray安装配置指南03网站改了域名，如何查找？04Labelme从安装到标注：零基础完整指南05AI 规范驱动开发“三剑客”深度对比：Spec-Kit、Kiro 与 OpenSpec 实战指南06安娜的档案(Anna’s Archive) 镜像网站/国内最新可访问入口（持续更新）07UV安装并设置国内源08Claude Code Skills 实用使用手册09百度网盘偷偷给电脑“降频”？102025-04-03 Latex学习1——本地配置Latex + VScode环境