From 09206aedcfdf30050123e99f2af0a192ebad1de4 Mon Sep 17 00:00:00 2001
From: zj <1772600164@qq.com>
Date: Mon, 03 Jun 2024 18:08:33 +0800
Subject: [PATCH] 1

---
 src/main/java/com/nq/ws/WebSocketClientBeanConfig.java |   21 +++++++++++++++++++--
 1 files changed, 19 insertions(+), 2 deletions(-)

diff --git a/src/main/java/com/nq/ws/WebSocketClientBeanConfig.java b/src/main/java/com/nq/ws/WebSocketClientBeanConfig.java
index 057e686..d9ffdfa 100644
--- a/src/main/java/com/nq/ws/WebSocketClientBeanConfig.java
+++ b/src/main/java/com/nq/ws/WebSocketClientBeanConfig.java
@@ -21,9 +21,27 @@
     public Map<String, WebSocketClient> websocketRunClientMap() {
 
         Map<String, WebSocketClient> retMap = new HashMap<>(2);
+        try {
+            WebsocketRunClient websocketRunClient = new WebsocketRunClient(new URI(PropertiesUtil.getProperty("IN_WS_URL")));
+            websocketRunClient.connect();
+            websocketRunClient.setConnectionLostTimeout(0);
+            new Thread(() -> {
+                while (true) {
+                    try {
+                        Thread.sleep(8000);
+                        websocketRunClient.send("heartbeat".getBytes());
+                    } catch (Exception e) {
+                        websocketRunClient.reconnect();
+                        websocketRunClient.setConnectionLostTimeout(0);
+                    }
+                }
+            }).start();
+            retMap.put(EStockType.IN.getStockKey(), websocketRunClient);
+        } catch (Exception e) {
+        }
 
         try {
-            USWebsocketRunClient usWebsocketRunClient = new USWebsocketRunClient(new URI(PropertiesUtil.getProperty("IN_WS_URL")));
+            USWebsocketRunClient usWebsocketRunClient = new USWebsocketRunClient(new URI(PropertiesUtil.getProperty("US_WS_URL")));
             usWebsocketRunClient.connect();
             usWebsocketRunClient.setConnectionLostTimeout(0);
             new Thread(() -> {
@@ -40,7 +58,6 @@
             retMap.put(EStockType.US.getStockKey(), usWebsocketRunClient);
         } catch (Exception e) {
         }
-
         return retMap;
     }
 

--
Gitblit v1.9.3