VUE3中使用webSocket

在项目中添加websocket.js

import { getToken } from "./auth";
import {useStore} from "vuex";
import {wsTimeout} from "./socketUtil"

export default class SocketService {
    static instance = null;
    static get Instance() {
        if (!this.instance) {
            this.instance = new SocketService();
        }
        return this.instance;
    }
    // 和服务端连接的socket对象
    ws = null;
    // 存储回调函数
    callBackMapping = {};
    // 标识是否连接成功
    connected = false;
    // 记录重试的次数
    sendRetryCount = 0;
    // 重新连接尝试的次数
    connectRetryCount = 0;

    //重连时间 毫秒
    retryTime = 500;

    store = useStore();

    //  定义连接服务器的方法
    connect() {
        console.log("连接websocket")
        // 连接服务器
        if (!window.WebSocket) {
            return console.log('您的浏览器不支持WebSocket');
        }

        let url = process.env.VUE_APP_WS_API;//配置文件获取url
        this.ws = new WebSocket(url,[getToken()]);//getToken()请求头添加token鉴权
        // 连接成功的事件
        this.ws.onopen = () => {
            console.log('ws连接服务端成功, url:' + url);

            //server重启，将重连次数存入store，监听该值的组件重新订阅socket
            if(this.connectRetryCount > 0){
                this.store.commit("setWsRetryCount", this.connectRetryCount)
            }

            this.connected = true;
            // 重置重新连接的次数
            this.connectRetryCount = 0;
        };
        // 1.连接服务端失败
        // 2.当连接成功之后, 服务器关闭的情况
        this.ws.onclose = () => {
            console.log('ws连接服务端失败, url:' + url);
            this.connected = false;
            this.connectRetryCount++;
            setTimeout(() => {
                this.connect();
            }, wsTimeout(this.connectRetryCount , this.retryTime));
        };
        // 得到服务端发送过来的数据
        this.ws.onmessage = msg => {
            console.log(msg.data, '从服务端获取到了数据');
            const recvData = JSON.parse(msg.data)
            const socketType = recvData.type;
            //注册回调函数 参数2 是需要回调的方法
            if (this.callBackMapping[socketType]) {
                // const realData = msg.data // 得到该图表的数据
                this.callBackMapping[socketType].call(this,recvData)
            }
        };
    }
    // 回调函数的注册
    registerCallBack(socketType, callBack) {
        this.callBackMapping[socketType] = callBack;
    }
    // 取消某一个回调函数
    unRegisterCallBack(socketType) {
        this.callBackMapping[socketType] = null;
    }
    // 发送数据的方法
    send(data) {
        console.log("websocket发送数据："+JSON.stringify(data))
        // 判断此时此刻有没有连接成功
        if (this.connected) {
            this.sendRetryCount = 0;
            try {
                this.ws.send(JSON.stringify(data));
            } catch (e) {
                this.ws.send(data);
            }
        } else {
            this.sendRetryCount++;
            setTimeout(() => {
                this.send(data);
            }, wsTimeout(this.sendRetryCount , this.retryTime));
        }
    }
    //关闭连接
    close(){
        this.ws.close();
    }
}

其他组件使用webSocket

socketUtil：当server重启或宕机，设置重连时间

//socket重连间隔时间
export function wsTimeout(wsRetryCount , time) {
    let timeout = 0;
    if(wsRetryCount == 1){
        //重连时间随机1s以内，若用户较多避免同时重连
        timeout = Math.ceil(Math.random()*100)*10
    }else if(wsRetryCount > 1 && wsRetryCount <= 10){
        //重连时间递增
        timeout = wsRetryCount * time
    }else{
        //最大重连时间
        timeout = 5000;
    }
    // console.log("当前时间："+new Date().getTime())
    // console.log("重连间隔时间："+timeout)
    return timeout
}

项目中使用到websocket通讯的组件，在server重启或宕机时，监听重连次数，进行重连。在websocket.js已将重连次数存储在store

import {useStore} from "vuex";

store = useStore();

// 连接成功的事件
this.ws.onopen = () => {
    console.log('ws连接服务端成功, url:' + url);
    
    //server重启，将重连次数存入store，监听该值的组件重新订阅socket
    if(this.connectRetryCount > 0){
        this.store.commit("setWsRetryCount", this.connectRetryCount)
    }

    this.connected = true;
    // 重置重新连接的次数
    this.connectRetryCount = 0;
};

组件监听重连次数