SpringBoot 整合 WebSocket

喜欢你没道理_

已于 2022-08-01 15:11:03 修改

阅读量1k

点赞数

文章标签： spring boot websocket java

于 2022-08-01 15:10:46 首次发布

本文链接：https://blog.csdn.net/sinat_25849999/article/details/126101573

版权

1、pom文件

<dependency>
	<groupId>org.springframework.boot</groupId>
	<artifactId>spring-boot-starter-websocket</artifactId>
</dependency>

2、工具类

/**
 * websocket 工具类
 *
 * @author cheneq
 */
@Slf4j
@Component
public class WebSocketSessionUtil {
   

    @Resource
    private RedisUtil redisUtil;


    /**
     * 链接Id
     */
    public static final String CONNECT_ID = "connectId";

    /**
     * 用户Id
     */
    public static final String USER_ID = "userId";

    /**
     * 雪花Id
     */
    public static final String SNOW_FLAKE_ID = "snowFlakeId";

    /**
     * 连接session映射
     */
    private static Map<String, WebSocketSession> SESSION_MAP = new ConcurrentHashMap<>();


    /**
     * 用户snowflakeId添加至Redis缓存
     *
     * @return r
     */
    public void addUserSeesion(String userId, String snowflakeId, WebSocketSession webSocketSession) {
   
        // 该用户所有对应的snowflakeId存储至redis
        SESSION_MAP.put(snowflakeId, webSocketSession);
        boolean bol = redisUtil.lSet(RedisKey.WEBSOCKET_USER + userId, snowflakeId);
        if (bol) {
   
            log.debug("用户【 {} 】连接 snowflakeId: {}", userId, snowflakeId);
        }
    }

    /**
     * 用户snowflakeId从Redis缓存移除
     */
    public void removeUserSeesion(String userId, String snowflakeId) {
   
        // 该用户所有对应的snowflakeId
        SESSION_MAP.remove(snowflakeId);
        long size = redisUtil.lRemove(RedisKey.WEBSOCKET_USER + userId, 0, snowflakeId);
        if (size > 0) {
   
            log.debug("用户【 {} 】退出 snowflakeId: {}", userId<