/* eslint-disable no-console */
|
/* eslint-disable unicorn/prefer-add-event-listener */
|
import { ref } from 'vue';
|
|
import { useMessage } from '@jnpf/hooks';
|
|
import { useAppConfig } from '@vben/hooks';
|
import { useAccessStore, useUserStore } from '@vben/stores';
|
|
import ReconnectingWebSocket from 'reconnecting-websocket';
|
|
import { $t } from '#/locales';
|
import { useAuthStore } from '#/store';
|
import { getRealJnpfAppEnCode } from '#/utils/jnpf';
|
|
const userStore = useUserStore();
|
const userInfo = userStore.getUserInfo;
|
const { createMessage, createWarningModal } = useMessage();
|
const { isDevMode, webSocketURL } = useAppConfig(import.meta.env, import.meta.env.PROD);
|
|
const url = isDevMode() ? `${webSocketURL}/api/message/websocket/` : webSocketURL ? `${webSocketURL}/websocket/` : `${window.location.origin}/websocket/`;
|
const webSocketUrl = url.replace('https://', 'wss://').replace('http://', 'ws://');
|
|
let ws: any;
|
const listeners = new Map();
|
|
export function useWebSocket() {
|
const accessStore = useAccessStore();
|
const authStore = useAuthStore();
|
const token = accessStore.accessToken;
|
const server = ref(webSocketUrl + encodeURIComponent(token as string));
|
/** 初始化WebSocket */
|
function initWebSocket() {
|
if (ws) {
|
ws.close();
|
ws = null;
|
}
|
ws = new ReconnectingWebSocket(server.value);
|
|
ws.addEventListener('open', onOpen);
|
ws.onerror = onError;
|
ws.onmessage = onMessage;
|
|
function onOpen() {
|
const onConnection = { method: 'OnConnection', mobileDevice: false, systemId: userInfo?.systemId, isSeparate: !!getRealJnpfAppEnCode() };
|
sendWsMsg(JSON.stringify(onConnection));
|
}
|
|
function onError(e) {
|
console.log('[WebSocket] 连接发生错误:', e);
|
}
|
|
function onMessage(res) {
|
if (res.data) {
|
try {
|
const dataStr = res.data;
|
const data = JSON.parse(dataStr);
|
for (const callback of listeners.keys()) {
|
try {
|
callback(data);
|
} catch (error) {
|
console.error(error);
|
}
|
}
|
// initMessage: //初始化
|
// sendMessage: //发送消息
|
// receiveMessage: //接收消息
|
// messageList: //消息列表
|
// messagePush: //消息推送
|
switch (data.method) {
|
// 断开websocket连接
|
case 'closeSocket': {
|
if (ws) {
|
ws.close();
|
ws = null;
|
}
|
break;
|
}
|
// 用户过期
|
case 'logout': {
|
if (data.token && data.token !== token) return location.reload();
|
if (ws) {
|
ws.close();
|
ws = null;
|
}
|
createMessage.error(data.msg || '登录过期,请重新登录').then(() => {
|
accessStore.setAccessToken(null);
|
authStore.logout();
|
});
|
break;
|
}
|
// 刷新页面
|
case 'refresh': {
|
createWarningModal({
|
title: $t('common.tipTitle'),
|
content: data.msg || '您的权限发生变更，请刷新界面！',
|
okText: '刷新',
|
onOk: () => {
|
location.reload();
|
},
|
});
|
break;
|
}
|
default: {
|
break;
|
}
|
}
|
} catch (error) {
|
console.error('[WebSocket] data解析失败：', error);
|
}
|
}
|
}
|
}
|
/**
|
* 添加 WebSocket 消息监听
|
* @param callback
|
*/
|
function onWebSocket(callback: (data: object) => any) {
|
if (!listeners.has(callback)) {
|
if (typeof callback === 'function') {
|
listeners.set(callback, null);
|
} else {
|
console.debug('[WebSocket] 添加 WebSocket 消息监听失败：传入的参数不是一个方法');
|
}
|
}
|
}
|
|
/**
|
* 解除 WebSocket 消息监听
|
*
|
* @param callback
|
*/
|
function offWebSocket(callback: (data: object) => any) {
|
listeners.delete(callback);
|
}
|
|
function getWebSocket() {
|
return ws;
|
}
|
|
function sendWsMsg(msg: string) {
|
try {
|
const msgObj = JSON.parse(msg);
|
msgObj.token = token;
|
const content = JSON.stringify(msgObj);
|
ws.send(content);
|
} catch {}
|
}
|
|
function closeWebSocket() {
|
if (ws) {
|
ws.close();
|
ws = null;
|
}
|
}
|
|
return {
|
initWebSocket,
|
sendWsMsg,
|
onWebSocket,
|
offWebSocket,
|
getWebSocket,
|
closeWebSocket,
|
};
|
}
|
