From 28cd840d5d383968bf1a2269193b18fa8e530725 Mon Sep 17 00:00:00 2001
From: Junjie <xjj@123>
Date: 星期三, 03 四月 2024 14:25:43 +0800
Subject: [PATCH] Merge remote-tracking branch 'origin/Four-Way-Rack' into Four-Way-Rack

---
 zy-asrs-flow/src/pages/map/websocket.js |  116 +++++++++++++++++++++++++++++++++++++++------------------
 1 files changed, 79 insertions(+), 37 deletions(-)

diff --git a/zy-asrs-flow/src/pages/map/websocket.js b/zy-asrs-flow/src/pages/map/websocket.js
index 20dea9b..1b4b9a5 100644
--- a/zy-asrs-flow/src/pages/map/websocket.js
+++ b/zy-asrs-flow/src/pages/map/websocket.js
@@ -1,42 +1,84 @@
-export const createWs = () => {
-    console.log('Creating WebSocket connection...');
-    const webSocket = new WebSocket('ws://127.0.0.1:9090/wcs/map/websocket');
+import { WEBSOCKET_BASE_URL } from '@/config/setting';
 
-    // 杩炴帴鎵撳紑鏃惰Е鍙�
-    webSocket.onopen = function (event) {
-        console.log('WebSocket connection opened:', event);
-        // 鍦ㄦ娣诲姞鍙互鍦ㄨ繛鎺ユ墦寮�鏃跺彂閫佺殑浠讳綍娑堟伅
-        // webSocket.send('Hello Server!');
-    };
+export default class WebSocketClient {
 
-    // 浠庢湇鍔″櫒鎺ユ敹鍒版秷鎭椂瑙﹀彂
-    webSocket.onmessage = function (event) {
-        console.log('WebSocket message received:', event.data);
-        // 鍦ㄨ繖閲屽鐞嗘帴鏀跺埌鐨勬秷鎭�
-    };
+    constructor(path) {
+        this.url = WEBSOCKET_BASE_URL + path;
+        this.webSocket = null;
+        this.heartbeatInterval = null; // Store the interval ID
+        this.heartbeatFrequency = 30000; // Heartbeat every 10 seconds
+    }
 
-    // 鍑虹幇閿欒鏃惰Е鍙�
-    webSocket.onerror = function (event) {
-        console.error('WebSocket error observed:', event);
-        // 鍦ㄨ繖閲屽鐞哤ebSocket閿欒
-    };
-
-    // 杩炴帴鍏抽棴鏃惰Е鍙�
-    webSocket.onclose = function (event) {
-        console.log('WebSocket connection closed:', event);
-        // 鍦ㄨ繖閲屽鐞嗚繛鎺ュ叧闂簨浠�
-        // 浣犲彲鑳芥兂瑕侀噸鏂拌繛鎺ユ垨瀵圭敤鎴疯繘琛屽弸濂芥彁绀�
-    };
-
-    // 璇锋眰鍏抽棴WebSocket杩炴帴锛堜富鍔ㄥ叧闂級锛屼綘鍙互浠庡叾浠栧嚱鏁拌皟鐢ㄨ繖涓�
-    const closeWebSocket = () => {
-        if (webSocket.readyState === WebSocket.OPEN) {
-            webSocket.close(); // 杩欏皢瑙﹀彂'onclose'浜嬩欢
+    connect() {
+        if (!this.url) {
+            console.error('WebSocketClient: Cannot connect without url.');
+            return;
         }
-    };
 
-    return {
-        webSocket,
-        closeWebSocket
-    };
-};
\ No newline at end of file
+        this.webSocket = new WebSocket(this.url);
+
+        this.webSocket.onopen = (event) => {
+            console.log('websocket connection opened.');
+            // Start the heartbeat
+            this.startHeartbeat();
+        };
+
+        this.webSocket.onmessage = (event) => {
+            console.log('websocket message received:', event.data);
+            this.onMessage(event.data);
+        };
+
+        this.webSocket.onerror = (event) => {
+            console.error('websocket error observed:', event);
+        };
+
+        this.webSocket.onclose = (event) => {
+            console.log('websocket connection closed!');
+            // Clear the heartbeat
+            this.stopHeartbeat();   
+            this.reconnect();
+        };
+    }
+
+    sendMessage(message) {
+        if (this.webSocket && this.webSocket.readyState === WebSocket.OPEN) {
+            this.webSocket.send(message);
+        } else {
+            console.error('WebSocketClient: Cannot send message, WebSocket connection is not open.');
+        }
+    }
+
+    onMessage(data) {
+        // 瀛愮被鍙兘浼氶噸鍐欒繖涓柟娉曟潵澶勭悊娑堟伅
+        console.log('WebSocketClient: Message received:', data);
+    }
+
+    close() {
+        if (this.webSocket && this.webSocket.readyState === WebSocket.OPEN) {
+            this.webSocket.close();
+        }
+    }
+
+    reconnect() {
+        setTimeout(() => {
+            console.log('WebSocketClient: Attempting to reconnect...');
+            this.connect();
+        }, 3000);
+    }
+
+    startHeartbeat() {
+        if(this.heartbeatInterval) {
+            clearInterval(this.heartbeatInterval);
+        }
+        this.heartbeatInterval = setInterval(() => {
+            this.sendMessage('1');
+        }, this.heartbeatFrequency);
+    }
+
+    stopHeartbeat() {
+        if (this.heartbeatInterval) {
+            clearInterval(this.heartbeatInterval);
+            this.heartbeatInterval = null;
+        }
+    }
+}
\ No newline at end of file

--
Gitblit v1.9.1