java基于websocket實現im聊天功能

更新時間：2023年11月17日 09:03:44 作者：查拉圖斯特拉如是

這篇文章主要為大家介紹了java基于websocket實現im聊天功能示例詳解,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進步,早日升職加薪

前言

文字聊天應該是很多人每天常用的功能，這篇文章就來分析一下聊天是怎么創(chuàng)建的，他的底層邏輯是什么，以及如何實現他的底層邏輯。設計實現消息的發(fā)送過程就像這張圖一樣，經過websocket進行一個消息的轉發(fā)，一對一是這樣，在群里里面發(fā)消息也是這個邏輯，一個消息發(fā)送給多個人。本質是也是一對一聊天，只是邏輯的干預是你覺得他就是一對多群聊。

用戶緩存的核心代碼，緩存用戶連接。SessionCache封裝了用戶的信息和客戶端連接的引用。

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

當用戶登錄的時候建立websocket長連接并且把連接信息存入到本地緩存，當有用戶給他發(fā)消息的時候進行一個消息轉發(fā)。

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));
}

當用戶下線之后的代碼，從本地清除緩存，如果有用戶給他發(fā)消息就存入到緩存。

/**
 * 關閉連接清除緩存
* @param sessionId
* @return
 */
public void wsClose(String sessionId) {
    SessionCache sessionCache = sessionCacheMap.remove( sessionId );
}

發(fā)送消息的代碼，當消息過來獲取緩存中的WebSocketSession進行一個字節(jié)流的邏輯寫入。

/**
 * 發(fā)送消息的方法
* @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("發(fā)送消息報錯：session.isOpen():{} ERROR: {}", session != null ? session.isOpen() : null,
            e.getMessage());
        } else {
            log.error("發(fā)送消息報錯：session.isOpen():{}", session != null ? session.isOpen() : null, e);
        }
        return false;
    }
}

消息體

消息體常見的消息分為文字消息，圖片消息還有語音消息。這里暫且不說視頻流。也就是我們常用的視頻聊天。首先從最基本的文字，圖片和語音來說，到后臺這邊都是使用的JSON格式。文字就不用說了，圖片的話是經過一個第三方的存儲文件服務器轉換成的一個鏈接。然后用戶端進行一個渲染。然后才會看到常用的一些圖片。然后語音消息其實也是一段文憑。音頻其實也是一個文件。到文件服務器之后也是一個文件的鏈接，然后用戶端進去。格式進行一個轉換。最后呈現給用戶的就是一段語音。

敏感字的處理

這里要說的有一個點，就是一個敏感字的處理。其實敏感字的檢索這就涉及到一些算法的操作。我們可能需要查看某一段文字當中是不是有一些敏感字？在常見的操作當然是使用字符檢索一下，也可以使用一些專門的檢索算法，當然邏輯上每一個系統(tǒng)里面會有一張專門用來配置的敏感字目錄，存在敏感字。進行拒絕或者打碼操作。