溫馨提示×

您好,登錄后才能下訂單哦!

密碼登錄×
登錄注冊(cè)×
其他方式登錄
點(diǎn)擊 登錄注冊(cè) 即表示同意《億速云用戶服務(wù)條款》

websocket在springboot+vue中的使用教程

發(fā)布時(shí)間:2020-09-18 19:24:25 來源:腳本之家 閱讀:183 作者:會(huì)飛的joy 欄目:編程語(yǔ)言

1、websocket在springboot中的一種實(shí)現(xiàn)

  在java后臺(tái)中,websocket是作為一種服務(wù)端配置,其配置如下

@Configuration
public class WebSocketConfig {
  
  @Bean(name="serverEndpointExporter")
  public ServerEndpointExporter getServerEndpointExporterBean(){
    return new ServerEndpointExporter();
  }
}

  加入上面的配置之后就可以編輯自己的websocket實(shí)現(xiàn)類了,如下

@Component
@ServerEndpoint(value = "/messageSocket/{userId}")
public class MessageWebSocket {
  private static final Logger logger = LoggerFactory.getLogger(MessageWebSocket.class);
  /**
   * 靜態(tài)變量,用來記錄當(dāng)前在線連接數(shù)。應(yīng)該把它設(shè)計(jì)成線程安全的。
   */
  private static int onlineCount = 0;
  /**
   * key: userId value: sessionIds
   */
  private static ConcurrentHashMap<Integer, ConcurrentLinkedQueue<String>> userSessionMap = new ConcurrentHashMap<>();
  /**
   * concurrent包的線程安全Map,用來存放每個(gè)客戶端對(duì)應(yīng)的MyWebSocket對(duì)象。
   */
  private static ConcurrentHashMap<String, MessageWebSocket> websocketMap = new ConcurrentHashMap<>();
  /**
   * key: sessionId value: userId
   */
  private static ConcurrentHashMap<String, Integer> sessionUserMap = new ConcurrentHashMap<>();
  /**
   * 當(dāng)前連接會(huì)話,需要通過它來給客戶端發(fā)送數(shù)據(jù)
   */
  private Session session;
  /**
   * 連接建立成功調(diào)用的方法
   * */
  @OnOpen
  public void onOpen(Session session, @PathParam("userId") Integer userId) {
    System.out.println(applicationContext);
    try {
      this.session = session;
      String sessionId = session.getId();
      //建立userId和sessionId的關(guān)系
      if(userSessionMap.containsKey(userId)) {
        userSessionMap.get(userId).add(sessionId);
      }else{
        ConcurrentLinkedQueue<String> queue = new ConcurrentLinkedQueue<>();
        queue.add(sessionId);
        userSessionMap.put(userId, queue);
      }
      sessionUserMap.put(sessionId, userId);
      //建立sessionId和websocket引用的關(guān)系
      if(!websocketMap.containsKey(sessionId)){
        websocketMap.put(sessionId, this);
        addOnlineCount();      //在線數(shù)加1
      }
    }catch (Exception e){
      logger.error("連接失敗");
      String es = ExceptionUtils.getFullStackTrace(e);
      logger.error(es);
    }
  }
  /**
   * 連接關(guān)閉調(diào)用的方法
   */
  @OnClose
  public void onClose() {
    String sessionId = this.session.getId();
    //移除userId和sessionId的關(guān)系
    Integer userId = sessionUserMap.get(sessionId);
    sessionUserMap.remove(sessionId);
    if(userId != null) {
      ConcurrentLinkedQueue<String> sessionIds = userSessionMap.get(userId);
      if(sessionIds != null) {
        sessionIds.remove(sessionId);
        if (sessionIds.size() == 0) {
          userSessionMap.remove(userId);
        }
      }
    }
    //移除sessionId和websocket的關(guān)系
    if (websocketMap.containsKey(sessionId)) {
      websocketMap.remove(sessionId);
      subOnlineCount();      //在線數(shù)減1
    }
  }
  /**
   * 收到客戶端消息后調(diào)用的方法
   *
   * @param messageStr 客戶端發(fā)送過來的消息
   **/
  @OnMessage
  public void onMessage(String messageStr, Session session, @PathParam("userId") Integer userId) throws IOException {
  }
  /**
   *
   * @param session
   * @param error 當(dāng)連接發(fā)生錯(cuò)誤時(shí)的回調(diào)
   */
  @OnError
  public void onError(Session session, Throwable error) {
    String es = ExceptionUtils.getFullStackTrace(error);
    logger.error(es);
  }
  /**
   * 實(shí)現(xiàn)服務(wù)器主動(dòng)推送
   */
  public void sendMessage(String message, Integer toUserId) throws IOException {
    if(toUserId != null && !StringUtil.isEmpty(message.trim())){
      ConcurrentLinkedQueue<String> sessionIds = userSessionMap.get(toUserId);
      if(sessionIds != null) {
        for (String sessionId : sessionIds) {
          MessageWebSocket socket = websocketMap.get(sessionId);
          socket.session.getBasicRemote().sendText(message);
        }
      }
    }else{
      logger.error("未找到接收用戶連接,該用戶未連接或已斷開");
    }
  }
  public void sendMessage(String message, Session session) throws IOException {
    session.getBasicRemote().sendText(message);
  }
   /**
  *獲取在線人數(shù)
  */
  public static synchronized int getOnlineCount() {
    return onlineCount;
  }
   /**
  *在線人數(shù)加一
  */
  public static synchronized void addOnlineCount() {
    MessageWebSocket.onlineCount++;
  }
  /**
  *在線人數(shù)減一
  */
  public static synchronized void subOnlineCount() {
    MessageWebSocket.onlineCount--;
  }
}

