快捷導(dǎo)航

vue 項目中使用websocket的正確姿勢

更新時間：2022年01月27日 16:24:16 作者：藍(lán)胖子的多啦A夢

這篇文章主要介紹了vue 項目中使用websocket的實例代碼，通過實例代碼給大家介紹了在utils下新建websocket.js文件的方法,本文給大家介紹的非常詳細(xì)，對大家的學(xué)習(xí)或工作具有一定的參考借鑒價值，需要的朋友可以參考下

1. 在utils下新建websocket.js文件

// import { showInfoMsg, showErrorMsg } from '@/utils/popInfo'
import ElementUI from 'element-ui';
function initWebSocket(e) {
    console.log(e)
    const wsUri = WS_API + "/webSocket/" + e;
    this.socket = new WebSocket(wsUri)//這里面的this都指向vue
    this.socket.onerror = webSocketOnError;
    this.socket.onmessage = webSocketOnMessage;
    this.socket.onclose = closeWebsocket;
}
function webSocketOnError(e) {
    ElementUI.Notification({
        title: '',
        message: "WebSocket連接發(fā)生錯誤" + e,
        type: 'error',
        duration: 0,
    });
}
function webSocketOnMessage(e) {
    const data = JSON.parse(e.data);
    console.log(data.msgType === "INFO", data.msgType === "INFO")
    if (data.msgType === "INFO") {
        ElementUI.Notification({
            title: '',
            message: data.msg,
            type: 'success',
            duration: 3000,
        });

    } else if (data.msgType === "ERROR") {
        ElementUI.Notification({
            title: '',
            message: data.msg,
            type: 'error',
            duration: 0,
        });
    }
}
// 關(guān)閉websiocket
function closeWebsocket() {
    console.log('連接已關(guān)閉...')
}
function close() {
    this.socket.close() // 關(guān)閉 websocket
    this.socket.onclose = function (e) {
        console.log(e)//監(jiān)聽關(guān)閉事件
        console.log('關(guān)閉')
    }
}
function webSocketSend(agentData) {
    this.socket.send(agentData);
}
export default {
    initWebSocket, close
}

如果想刷新重新鏈接websocket 可以在App.vue頁面里添加個鉤子函數(shù)

 mounted() {
    //當(dāng)在任一路由頁面被刷新時,便是根組件app被從新建立,此時能夠進(jìn)行webSocket重連
    //從localStorage中獲取用戶信息,是登陸狀態(tài)則能夠進(jìn)行webSocket重連
    let token = localStorage.getItem("token");
    if (token) {
      // userMessage = JSON.parse(userMessage);
      this.$websocket.initWebSocket(token);
    }
  },

客戶端主動關(guān)閉websocket 在關(guān)閉的地方觸發(fā)函數(shù)就可以

 logout() {
      // localStorage.clear();
      localStorage.removeItem("token");
      this.$websocket.close();
      this.$store.dispatch("LogOut").then(() => {
        location.reload();
      });
    },