import { getWebsocketUrl } from '@/websocket/util/config'
|
import { useConnectWebSocket } from '@/utils/websocket/connect-websocket'
|
|
/**
|
* 使用无人机WebSocket连接，根据工作区ID动态管理WebSocket连接。
|
*
|
* @param {Ref<string>} workspaceIdRef - 工作区ID的引用，用于动态更新WebSocket连接。
|
* @returns {Object} - 包含wsInfo和removeWS的响应式对象。
|
*/
|
export const useDroneWS = workspaceIdRef => {
|
const wsInfo = ref({})
|
let droneWebSocket = null
|
|
// ws消息钩子 todo后面加节流参数处理
|
const messageHandler = result => {
|
let payload = JSON.parse(result)
|
if (payload.biz_code) {
|
wsInfo.value[payload.biz_code] = payload
|
}
|
}
|
|
function init() {
|
let webSocketUrl = getWebsocketUrl() + '&workspace-id=' + workspaceIdRef.value
|
// 监听ws 消息
|
droneWebSocket = useConnectWebSocket(messageHandler, webSocketUrl)
|
}
|
|
watch(workspaceIdRef, (newValue, oldValue) => {
|
removeWS()
|
if (workspaceIdRef.value) {
|
init()
|
}
|
})
|
|
// 移除ws 事件
|
function removeWS() {
|
droneWebSocket?.close()
|
droneWebSocket = null
|
wsInfo.value = {}
|
}
|
|
onBeforeUnmount(() => {
|
removeWS()
|
})
|
|
return {wsInfo,removeWS}
|
}
|
