java基于websocket实现im聊天功能

/**
 * 保存sessionId与用户的map
 */
Map<String, SessionCache> sessionCacheMap = new ConcurrentHashMap<>();

public void wsOpen(WebSocketSession session) throws IOException {
    CacheToken cache = validateTokenAndReturnCache(session, tokenCache);
    if (cache == null) return;
    log.info("websocket id:{} 建立连接，用户token:{} 连接成功，放入在线用户缓存", session.getId(), cache.getToken());
    javsessionCacheMap.put(session.getId(), new SessionCache(cache.getToken(), session));
}

/**
 * 关闭连接清除缓存
* @param sessionId
* @return
 */
public void wsClose(String sessionId) {
    SessionCache sessionCache = sessionCacheMap.remove( sessionId );
}

/**
 * 发送消息的方法
* @param session
* @param payload
* @return
 */
public boolean sendData(WebSocketSession session, String payload){
    if (session == null || !session.isOpen()) {
return false;
}
    try {
        if (session.isOpen()) {
            session.sendMessage(new TextMessage(payload));
            return true;
        } else {
        // 连接关闭，丢弃消息
        // String token = (String) session.getAttributes().get("token");
         // log.debug( "websocket id:{} 已经关闭, token:{}, 丢弃消息。", session.getId(), token);
         // log.debug("websocket关闭丢弃消息：{}", payload);
            return false;
        }
    } catch (IOException e) {
        if (e instanceof IOException || e instanceof SocketTimeoutException) {
            log.info("发送消息报错：session.isOpen():{} ERROR: {}", session != null ? session.isOpen() : null,
            e.getMessage());
        } else {
            log.error("发送消息报错：session.isOpen():{}", session != null ? session.isOpen() : null, e);
        }
        return false;
    }
}