From 2c2bc8614c8ea0ce386369eb4924da1e6aa052d1 Mon Sep 17 00:00:00 2001
From: husq <931347610@qq.com>
Date: Wed, 27 Sep 2023 09:35:57 +0800
Subject: [PATCH] 添加环境配置区分
---
src/hooks/use-connect-websocket.ts | 6 +++---
1 files changed, 3 insertions(+), 3 deletions(-)
diff --git a/src/hooks/use-connect-websocket.ts b/src/hooks/use-connect-websocket.ts
index c0ba921..805c14d 100644
--- a/src/hooks/use-connect-websocket.ts
+++ b/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())
+export function useConnectWebSocket (messageHandler: MessageHandler, url?:string) {
+ const webSocket = new ConnectWebSocket(url || getWebsocketUrl())
onMounted(() => {
webSocket?.registerMessageHandler(messageHandler)
--
Gitblit v1.9.3