优化WebSocket组件
This commit is contained in:
parent
03e4a66f18
commit
e44bf151dd
@ -7,12 +7,18 @@
|
|||||||
|
|
||||||
const emit = defineEmits(['rollback']);
|
const emit = defineEmits(['rollback']);
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 定义接收的参数
|
||||||
|
*/
|
||||||
const props = defineProps({
|
const props = defineProps({
|
||||||
uri: {
|
uri: {
|
||||||
type: String,
|
type: String,
|
||||||
},
|
},
|
||||||
});
|
});
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 定义Socket状态
|
||||||
|
*/
|
||||||
const state = reactive({
|
const state = reactive({
|
||||||
webSocket: ref(), // webSocket实例
|
webSocket: ref(), // webSocket实例
|
||||||
lockReconnect: false, // 重连锁,避免多次重连
|
lockReconnect: false, // 重连锁,避免多次重连
|
||||||
@ -27,6 +33,9 @@ const state = reactive({
|
|||||||
},
|
},
|
||||||
});
|
});
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 获取Token令牌
|
||||||
|
*/
|
||||||
const token = computed(() => {
|
const token = computed(() => {
|
||||||
return useUserStore().getToken;
|
return useUserStore().getToken;
|
||||||
});
|
});
|
||||||
@ -35,6 +44,9 @@ const tenant = computed(() => {
|
|||||||
return Session.getTenant();
|
return Session.getTenant();
|
||||||
});
|
});
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 钩子函数
|
||||||
|
*/
|
||||||
onMounted(() => {
|
onMounted(() => {
|
||||||
initWebSocket();
|
initWebSocket();
|
||||||
});
|
});
|
||||||
@ -44,6 +56,9 @@ onUnmounted(() => {
|
|||||||
clearTimeoutObj(state.heartbeat);
|
clearTimeoutObj(state.heartbeat);
|
||||||
});
|
});
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 初始化WebSocket对象
|
||||||
|
*/
|
||||||
const initWebSocket = () => {
|
const initWebSocket = () => {
|
||||||
// ws地址
|
// ws地址
|
||||||
let host = window.location.host;
|
let host = window.location.host;
|
||||||
@ -61,11 +76,17 @@ const initWebSocket = () => {
|
|||||||
state.webSocket.onclose = onClose;
|
state.webSocket.onclose = onClose;
|
||||||
};
|
};
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 重连机制
|
||||||
|
*/
|
||||||
const reconnect = () => {
|
const reconnect = () => {
|
||||||
if (!token) {
|
if (!token.value) {
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
if (state.lockReconnect || (state.maxReconnect !== -1 && state.reconnectTime > state.maxReconnect)) {
|
if (
|
||||||
|
state.lockReconnect ||
|
||||||
|
(state.maxReconnect !== -1 && state.reconnectTime > state.maxReconnect)
|
||||||
|
) {
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
state.lockReconnect = true;
|
state.lockReconnect = true;
|
||||||
@ -76,6 +97,7 @@ const reconnect = () => {
|
|||||||
state.lockReconnect = false;
|
state.lockReconnect = false;
|
||||||
}, 5000);
|
}, 5000);
|
||||||
};
|
};
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* 清空定时器
|
* 清空定时器
|
||||||
*/
|
*/
|
||||||
@ -83,6 +105,7 @@ const clearTimeoutObj = (heartbeat: any) => {
|
|||||||
heartbeat.pingTimeoutObj && clearTimeout(heartbeat.pingTimeoutObj);
|
heartbeat.pingTimeoutObj && clearTimeout(heartbeat.pingTimeoutObj);
|
||||||
heartbeat.pongTimeoutObj && clearTimeout(heartbeat.pongTimeoutObj);
|
heartbeat.pongTimeoutObj && clearTimeout(heartbeat.pongTimeoutObj);
|
||||||
};
|
};
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* 开启心跳
|
* 开启心跳
|
||||||
*/
|
*/
|
||||||
@ -116,6 +139,7 @@ const onOpen = () => {
|
|||||||
startHeartbeat();
|
startHeartbeat();
|
||||||
state.reconnectTime = 0;
|
state.reconnectTime = 0;
|
||||||
};
|
};
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* 连接失败事件
|
* 连接失败事件
|
||||||
* @param e
|
* @param e
|
||||||
@ -133,9 +157,10 @@ const onClose = () => {
|
|||||||
//重连
|
//重连
|
||||||
reconnect();
|
reconnect();
|
||||||
};
|
};
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* 接收服务器推送的信息
|
* 接收服务器推送的信息
|
||||||
* @param msgEvent
|
* @param msgEvent 消息事件
|
||||||
*/
|
*/
|
||||||
const onMessage = (msgEvent: any) => {
|
const onMessage = (msgEvent: any) => {
|
||||||
// 收到服务器信息,心跳重置并发送
|
// 收到服务器信息,心跳重置并发送
|
||||||
|
Loading…
Reference in New Issue
Block a user