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 |    4 ++--
 1 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/hooks/use-connect-websocket.ts b/src/hooks/use-connect-websocket.ts
index 43fb527..805c14d 100644
--- a/src/hooks/use-connect-websocket.ts
+++ b/src/hooks/use-connect-websocket.ts
@@ -7,8 +7,8 @@
  * 接收一个message函数
  * @param messageHandler
  */
-export function useConnectWebSocket (messageHandler: MessageHandler) {
-  const webSocket = new ConnectWebSocket(getWebsocketUrl())
+export function useConnectWebSocket (messageHandler: MessageHandler, url?:string) {
+  const webSocket = new ConnectWebSocket(url || getWebsocketUrl())
 
   onMounted(() => {
     webSocket?.registerMessageHandler(messageHandler)

--
Gitblit v1.9.3