From f95f6e8390c355bd601849de26b338bb64c97b90 Mon Sep 17 00:00:00 2001
From: gssong <1742057357@qq.com>
Date: 星期五, 07 六月 2024 21:38:17 +0800
Subject: [PATCH] update 优化审批按钮判断

---
 src/utils/websocket.ts |   23 +++++++++++++++--------
 1 files changed, 15 insertions(+), 8 deletions(-)

diff --git a/src/utils/websocket.ts b/src/utils/websocket.ts
index 90e86bf..d4dd8a8 100644
--- a/src/utils/websocket.ts
+++ b/src/utils/websocket.ts
@@ -19,9 +19,8 @@
  */
 
 import { getToken } from '@/utils/auth';
+import { ElNotification } from 'element-plus';
 import useNoticeStore from '@/store/modules/notice';
-
-const { addNotice } = useNoticeStore();
 
 let socketUrl: any = ''; // socket鍦板潃
 let websocket: any = null; // websocket 瀹炰緥
@@ -32,6 +31,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);
@@ -115,18 +117,23 @@
 // socket 鎺ユ敹鏁版嵁
 export const websocketonmessage = () => {
   websocket.onmessage = function (e: any) {
-    const msg = JSON.parse(e.data) as any;
-    if (msg.type === 'heartbeat') {
+    if (e.data.indexOf('heartbeat') > 0) {
       resetHeart();
     }
-    if (msg.type === 'ping') {
+    if (e.data.indexOf('ping') > 0) {
       return;
     }
-    addNotice({
-      message: msg,
+    useNoticeStore().addNotice({
+      message: e.data,
       read: false,
       time: new Date().toLocaleString()
     });
-    return msg;
+    ElNotification({
+      title: '娑堟伅',
+      message: e.data,
+      type: 'success',
+      duration: 3000
+    });
+    return e.data;
   };
 };

--
Gitblit v1.9.3