SpringBoot中webSocket實(shí)現即時(shí)聊天

即時(shí)聊天

這個(gè)使用了websocket，在springboot下使用很簡(jiǎn)單。前端是小程序，這個(gè)就比較坑，小程序即時(shí)聊天上線(xiàn)需要域名并且使用wss協(xié)議，就是ws+ssl更加安全。但是要上線(xiàn)這還不夠，你必須為企業(yè)主體開(kāi)發(fā)者。個(gè)人開(kāi)發(fā)者即時(shí)聊天屬于社交、不在服務(wù)類(lèi)目?jì)?，審核?huì )不通過(guò)?。?！

功能 ：我們的小程序是個(gè)二手交易小程序，即時(shí)聊天對于一個(gè)后臺服務(wù)器只是單核2g的來(lái)說(shuō)有點(diǎn)抗不住。所以在雙方都在線(xiàn)的時(shí)候沒(méi)有存儲聊天消息，只是在單方不在線(xiàn)時(shí)存儲了離線(xiàn)消息。而且只能發(fā)三條離線(xiàn)消息。仿照了csdn的聊天。

使用:我們是點(diǎn)擊進(jìn)入聊天之后才發(fā)起websocket，這就造成了一個(gè)問(wèn)題，就是用戶(hù)退出到消息列表又重新點(diǎn)進(jìn)入就會(huì )重新發(fā)送一個(gè)websocket請求。每次請求session都不一樣。而且微信限制一個(gè)用戶(hù)只能同時(shí)發(fā)起5個(gè)請求。一開(kāi)始前端沒(méi)能退出聊天頁(yè)面就端開(kāi)，就錯誤唉??！。只能后臺去斷使用sessioin.close()會(huì )調用onClose()方法 這個(gè)session是你要斷的session。不過(guò)后來(lái)前端可以自己斷了就nice了！

效果：

數據庫設計：

對于展示消息聊天列表使用了一張表。last_context為對方發(fā)送的最后一條消息。只要有一方點(diǎn)擊了私信進(jìn)入聊天頁(yè)面就會(huì )往表中插入兩條記錄。方便之后刪除聊天，畢竟一方刪除不能讓另一方也看不到信息

對于消息詳細離線(xiàn)內容，則使用了另外一張表。

后臺代碼：

package com.w.wx.controller.WebSocket;

import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONObject;
import com.w.wx.domain.ChatMessage;
import com.w.wx.service.ChatService;
import com.w.wx.utils.ALToHMUtil;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import javax.annotation.PostConstruct;
import javax.websocket.*;
import javax.websocket.server.PathParam;
import javax.websocket.server.ServerEndpoint;
import java.io.IOException;
import java.util.ArrayList;
import java.util.Iterator;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.atomic.AtomicInteger;

@Slf4j
@ServerEndpoint("/wx/{fromOpenid}/{toOpenid}")
@Component
public class WebSocketServer {

    public static WebSocketServer webSocketServer;
    @Autowired
    private ChatService chatService;

    @PostConstruct        //此注解的方法在bean加載前執行
    private void init() {
        webSocketServer = this;
        //初始化時(shí)將靜態(tài)化的interFaceInfoMapper進(jìn)行了實(shí)例化
        webSocketServer.chatService = this.chatService;
    }


	 //靜態(tài)變量，用來(lái)記錄當前在線(xiàn)連接數。應該把它設計成線(xiàn)程安全的。
    private static AtomicInteger onlineNum = new AtomicInteger();

    //concurrent包的線(xiàn)程安全HashMap，用來(lái)存放每個(gè)客戶(hù)端對應的WebSocketServer對象。
    private static ConcurrentHashMap<String, Session> sessionPools = new ConcurrentHashMap<>();

    //發(fā)送消息
    public void sendMessage(Session session, ChatMessage message) throws IOException {
        if(session != null){
            synchronized (session) {

                String s = JSONObject.toJSONString(message);
                System.out.println("52 發(fā)送數據：" + s);
                session.getBasicRemote().sendText(s);
            }
        }
    }
    //給指定用戶(hù)發(fā)送信息
    public void sendInfo(String to_openid, ChatMessage message){
        Session session = sessionPools.get(to_openid);

        if(session == null){
            webSocketServer.chatService.addDeInfo(message);
        }else{
            try {
                sendMessage(session, message);
            }catch (Exception e){
                e.printStackTrace();
            }
        }


    }
    // 群發(fā)消息
    public void broadcast(ChatMessage message){
    	for (Session session: sessionPools.values()) {
            try {
                sendMessage(session, message);
            } catch(Exception e){
                e.printStackTrace();
                continue;
            }
        }
    }



    //收到客戶(hù)端信息后，根據接收人的username把消息推下去或者群發(fā)
    // to=-1群發(fā)消息
    @OnMessage
    public void onMessage(String message) throws IOException{
        ChatMessage msg=JSON.parseObject(message, ChatMessage.class);
        sessionPools.get(msg.getToOpenid());

        webSocketServer.chatService.addInfo(message);


        if (msg.getToOpenid().equals("-1")) {
            broadcast(msg);
        } else {
            sendInfo(msg.getToOpenid(),msg);
        }

    }



    //建立連接成功調用
    @OnOpen
    public void onOpen(Session session, @PathParam(value = "fromOpenid") String fromOpenid,@PathParam(value = "toOpenid") String toOpenid) throws IOException {

        ArrayList<ChatMessage> list = webSocketServer.chatService.getAllNotRead(fromOpenid,toOpenid);
        if (!list.isEmpty()) {
            Iterator<ChatMessage> it = list.iterator();
            while (it.hasNext()) {
                ChatMessage chatMessage = it.next();
                chatMessage.setContent(ALToHMUtil.toUnicode(chatMessage.getContent()));
                sendMessage(session, chatMessage);
                log.info("115 當前用戶(hù)接收離線(xiàn)消息" + chatMessage.toString());
            }
        }

        sessionPools.put(fromOpenid, session);

        addOnlineCount();
        System.out.println("125 "+fromOpenid + "加入webSocket！當前人數為" + onlineNum);


    }

    //關(guān)閉連接時(shí)調用
    @OnClose
    public void onClose(@PathParam(value = "fromOpenid") String fromOpenid) throws IOException {

        Session session = sessionPools.get(fromOpenid);

        session.close();
        sessionPools.remove(fromOpenid);

        subOnlineCount();
        System.out.println(fromOpenid + "斷開(kāi)webSocket連接！當前人數為" + onlineNum);

    }
    //錯誤時(shí)調用
    @OnError
    public void onError(Session session, Throwable throwable){
       // System.out.println("發(fā)生錯誤");
        throwable.printStackTrace();
    }

    public static void addOnlineCount(){
        onlineNum.incrementAndGet();
    }

    public static void subOnlineCount() {
        onlineNum.decrementAndGet();
    }
    
    public static AtomicInteger getOnlineNumber() {
        return onlineNum;
    }
    
    public static ConcurrentHashMap<String, Session> getSessionPools() {
        return sessionPools;
    }
}

到此這篇關(guān)于SpringBoot中webSocket實(shí)現即時(shí)聊天的文章就介紹到這了,更多相關(guān)SpringBoot中webSocket實(shí)現即時(shí)聊天內容請搜索腳本之家以前的文章或繼續瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！