From d84fee9319f3377fb9bc6cd394bb211fc7b06767 Mon Sep 17 00:00:00 2001
From: 疯狂的狮子Li <15040126243@163.com>
Date: 星期四, 16 十一月 2023 12:24:52 +0800
Subject: [PATCH] !58 修复vite环境变量类型原因导致websocket关闭失效问题 Merge pull request !58 from Lau/N/A

---
 src/utils/websocket.ts |   11 ++++++++++-
 1 files changed, 10 insertions(+), 1 deletions(-)

diff --git a/src/utils/websocket.ts b/src/utils/websocket.ts
index 0d5e9ee..a4529f9 100644
--- a/src/utils/websocket.ts
+++ b/src/utils/websocket.ts
@@ -20,6 +20,7 @@
 
 import { getToken } from '@/utils/auth';
 import useNoticeStore from '@/store/modules/notice';
+import { ElNotification } from "element-plus";
 
 const { addNotice } = useNoticeStore();
 
@@ -32,6 +33,9 @@
 
 // 鍒濆鍖杝ocket
 export const initWebSocket = (url: any) => {
+  if (import.meta.env.VITE_APP_WEBSOCKET === 'false') {
+    return;
+  }
   socketUrl = url;
   // 鍒濆鍖� websocket
   websocket = new WebSocket(url + '?Authorization=Bearer ' + getToken() + '&clientid=' + import.meta.env.VITE_APP_CLIENT_ID);
@@ -126,7 +130,12 @@
       read: false,
       time: new Date().toLocaleString()
     });
-    ElMessage.success(e.data);
+    ElNotification({
+      title: '娑堟伅',
+      message: e.data,
+      type: 'success',
+      duration: 3000
+    })
     return e.data;
   };
 };

--
Gitblit v1.9.3