activemq如何跨ip推送消息_Netty 实战：如何实现自定义协议消息推送？

推送业务

思路

消息推送一般的思路就是：

1.轮询(Pull)客户端定时的去询问服务器是否有新消息需要下发；缺点很明显Android后台不停的访问网络费电还浪费流量。

2.推送(Push)服务端有新消息立即发送给客户端，这就没有时间的延迟，消息及时到达。

当时需求过来之后就首先考虑的这两个，开发的角度Pull实现起来简单省事，但从用户来说省电和省流量才是主要的，所以最后选用Push。

客户端与服务端使用长连接，客户端定时向服务端发送心跳包维持长连接。

心跳周期

那么这里会有一个问题，心跳包的周期多少才合理？

由于移动无线网络的特点，推送服务的心跳周期并不能设置的太长，否则长连接会被释放，造成频繁的客户端重连，但是也不能设置太短，否则在当前缺乏统一心跳框架的机制下很容易导致信令风暴(例如微信心跳信令风暴问题)。

具体的心跳周期并没有统一的标准，180S也许是个不错的选择，微信为300S。

代码实现

公共代码

• PushMsg.java

定义消息结构体：

public class PushMsg implements Serializable {    private static final long serialVersionUID = 4631960168572447268L;    public static PushMsg newInstance() {        return new PushMsg();    }    /**     * 推送类型     */    private PushTypeEnum pushType;    /**     * 推送消息     */    private String info;    public PushTypeEnum pushType() {        return pushType;    }    public PushMsg pushType(PushTypeEnum pushType) {        this.pushType = pushType;        return this;    }    public String info() {        return info;    }    public PushMsg info(String info) {        this.info = info;        return this;    }    @Override    public String toString() {        return "PushMsg{" +                "pushType=" + pushType +                ", info='" + info + ''' +                '}';    }}

• PushTypeEnum.java

定义各种消息推送的类型枚举

public enum PushTypeEnum {    /**     * 请求连接     */    CONNECT_REQ,    /**     * 连接成功     */    CONNECT_SUCCESS,    /**     * 连接失败     */    CONNECT_FAIL,    /**     * 心跳请求     */    HEARTBEAT_REQ,    /**     * 心跳响应     */    HEARTBEAT_RESP,    /**     * 消息推送     */    MSG_PUSH;}

• ChannelMaps.java

用于存放 channel 信息，此处使用 channel.id() 对应的长文本，作为 key。

因为 id 长文本是唯一的。

public class ChannelMaps {    private ChannelMaps(){}    private static final Map CHANNEL_MAP = new HashMap<>();    public static void addChannel(Channel channel) {        CHANNEL_MAP.put(channel.id().asLongText(), channel);    }    public static void removeChannel(Channel channel) {        CHANNEL_MAP.remove(channel.id().asLongText());    }    public static Collection getAllChannel() {        return CHANNEL_MAP.values();    }}

服务端

• PushServer.java

服务端核心启动代码。

push() 负责向客户端推送消息。

public class PushServer {    public static void main(String[] args) {        EventLoopGroup workerGroup = new NioEventLoopGroup();        EventLoopGroup bossGroup = new NioEventLoopGroup();        try {            ServerBootstrap serverBootstrap = new ServerBootstrap();            ChannelFuture channelFuture = serverBootstrap.group(bossGroup, workerGroup)                    .channel(NioServerSocketChannel.class)                    .option(ChannelOption.SO_BACKLOG, 100)                    .handler(new LoggingHandler(LogLevel.INFO))                    .childHandler(new ChannelInitializer() {                        @Override                        protected void initChannel(SocketChannel ch) throws Exception {                            ch.pipeline().addLast(                                new ObjectEncoder(),                                new ObjectDecoder(ClassResolvers.cacheDisabled(null)),                                new ReadTimeoutHandler(100),                                new ConnectServerHandler(),                                new HeartBeatServerHandler()                            );                        }                    })                    .bind(8888)                    .syncUninterruptibly();            System.out.println("server start on 8888...");            // 这里可以推送信息            // 需要将 channel 有效信息存储在 map 中。            push();            channelFuture.channel().closeFuture().syncUninterruptibly();        } finally {            workerGroup.shutdownGracefully();            bossGroup.shutdownGracefully();        }    }    /**     * 推送到所有客户端     */    private static void push() {        try {            while (ChannelMaps.getAllChannel().isEmpty()) {                TimeUnit.SECONDS.sleep(5);                // 循环等待，直到有客户端加入。            }            System.out.println("Start push...");            PushMsg pushMsg = new PushMsg();            pushMsg.pushType(PushTypeEnum.MSG_PUSH).info("hello client");            // 循环所有的客户端，发送消息            for(Channel channel : ChannelMaps.getAllChannel()) {                System.out.println("start push client channel : " + channel.id().asLongText());                channel.writeAndFlush(pushMsg);            }            System.out.println("End push...");        } catch (InterruptedException e) {            e.printStackTrace();        }    }}

• ConnectServerHandler.java

这里是对连接相关的处理。

我们这里模拟了 auth 验证，如果用户输入的信息为 "天王盖地虎"，我们才会认为认证成功。

public class ConnectServerHandler extends SimpleChannelInboundHandler {    @Override    protected void channelRead0(ChannelHandlerContext ctx, PushMsg msg) throws Exception {        //如果是连接信息，判断是否是黑名单ip        if(PushTypeEnum.CONNECT_REQ.equals(msg.pushType())){            PushMsg response = new PushMsg();            // 口号判断            if("天王盖地虎".equals(msg.info())) {                response.pushType(PushTypeEnum.CONNECT_SUCCESS).info("宝塔镇河妖");            } else {                response.pushType(PushTypeEnum.CONNECT_FAIL).info("有内鬼，终止交易");            }            ctx.writeAndFlush(response);        }else{            ctx.fireChannelRead(msg);        }    }}

• HeartBeatServerHandler.java

服务端的心跳包处理。

如果接收到心跳包信息，则直接响应 pong。

并且将发送 ping 的客户端，认为是活着的，放在 channel 列表中

public class HeartBeatServerHandler extends SimpleChannelInboundHandler {    @Override    protected void channelRead0(ChannelHandlerContext ctx, PushMsg msg) throws Exception {        //如果是心跳包ping，则返回pong        if(PushTypeEnum.HEARTBEAT_REQ.equals(msg.pushType())){            System.out.println("Received client heart beat : " + msg.info());            // 如果接收到心跳，则认为连接成功。            ChannelMaps.addChannel(ctx.channel());            // 返回响应            PushMsg response = new PushMsg();            response.pushType(PushTypeEnum.HEARTBEAT_RESP).info("pong");            ctx.writeAndFlush(response);        }else{            ctx.fireChannelRead(msg);        }    }    @Override    public void exceptionCaught(ChannelHandlerContext ctx, Throwable cause) throws Exception {        // 如果接收到心跳异常，则直接移除当前连接        ChannelMaps.removeChannel(ctx.channel());        ctx.close();    }}

客户端

• PushClient.java

客户端启动代码如下：

public class PushClient {    public static void main(String[] args) {        EventLoopGroup workGroup = new NioEventLoopGroup();        try {            Bootstrap bootstrap = new Bootstrap();            ChannelFuture channelFuture = bootstrap                    .group(workGroup)                    .channel(NioSocketChannel.class)                    .option(ChannelOption.SO_KEEPALIVE, true)                    .handler(new ChannelInitializer() {                        @Override                        protected void initChannel(SocketChannel ch) throws Exception {                            ChannelPipeline p = ch.pipeline();                            p.addLast(new IdleStateHandler(20, 10, 0));                            p.addLast(new ObjectEncoder());                            p.addLast(new ObjectDecoder(ClassResolvers.cacheDisabled(null)));                            p.addLast(new ReadTimeoutHandler(100));                            p.addLast(new ConnectClientHandler());                            p.addLast(new HeatBeatClientHandler());                            p.addLast(new PushClientHandler());                        }                    })                    .connect("localhost", 8888)                    .syncUninterruptibly();            final String id = channelFuture.channel().id().asLongText();            System.out.println("client started: " + id);            channelFuture.channel().closeFuture().syncUninterruptibly();            System.out.println("client closed: " + id);        } finally {            workGroup.shutdownGracefully();            // 这里可以进行重登尝试        }    }}

• ConnectClientHandler.java

连接处理类信息。

会在 channel 连接上服务端之后，发送一个请求信息。

并且输出客户端对应的响应信息。

public class ConnectClientHandler extends SimpleChannelInboundHandler {    // 三次握手完成，发送连接请求    @Override    public void channelActive(ChannelHandlerContext ctx) throws Exception {        PushMsg pushMsg = PushMsg.newInstance().pushType(PushTypeEnum.CONNECT_REQ).info("天王盖地虎");        ctx.writeAndFlush(pushMsg);    }    @Override    protected void channelRead0(ChannelHandlerContext ctx, PushMsg msg) throws Exception {        // 判断是否验证成功        if(PushTypeEnum.CONNECT_SUCCESS.equals(msg.pushType())) {            // 发送心跳            System.out.println("连接成功，服务端信息: " + msg.info());            ctx.fireChannelRead(msg);        } else if(PushTypeEnum.CONNECT_FAIL.equals(msg.pushType())) {            System.err.println("连接失败");            ctx.close();        } else {            ctx.fireChannelRead(msg);        }    }}

• HeatBeatClientHandler.java

心跳包处理信息。

登录成功的 client 端，会定时 30s 发送一次心跳请求到服务端。

此处也会处理心跳的响应信息。

public class HeatBeatClientHandler extends SimpleChannelInboundHandler {    @Override    protected void channelRead0(ChannelHandlerContext ctx, PushMsg msg) throws Exception {        // 判断是否验证成功，则循环发送心跳包        if(PushTypeEnum.CONNECT_SUCCESS.equals(msg.pushType())) {            System.out.println("连接成功，开始发送心跳包。。。");            //30s 发送一次心跳            ctx.executor().scheduleAtFixedRate(                    new HeatBeatClientTask(ctx), 0, 30, TimeUnit.SECONDS);        } else if(PushTypeEnum.HEARTBEAT_RESP.equals(msg.pushType())) {            // 处理响应信息            System.out.println("接收到 server 响应: " + msg.info());        } else {            ctx.fireChannelRead(msg);        }    }    private class HeatBeatClientTask implements Runnable {        private ChannelHandlerContext ctx;        public HeatBeatClientTask(ChannelHandlerContext ctx) {            this.ctx = ctx;        }        @Override        public void run() {            ctx.writeAndFlush(PushMsg.newInstance().pushType(PushTypeEnum.HEARTBEAT_REQ).info("hello client"));        }    }}

• PushClientHandler.java

对于服务端的消息，进行处理。

此处非常简单，直接做一个输出即可，实际业务可以根据自己的需求进行处理。

public class PushClientHandler extends SimpleChannelInboundHandler {    @Override    protected void channelRead0(ChannelHandlerContext ctx, PushMsg msg) throws Exception {        //TODO: 你可以在这里实现更加复杂的逻辑。        System.out.println("Received info from server: " + msg);    }}

测试验证

启动服务端

• 服务端日志

九月 29, 2019 4:57:08 下午 io.netty.handler.logging.LoggingHandler channelRegistered信息: [id: 0xe899a8b1] REGISTERED九月 29, 2019 4:57:08 下午 io.netty.handler.logging.LoggingHandler bind信息: [id: 0xe899a8b1] BIND: 0.0.0.0/0.0.0.0:8888server start on 8888...九月 29, 2019 4:57:08 下午 io.netty.handler.logging.LoggingHandler channelActive信息: [id: 0xe899a8b1, L:/0:0:0:0:0:0:0:0:8888] ACTIVE九月 29, 2019 4:57:17 下午 io.netty.handler.logging.LoggingHandler channelRead信息: [id: 0xe899a8b1, L:/0:0:0:0:0:0:0:0:8888] READ: [id: 0xcd6512eb, L:/127.0.0.1:8888 - R:/127.0.0.1:57451]九月 29, 2019 4:57:17 下午 io.netty.handler.logging.LoggingHandler channelReadComplete信息: [id: 0xe899a8b1, L:/0:0:0:0:0:0:0:0:8888] READ COMPLETE

启动客户端

• 服务端日志

Received client heart beat : hello clientStart push...start push client channel : 00155dfffe2b601d-00004658-00000001-2d40d35bac46078d-cd6512ebEnd push...

• 客户端日志

client started: 00155dfffe2b601d-0000124c-00000000-d9cc2eabac460700-140980b9连接成功，服务端信息: 宝塔镇河妖连接成功，开始发送心跳包。。。接收到 server 响应: pongReceived info from server: PushMsg{pushType=MSG_PUSH, info='hello client'}

后续心跳信息

• 服务端

Received client heart beat : hello clientReceived client heart beat : hello clientReceived client heart beat : hello clientReceived client heart beat : hello clientReceived client heart beat : hello clientReceived client heart beat : hello clientReceived client heart beat : hello client

• 客户端

接收到 server 响应: pong接收到 server 响应: pong接收到 server 响应: pong接收到 server 响应: pong接收到 server 响应: pong接收到 server 响应: pong接收到 server 响应: pong

小结

实际上掌握这个能力，就可以实现很多想要实现的功能。

比如最常见的配置中心，就是这种推拉结合的策略。也可以自己从零写一个 RPC 框架之类的。

希望本文对你有所帮助，如果喜欢，欢迎点赞收藏转发一波。

我是老马，期待与你的下次相遇。

拓展阅读

面试经典：linux 的 5 种 I/O 模型详解

java 中的 BIO/NIO/AIO 详解

为什么选择 Netty？Netty 入门教程

netty 解决拆包和粘包问题的 4 种方法，你会几种？

netty 服务端启动流程源码详解

netty 客户端启动流程及源码详解

Netty 实战：如何实现文件服务器？

Netty 实战：如何实现 HTTP 服务器？