Netty 通道怎么区分对应的用户？

痛定思痛。 2022-10-13 15:52 102阅读 0赞

作者：南城之南  
出处：[https://www.cnblogs.com/liangshu/p/12459657.html][https_www.cnblogs.com_liangshu_p_12459657.html]

## 前言 ##

考虑一个功能业务,在`web`程序中向指定的某个用户进行实时通讯

在Web运用的`Socket`通讯功能中(如在线客服),为保证点对点通讯.而这个看似简单的根据用户寻到起`channel`通道实际会碰到不少问题

1.  web程序中的`Http`协议是无状态的
2.  一般项目中`socket`服务和`web`项目是独立部署的
3.  `socket`连接存在重连的情况,而`Channel`对象每次都不一样
4.  `Channel`是面向网卡绑定的,无法序列化

## 解决方案 ##

通过管理一个线程安全的`用户标识`(如用户主键)和对应`channel`的`map`链表

private final ConcurrentHashMap<String, Channel> channelMap = new ConcurrentHashMap<>();

那么问题来了,

*  在`netty`模块中怎么得到这个`用户标识`?
 *  又如何保证`netty socket`模块可以安全的识别某个通道属于某个用户?(这个可以像上面一样的方式解决)
 *  `netty socket`模块接收到一条消息又任何证明这条通道是可信的?

在`netty`的实现中是没有认证也没有`HttpSession`这个东西的,也就是说.在`netty`程序线程中是无法得到web项目登录的用户情况的.

出于这点,参考`web`项目集群的`session`共享方案.可以在Redis等缓存中保存登录信息.

![20210707171822334.png][]

1.  在`web`项目中登录之后在redis中在这个以`用户id`为名的`key`中保存一个`token`,
2.  在客户端`socket`通道建立之后立马发送包含一个`用户标识`和`ASK`到`socket`服务端,
3.  服务端根据`ASK`计算一个`token`和`redis`比对.一旦比对成功,则绑定当前`channel`和用户之间的关系;
4.  之后`server`每接收到一条消息就检测当前通道有没有绑定用户信息

> 这个`key`是**一次性**的.这点非常重要,试想一下.在你前台项目可能因为cookie过期或者后台已经自动将该用户下线,而你的`用户标识`和`ASK`暴露.那么就可能被恶意连接发送消息;

另外关于`token`和`ASK`之类的验证传输如果仅仅是为了识别和绑定用户与`channel`的关系,这点也是可以忽略的,只要redis中保存该用户的登录状态即可,通道建立的第一次通讯就传输当前浏览器的登录用户标识,再去redis中比对即可,但是redis中的这个`key`还是一次性的好,避免一个用户建立多条`socket`通道

#### 正确的绑定通道`Channel`和用户之间的关系 ####

如果我们仅仅有一个`ConcurrentHashMap<String, Channel>`,是无法快速优雅的判断当前`channel`是属于哪个用户的;我看到别人绝大多数的实现是在创建一个`channelId`和`用户标识`的Map来管理

//key为channel的长id,channel.id().asLongText();value为用户id
    private final ConcurrentHashMap<String, String> channelAndUserMap = new ConcurrentHashMap<>();

其实这不是最合理的做法,正确的做法是利用`Channel`对象提供的`AttributeMap`来保存该通道的附带信息,很多人不知道`Channel`对象提供了一个绑定自定义数据的Map

使用示例：

//用户id=>channel示例
    private final ConcurrentHashMap<String, Channel> channelMap = new ConcurrentHashMap<>();
    
    
    /**
     * 判断一个通道是否有用户在使用
     * 可做信息转发时判断该通道是否合法
     * @param channel
     * @return
     */
    public boolean hasUser(Channel channel) {
        AttributeKey<String> key = AttributeKey.valueOf("user");
        return (channel.hasAttr(key) || channel.attr(key).get() != null);//netty移除了这个map的remove方法,这里的判断谨慎一点
    }
    
    /**
     * 上线一个用户
     *
     * @param channel
     * @param userId
     */
    public void online(Channel channel, String userId) {
        //先判断用户是否在web系统中登录?
        //这部分代码个人实现,参考上面redis中的验证
    
            this.channelMap.put(userId, channel);
            AttributeKey<String> key = AttributeKey.valueOf("user");
            channel.attr(key).set(userId);
    
    
    }
    
    /**
     * 根据用户id获取该用户的通道
     *
     * @param userId
     * @return
     */
    public Channel getChannelByUserId(String userId) {
        return this.channelMap.get(userId);
    }
    
    /**
     * 判断一个用户是否在线
     *
     * @param userId
     * @return
     */
    public Boolean online(String userId) {
        return this.channelMap.containsKey(userId) && this.channelMap.get(userId) != null;
    }

**注意!!**

很多人拿`channel.id().asShortText()`来记录标识`channel`,这是错误的!!!!!短id不保证全局唯一!!

![20210707171822753.png][]

**近期热文推荐：**

1.[1,000+ 道 Java面试题及答案整理(2021最新版)][1_000_ _ Java_2021]

2.[终于靠开源项目弄到 IntelliJ IDEA 激活码了，真香！][IntelliJ IDEA]

3.[阿里 Mock 工具正式开源，干掉市面上所有 Mock 工具！][Mock _ Mock]

4.[Spring Cloud 2020.0.0 正式发布，全新颠覆性版本！][Spring Cloud 2020.0.0]

5.[《Java开发手册（嵩山版）》最新发布，速速下载！][Java]

觉得不错，别忘了随手点赞+转发哦！

[https_www.cnblogs.com_liangshu_p_12459657.html]: https://www.cnblogs.com/liangshu/p/12459657.html
[20210707171822334.png]: /images/20221005/a625f4c2c3e04d9c8cd2e01444e922e7.png
[20210707171822753.png]: /images/20221005/56b992f76bc148a8ab9726547999d94e.png
[1_000_ _ Java_2021]: https://www.javastack.cn/mst/
[IntelliJ IDEA]: http://www.javastack.cn/article/2020/intellij-idea-by-open-source-project/
[Mock _ Mock]: http://www.javastack.cn/article/2021/alibaba-testable-mock-released/
[Spring Cloud 2020.0.0]: http://www.javastack.cn/article/2020/spring-cloud-2020.0.0-released/
[Java]: http://www.javastack.cn/article/2020/alibaba-release-java-develop-rules-songshan/