vue3 封装WebSocket（直接复制）

路过的假面骑士dcd2024-02-04 15:29

创建文件WebSocket.js

只有基本的功能

javascript 复制代码

import { ref } from 'vue';
import store from "@/store";

export default function useWebSocket(url) {
    const messagesData = ref([]);

    // 创建WebSocket连接和相关方法
    const socket = new WebSocket(url);

    socket.onopen = () => {
        console.log('WebSocket连接已建立');
    };

    socket.onmessage = (event) => {
        const receivedMessage = JSON.parse(event.data);
        //接收到后端消息，根据业务逻辑自行处理
    };

    socket.onclose = () => {
        console.log('WebSocket连接已关闭');
    };
    return {
        messagesData,
        sendMessage: (message) => {
            if (socket.readyState === WebSocket.OPEN) {
                socket.send(message);
            }
        },
        closeConnection: () => {
            socket.close();
        }
    };
}

页面使用

一般是登陆成功时调用，可以在登陆页面引入(如果不是全局需要，那就在你需要的页面引入)

javascript 复制代码

import useWebSocket from "./utils/WebSocket";

javascript 复制代码

/**
 * 这段代码是放在登陆成功时调用，或者自身设置特定条件调用
 * 实例化useWebSocket
 * 获取token，无论vuex还是本地存储
 * 写好地址调用
 * store.dispatch可以保存更新后端的信息(根据自身业务需要，没有就删除)
 */
let socket = ref(null);
const token = store.state.user.token;
socket.value = useWebSocket(`地址?token=${token}`);
store.dispatch("setMessagesData", socket.value.messagesData);

销毁

javascript 复制代码

//一般是退出登录、离开某个页面时销毁
socket.value.closeConnection();

注意

当使用f5或者点击刷新页面以后, websocket连接就会强制关闭

我的处理方法是：每次进入程序前都会重新加载App.vue文件，因此在此文件执行创建websocket实例化最为合适，无论刷新进入哪个页面都将重新创建websocket。(可以写一些判断，例如判断是否有token，有则重新创建连接，否则啥也不做)

接受和发送消息直接调用封装的函数即可