husq
2023-09-13 8b2796fffef8d0436639b997b662a8b5790404e5
src/hooks/use-connect-websocket.ts
@@ -1,14 +1,14 @@
import { onMounted, onUnmounted } from 'vue'
import ReconnectingWebSocket from 'reconnecting-websocket'
import ConnectWebSocket, { MessageHandler } from '/@/websocket'
import { getWebsocketUrl, getMyWebSocketUrl } from '/@/websocket/util/config'
import { getWebsocketUrl } from '/@/websocket/util/config'
/**
 * 接收一个message函数
 * @param messageHandler
 */
export function useConnectWebSocket (messageHandler: MessageHandler) {
  const webSocket = new ConnectWebSocket(getMyWebSocketUrl())
  const webSocket = new ConnectWebSocket(getWebsocketUrl())
  onMounted(() => {
    webSocket?.registerMessageHandler(messageHandler)