From ba9e35f79fa9a703992c58c5951cd1593749172e Mon Sep 17 00:00:00 2001
From: ydj <yangsink@163.com>
Date: Mon, 03 Jun 2024 17:58:58 +0800
Subject: [PATCH] 1

---
 src/main/java/com/nq/ws/WebSocketClientBeanConfig.java |  134 ++++++++++++++++++++++++++++++++------------
 1 files changed, 96 insertions(+), 38 deletions(-)

diff --git a/src/main/java/com/nq/ws/WebSocketClientBeanConfig.java b/src/main/java/com/nq/ws/WebSocketClientBeanConfig.java
index 720492e..b87e2da 100644
--- a/src/main/java/com/nq/ws/WebSocketClientBeanConfig.java
+++ b/src/main/java/com/nq/ws/WebSocketClientBeanConfig.java
@@ -19,49 +19,107 @@
 
     @Bean
     public Map<String, WebSocketClient> websocketRunClientMap() {
-
         Map<String, WebSocketClient> retMap = new HashMap<>(2);
 
-
-
-
-        try {
-            WebsocketRunClient websocketRunClient = new WebsocketRunClient(new URI(PropertiesUtil.getProperty("US_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);
+        for (EStockType value : EStockType.values()) {
+            try {
+                WebsocketRunClient websocketRunClient = new WebsocketRunClient(new URI(PropertiesUtil.getProperty(value.getWsUrl())));
+                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) {
+                }).start();
+                retMap.put(EStockType.MAS.getStockKey(), websocketRunClient);
+            } catch (Exception e) {
+            }
         }
 
-        try {
-            USWebsocketRunClient usWebsocketRunClient = new USWebsocketRunClient(new URI(PropertiesUtil.getProperty("MAS_WS_URL")));
-            usWebsocketRunClient.connect();
-            usWebsocketRunClient.setConnectionLostTimeout(0);
-            new Thread(() -> {
-                while (true) {
-                    try {
-                        Thread.sleep(8000);
-                        usWebsocketRunClient.send("heartbeat".getBytes());
-                    } catch (Exception e) {
-                        usWebsocketRunClient.reconnect();
-                        usWebsocketRunClient.setConnectionLostTimeout(0);
-                    }
-                }
-            }).start();
-            retMap.put(EStockType.US.getStockKey(), usWebsocketRunClient);
-        } catch (Exception e) {
-        }
+
+//        try {
+//            WebsocketRunClient websocketRunClient = new WebsocketRunClient(new URI(PropertiesUtil.getProperty("MAS_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.MAS.getStockKey(), websocketRunClient);
+//        } catch (Exception e) {
+//        }
+//
+//        try {
+//            USWebsocketRunClient usWebsocketRunClient = new USWebsocketRunClient(new URI(PropertiesUtil.getProperty("US_WS_URL")));
+//            usWebsocketRunClient.connect();
+//            usWebsocketRunClient.setConnectionLostTimeout(0);
+//            new Thread(() -> {
+//                while (true) {
+//                    try {
+//                        Thread.sleep(8000);
+//                        usWebsocketRunClient.send("heartbeat".getBytes());
+//                    } catch (Exception e) {
+//                        usWebsocketRunClient.reconnect();
+//                        usWebsocketRunClient.setConnectionLostTimeout(0);
+//                    }
+//                }
+//            }).start();
+//            retMap.put(EStockType.US.getStockKey(), usWebsocketRunClient);
+//        } catch (Exception e) {
+//        }
+//
+//
+//        try {
+//            HKWebsocketRunClient hkWebsocketRunClient = new HKWebsocketRunClient(new URI(PropertiesUtil.getProperty("HK_WS_URL")));
+//            hkWebsocketRunClient.connect();
+//            hkWebsocketRunClient.setConnectionLostTimeout(0);
+//            new Thread(() -> {
+//                while (true) {
+//                    try {
+//                        Thread.sleep(8000);
+//                        hkWebsocketRunClient.send("heartbeat".getBytes());
+//                    } catch (Exception e) {
+//                        hkWebsocketRunClient.reconnect();
+//                        hkWebsocketRunClient.setConnectionLostTimeout(0);
+//                    }
+//                }
+//            }).start();
+//            retMap.put(EStockType.HK.getStockKey(), hkWebsocketRunClient);
+//        } catch (Exception e) {
+//        }
+//
+//        try {
+//            HKWebsocketRunClient hkWebsocketRunClient = new HKWebsocketRunClient(new URI(PropertiesUtil.getProperty("IN_WS_URL")));
+//            hkWebsocketRunClient.connect();
+//            hkWebsocketRunClient.setConnectionLostTimeout(0);
+//            new Thread(() -> {
+//                while (true) {
+//                    try {
+//                        Thread.sleep(8000);
+//                        hkWebsocketRunClient.send("heartbeat".getBytes());
+//                    } catch (Exception e) {
+//                        hkWebsocketRunClient.reconnect();
+//                        hkWebsocketRunClient.setConnectionLostTimeout(0);
+//                    }
+//                }
+//            }).start();
+//            retMap.put(EStockType.HK.getStockKey(), hkWebsocketRunClient);
+//        } catch (Exception e) {
+//        }
+
 
         return retMap;
     }

--
Gitblit v1.9.3