IM系统第六章 -- 断线重连恢复通信

最新推荐文章于 2023-03-15 09:51:13 发布

原创最新推荐文章于 2023-03-15 09:51:13 发布 · 642 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#java #网络

学习笔记同时被 2 个专栏收录

20 篇文章

订阅专栏

Java学习

15 篇文章

订阅专栏

本文介绍了IM系统中实现断线重连的机制，包括客户端通过定时检测连接状态，当发现断开时发起重连请求。服务端接收到请求后，更新用户和群组的Channel映射。该机制确保了在网络恢复后能及时恢复通信。

IM系统第六章 – 断线重连恢复通信

你是如何实现该功能的？

断线重连该机制在通信中是一种很重要的机制，假如没有该机制，无法及时在恢复网络的时候接收到对方发来的消息，需要退出登录重新连接才能接收的到；为了解决这个情况，”断线重连“机制就应运而生了。

那这种机制你如何实现？利用在客户端中 定时检测连接状态的方式，来判断是否保持良好的链接，如果断开就进行重连恢复通信。

下图为实现该逻辑流程图（参考小傅哥）

实现功能所需的处理：

自定义协议；
服务端断线重连处理；
客户端实现心跳机制处理；

自定义协议：ReconnectRequest

// 断线重连请求  协议
public class ReconnectRequest extends Packet {
    private String useId; // 发起重连的用户Id
    public ReconnectRequest(String userId) {
        this.userId = userId;
    }
    public String getUserId() {
        return userId;
    }
    public void setUserId(String userId) {
        this.userId = userId;
    }
}

自定义完协议后，客户端的Application中设定 定时检测连接状态 机制；

public class Application extends javafx.application.Application {
    // 默认线程池
    private static final ExecutorService executorService = Executors.newFixedThreadPool(2);
    private final ScheduledExecutorService scheduledExecutorService = Executors.newScheduledThreadPool(1);
    // Channel状态定时巡检，3s后每5s执行一次连接检测
    scheduledExecutorService.scheduleAtFixedRate(() -> {
        while(!nettyClient.isActive()) {
            try{
                System.out.println("通信管道巡检：断线重连[begin]");
                // 重新提交 nettyClient任务到executorService线程池中
                Channel freshChennel = executorService.submit(nettyClient).get();
                // 如果缓存中的userId为空，则跳过
                if(null == CacheUtil.userId) continue;
                // 否则将userId作为 重连请求协议的参数 传送过去
                freshChannel.writeAndFlush(new ReconnectRequest(CacheUtil.userId));
            }catch(Exception e) {
                System.out.println("通信管道巡检：断线重连[error]");
            }
        }
    } 3,5,TimeUnit.SECONDS);
}

服务端的主要任务：发生断线重连后，客户端发来重连请求协议，利用这个协议的userId 结合SocketUtil 工具类来将原有的用户管道删除，然后增添新的现在参数的channel（添加用户channel的操作）；从数据库中查到该用户的群组的对话框集合，然后进行遍历更新。

public class ReconnectHandler extends MyBizHandler<ReconnectRequest> {
    @Override
    public void channelRead(Channel channel, ReconnectRequest msg) {
        // 添加个人channel[删除原有的channel，增添新的channel与用户id的映射]
        SocketChannelUtil.removeUserChannelByUserId(msg.getUserId());
        SocketChannelUtil.addChannel(msg.getUserId(), channel);
        
        // 添加群组channel[通过查询用户的 对话群组，然后遍历来更新]
        List<String> groupsList = userService.queryTalkBoxGroupsIdList(msg.getUserId());
        for(String groupId : groupsList) {
            SocketChannelUtil.addChannelGroup(groupId,channel);
        }
    }
}

总结

分三步：

自定义断线重连协议；
客户端中Application 要实现 定时检测连接状态 机制，每5s就进行一次心跳机制；如果CacheUtil.userId 不为空，就证明该用户已经断开连接，需要发重连请求协议过去给服务端；
服务端中，主要是对用户原有的channel进行删除和新的channel进行了增添，群组就从数据库中根据用户的id 查询到群组的对话框集合，然后进行遍历更新即可。