You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
99 lines
2.6 KiB
99 lines
2.6 KiB
import { SSEEvent } from '@vben/constants';
|
|
import { useAppConfig } from '@vben/hooks';
|
|
import { useAccessStore } from '@vben/stores';
|
|
|
|
import { useEventSource, useWebSocket } from '@vueuse/core';
|
|
|
|
const { apiURL, clientId, sseEnable, websocketEnable } = useAppConfig(
|
|
import.meta.env,
|
|
import.meta.env.PROD,
|
|
);
|
|
|
|
export function useSseMessage() {
|
|
/**
|
|
* 未开启 不监听
|
|
*/
|
|
if (!sseEnable) {
|
|
console.warn('当前未开启sse.');
|
|
return;
|
|
}
|
|
const accessStore = useAccessStore();
|
|
const token = accessStore.accessToken;
|
|
|
|
const sseAddr = `${apiURL}/resource/sse?clientid=${clientId}&Authorization=Bearer ${token}`;
|
|
|
|
const sseReturnData = useEventSource(
|
|
sseAddr,
|
|
[SSEEvent.MESSAGE, SSEEvent.HAZARD],
|
|
{
|
|
autoReconnect: {
|
|
delay: 1000,
|
|
onFailed() {
|
|
console.error('sse重连失败.');
|
|
},
|
|
retries: 3,
|
|
},
|
|
},
|
|
);
|
|
|
|
return sseReturnData;
|
|
}
|
|
|
|
function isUrl(path?: string) {
|
|
return /^https?:\/\//.test(path || '');
|
|
}
|
|
|
|
export function useWebSocketMessage() {
|
|
if (!websocketEnable) {
|
|
console.warn('当前未开启websocket.');
|
|
return;
|
|
}
|
|
let apiUrlStr = String(apiURL);
|
|
/**
|
|
* 这里可能有两种情况 兼容dev模式的proxy或者prod模式但是没有用全路径比如http://xxx/xxx
|
|
* 1. apiUrl为https://xxx.com/xxx
|
|
* 2. apiUrl为/xxx
|
|
* 转换后为http链接形式
|
|
*/
|
|
if (!isUrl(apiURL)) {
|
|
// 协议+域名
|
|
apiUrlStr = `${window.location.protocol}//${window.location.host}${apiURL}`;
|
|
}
|
|
const accessStore = useAccessStore();
|
|
const token = accessStore.accessToken;
|
|
// 这里是http链接形式
|
|
let websocketAddr = `${apiUrlStr}/resource/websocket?clientid=${clientId}&Authorization=Bearer ${token}`;
|
|
// http/https处理
|
|
websocketAddr = window.location.protocol.includes('https')
|
|
? websocketAddr.replace('https://', 'wss://')
|
|
: websocketAddr.replace('http://', 'ws://');
|
|
// console.log('websocketUrl: ' + websocketAddr);
|
|
|
|
const websocketResponse = useWebSocket(websocketAddr, {
|
|
autoReconnect: {
|
|
// 重连最大次数
|
|
retries: 3,
|
|
// 重连间隔
|
|
delay: 1000,
|
|
onFailed() {
|
|
console.error('websocket重连失败.');
|
|
},
|
|
},
|
|
heartbeat: {
|
|
message: JSON.stringify({ type: 'ping' }),
|
|
// 发送心跳的间隔
|
|
interval: 10_000,
|
|
// 接收到心跳response的超时时间
|
|
pongTimeout: 2000,
|
|
},
|
|
onConnected() {
|
|
console.info('websocket已经连接');
|
|
},
|
|
onDisconnected() {
|
|
console.warn('websocket已经断开');
|
|
},
|
|
});
|
|
|
|
return websocketResponse;
|
|
}
|