import {ref} from "vue";
|
|
/**
|
* websocket公用方法
|
*/
|
const socket = ()=>{
|
let timer = null;
|
let ws = null;
|
let isConnect = ref(false);
|
let count = 0;
|
|
// WebSocket连接
|
const connectWebsocket = (url) => {
|
if (count > 5) {
|
connectError()
|
return
|
}
|
try { // 正在建立连接
|
ws = new WebSocket(URL)
|
initWebSocket() // 初始化WebSocket连接
|
} catch { // 建立连接出错，重新连接
|
connect()
|
}
|
}
|
|
// 重新连接WebSocket
|
const connect = ()=>{
|
if(isConnect.value) return
|
console.info('尝试重新连接WebSocket');
|
if(timer) clearTimeout(timer)
|
timer = setTimeout(()=>{
|
count++;
|
connectWebsocket()
|
}, 5000);
|
}
|
|
// 初始化WebSocket连接
|
const initWebSocket = ()=>{
|
ws.onopen = function () { // WebSocket连接成功
|
isConnect.value = true;
|
console.success('WebSocket连接成功')
|
}
|
ws.onerror = function () { // WebSocket连接发生错误
|
isConnect = false;
|
connect()
|
console.error('WebSocket连接发生错误,正在尝试重新连接')
|
}
|
ws.onclose = function (e) { // 已关闭WebSocket连接
|
isConnect.value = false
|
console.warn('已关闭WebSocket连接')
|
}
|
ws.onmessage = function (data) { // 接收到服务端发送的数据
|
receive(data)
|
}
|
}
|
|
const connectError = () => {
|
console.log('WebSocket连接失败')
|
clearTimeout(timer)
|
}
|
|
|
return {
|
ws,
|
connectWebsocket,
|
};
|
};
|
export default socket;
|
