| | |
| | | package com.nq.ws; |
| | | |
| | | |
| | | import com.nq.enums.EStockType; |
| | | import com.nq.utils.PropertiesUtil; |
| | | import lombok.extern.slf4j.Slf4j; |
| | |
| | | public Map<String, WebSocketClient> websocketRunClientMap() { |
| | | |
| | | Map<String, WebSocketClient> retMap = new HashMap<>(2); |
| | | // try { |
| | | // JPWebsocketRunClient jpWebsocketRunClient = new JPWebsocketRunClient(new URI(PropertiesUtil.getProperty("JP_WS_URL")), EStockType.JP); |
| | | // jpWebsocketRunClient.connect(); |
| | | // jpWebsocketRunClient.setConnectionLostTimeout(0); |
| | | // startHeartbeatThread(jpWebsocketRunClient); |
| | | // retMap.put(EStockType.JP.getStockKey(), jpWebsocketRunClient); |
| | | // } catch (Exception e) { |
| | | // log.error("jpWebsocketRunClient 异常: {}", e.getMessage()); |
| | | // } |
| | | try { |
| | | WebsocketRunClient websocketRunClient = new WebsocketRunClient(new URI(PropertiesUtil.getProperty("JS_IN_WS_URL")),EStockType.IN); |
| | | websocketRunClient.connect(); |
| | | websocketRunClient.setConnectionLostTimeout(0); |
| | | new Thread(() -> { |
| | | while (true) { |
| | | USWebsocketRunClient usWebsocketRunClient = new USWebsocketRunClient(new URI(PropertiesUtil.getProperty("US_WS_URL")), EStockType.US); |
| | | usWebsocketRunClient.connect(); |
| | | usWebsocketRunClient.setConnectionLostTimeout(0); |
| | | startHeartbeatThread(usWebsocketRunClient); |
| | | retMap.put(EStockType.US.getStockKey(), usWebsocketRunClient); |
| | | } catch (Exception e) { |
| | | log.error("usWebsocketRunClient 异常: {}", e.getMessage()); |
| | | } |
| | | return retMap; |
| | | } |
| | | |
| | | private void startHeartbeatThread(WebSocketClient client) { |
| | | new Thread(() -> { |
| | | while (true) { |
| | | try { |
| | | Thread.sleep(8000); |
| | | if (client.isOpen()) { // 先检查连接状态 |
| | | client.send("heartbeat".getBytes()); |
| | | } else { |
| | | client.reconnect(); |
| | | client.setConnectionLostTimeout(0); |
| | | } |
| | | } catch (Exception e) { |
| | | log.error("心跳线程异常, 尝试重连: {}", e.getMessage()); |
| | | try { |
| | | Thread.sleep(8000); |
| | | websocketRunClient.send("heartbeat".getBytes()); |
| | | } catch (Exception e) { |
| | | websocketRunClient.reconnect(); |
| | | websocketRunClient.setConnectionLostTimeout(0); |
| | | client.reconnect(); |
| | | client.setConnectionLostTimeout(0); |
| | | } catch (Exception re) { |
| | | log.error("重连失败: {}", re.getMessage()); |
| | | } |
| | | } |
| | | }).start(); |
| | | } catch (Exception e) { |
| | | } |
| | | |
| | | |
| | | |
| | | return retMap; |
| | | } |
| | | }).start(); |
| | | } |
| | | |
| | | |
| | | } |
| | | |