到此后臺(tái)服務(wù)端的工作已經(jīng)做好了,前端如何作為客戶端進(jìn)行連接呢,請(qǐng)繼續(xù)往下看。。

為了實(shí)現(xiàn)斷開自動(dòng)重連,我們使用的reconnecting-websocket.js組件

//websocket連接實(shí)例
let websocket = null;
//初始話websocket實(shí)例
function initWebSocket(userId) {
  // ws地址 -->這里是你的請(qǐng)求路徑
  let host = urlConfig.wsUrl + 'messageSocket/' + userId;
  if ('WebSocket' in window) {
    websocket = new ReconnectingWebSocket(host);
    // 連接錯(cuò)誤
    websocket.onerror = function () {
    }
    // 連接成功
    websocket.onopen = function () {
    }
    // 收到消息的回調(diào),e.data為收到的信息
    websocket.onmessage = function (e) {
    }
    // 連接關(guān)閉的回調(diào)
    websocket.onclose = function () {
    }
    //監(jiān)聽窗口關(guān)閉事件,當(dāng)窗口關(guān)閉時(shí),主動(dòng)去關(guān)閉websocket連接,防止連接還沒斷開就關(guān)閉窗口,server端會(huì)拋異常。
    window.onbeforeunload = function () {
      closeWebSocket();
    }
  } else {
    alert('當(dāng)前瀏覽器不支持websocket')
    return;
  }
}
//關(guān)閉WebSocket連接
function closeWebSocket() {
  websocket.close();
}
//發(fā)送消息
function sendMessage(message){
  websocket.send(message);
}

至此一個(gè)簡(jiǎn)易的完整的websocket已經(jīng)完成了,具體功能可以依此為基本進(jìn)行擴(kuò)展。

總結(jié)

以上所述是小編給大家介紹的websocket在springboot+vue中的使用教程,希望對(duì)大家有所幫助,如果大家有任何疑問歡迎給大家留言,小編會(huì)及時(shí)回復(fù)大家的!

向AI問一下細(xì)節(jié)

免責(zé)聲明:本站發(fā)布的內(nèi)容(圖片、視頻和文字)以原創(chuàng)、轉(zhuǎn)載和分享為主,文章觀點(diǎn)不代表本網(wǎng)站立場(chǎng),如果涉及侵權(quán)請(qǐng)聯(lián)系站長(zhǎng)郵箱:is@yisu.com進(jìn)行舉報(bào),并提供相關(guān)證據(jù),一經(jīng)查實(shí),將立刻刪除涉嫌侵權(quán)內(nèi)容。

AI