From b07a82a17bb243739e7457a8885be1e8ee2bec6e Mon Sep 17 00:00:00 2001
From: Administrator <15274802129@163.com>
Date: Fri, 12 Dec 2025 17:40:00 +0800
Subject: [PATCH] feat(okx): 调整合约交易参数配置

---
 src/main/java/com/xcong/excoin/modules/okxNewPrice/OkxQuantWebSocketClient.java |  173 ++++++++++++++++++++++++++++++++++++++++++++-------------
 1 files changed, 132 insertions(+), 41 deletions(-)

diff --git a/src/main/java/com/xcong/excoin/modules/okxNewPrice/OkxQuantWebSocketClient.java b/src/main/java/com/xcong/excoin/modules/okxNewPrice/OkxQuantWebSocketClient.java
index 6a0183f..814d754 100644
--- a/src/main/java/com/xcong/excoin/modules/okxNewPrice/OkxQuantWebSocketClient.java
+++ b/src/main/java/com/xcong/excoin/modules/okxNewPrice/OkxQuantWebSocketClient.java
@@ -1,11 +1,13 @@
 package com.xcong.excoin.modules.okxNewPrice;
 
+import cn.hutool.core.util.ObjectUtil;
 import cn.hutool.core.util.StrUtil;
 import com.alibaba.fastjson.JSON;
 import com.alibaba.fastjson.JSONObject;
 import com.xcong.excoin.modules.okxNewPrice.celue.CaoZuoService;
 import com.xcong.excoin.modules.okxNewPrice.okxWs.*;
 import com.xcong.excoin.modules.okxNewPrice.okxWs.enums.CoinEnums;
+import com.xcong.excoin.modules.okxNewPrice.okxWs.enums.ExchangeInfoEnum;
 import com.xcong.excoin.modules.okxNewPrice.okxWs.enums.OrderParamEnums;
 import com.xcong.excoin.modules.okxNewPrice.utils.SSLConfig;
 import com.xcong.excoin.modules.okxNewPrice.wangge.WangGeService;
@@ -23,6 +25,7 @@
 import java.net.URI;
 import java.net.URISyntaxException;
 import java.util.concurrent.*;
+import java.util.concurrent.atomic.AtomicBoolean;
 import java.util.concurrent.atomic.AtomicReference;
 
 /**
@@ -46,10 +49,17 @@
     private ScheduledExecutorService heartbeatExecutor;
     private volatile ScheduledFuture<?> pongTimeoutFuture;
     private final AtomicReference<Long> lastMessageTime = new AtomicReference<>(System.currentTimeMillis());
+    
+    // 连接状态标志
+    private final AtomicBoolean isConnected = new AtomicBoolean(false);
+    private final AtomicBoolean isConnecting = new AtomicBoolean(false);
 
     private static final String WS_URL_MONIPAN = "wss://wspap.okx.com:8443/ws/v5/private";
     private static final String WS_URL_SHIPAN = "wss://ws.okx.com:8443/ws/v5/private";
-    private static final boolean INTERNET = false;
+
+    private ScheduledExecutorService reconnectScheduler;
+    private final AtomicReference<Long> lastReconnectTime = new AtomicReference<>(System.currentTimeMillis());
+
 
     /**
      * 订阅频道指令
@@ -75,6 +85,9 @@
     public void init() {
         connect();
         startHeartbeat();
+        
+        // 添加每小时重连的定时任务
+        schedulePeriodicReconnect();
     }
 
     /**
@@ -94,6 +107,10 @@
             pongTimeoutFuture.cancel(true);
         }
         shutdownExecutorGracefully(sharedExecutor);
+
+        if (reconnectScheduler != null) {
+            reconnectScheduler.shutdownNow();
+        }
     }
 
     private void shutdownExecutorGracefully(ExecutorService executor) {
@@ -116,22 +133,52 @@
      * 设置回调函数以监听连接打开、接收消息、关闭和错误事件。
      */
     private void connect() {
+        // 避免重复连接
+        if (isConnecting.get()) {
+            log.info("连接已在进行中,跳过重复连接请求");
+            return;
+        }
+        
+        if (!isConnecting.compareAndSet(false, true)) {
+            log.info("连接已在进行中,跳过重复连接请求");
+            return;
+        }
+        
         try {
-            InstrumentsWs.handleEvent(redisUtils);
+            InstrumentsWs.handleEvent();
             wangGeService.initWangGe();
             SSLConfig.configureSSL();
             System.setProperty("https.protocols", "TLSv1.2,TLSv1.3");
             String WS_URL = WS_URL_MONIPAN;
-            if (INTERNET){
+            if (ExchangeInfoEnum.OKX_UAT.isAccountType()){
                 WS_URL = WS_URL_SHIPAN;
             }
             URI uri = new URI(WS_URL);
+            
+            // 关闭之前的连接(如果存在)
+            if (webSocketClient != null) {
+                try {
+                    webSocketClient.closeBlocking();
+                } catch (InterruptedException e) {
+                    Thread.currentThread().interrupt();
+                    log.warn("关闭之前连接时被中断");
+                }
+            }
+            
             webSocketClient = new WebSocketClient(uri) {
                 @Override
                 public void onOpen(ServerHandshake handshake) {
                     log.info("OKX account-order WebSocket连接成功");
-                    resetHeartbeatTimer();
-                    websocketLogin();
+                    isConnected.set(true);
+                    isConnecting.set(false);
+                    
+                    // 棜查应用是否正在关闭
+                    if (!sharedExecutor.isShutdown()) {
+                        resetHeartbeatTimer();
+                        websocketLogin();
+                    } else {
+                        log.warn("应用正在关闭,忽略WebSocket连接成功回调");
+                    }
                 }
 
                 @Override
@@ -144,6 +191,8 @@
                 @Override
                 public void onClose(int code, String reason, boolean remote) {
                     log.warn("OKX account-order WebSocket连接关闭: code={}, reason={}", code, reason);
+                    isConnected.set(false);
+                    isConnecting.set(false);
                     cancelPongTimeout();
 
                     if (sharedExecutor != null && !sharedExecutor.isShutdown() && !sharedExecutor.isTerminated()) {
@@ -165,12 +214,14 @@
                 @Override
                 public void onError(Exception ex) {
                     log.error("OKX account-order WebSocket发生错误", ex);
+                    isConnected.set(false);
                 }
             };
 
             webSocketClient.connect();
         } catch (URISyntaxException e) {
             log.error("WebSocket URI格式错误", e);
+            isConnecting.set(false);
         }
     }
 
@@ -203,7 +254,7 @@
     private void handleWebSocketMessage(String message) {
         try {
             if ("pong".equals(message)) {
-                log.info("收到心跳响应");
+                log.debug("收到心跳响应");
                 cancelPongTimeout();
                 return;
             }
@@ -244,50 +295,35 @@
      * @param response 包含价格数据的 JSON 对象
      */
     private void processPushData(JSONObject response) {
+        String op = response.getString("op");
+        if (op != null){
+            if (TradeOrderWs.ORDERWS_CHANNEL.equals(op)) {
+                // 直接使用Object类型接收,避免强制类型转换
+                Object data = response.get("data");
+                log.info("收到下单推送结果: {}", JSON.toJSONString(data));
+                return;
+            }
+        }
         JSONObject arg = response.getJSONObject("arg");
         if (arg == null) {
-            log.warn("无效的推送数据,缺少 'arg' 字段");
+            log.warn("无效的推送数据,缺少 'arg' 字段 :{}",response);
             return;
         }
 
         String channel = arg.getString("channel");
         if (channel == null) {
-            log.warn("无效的推送数据,缺少 'channel' 字段");
+            log.warn("无效的推送数据,缺少 'channel' 字段{}",response);
             return;
         }
 
         if (OrderInfoWs.ORDERINFOWS_CHANNEL.equals(channel)) {
             OrderInfoWs.handleEvent(response, redisUtils);
         }else if (AccountWs.ACCOUNTWS_CHANNEL.equals(channel)) {
-            AccountWs.handleEvent(response, redisUtils);
-        } else if (PositionsWs.POSITIONSWS_CHANNEL.equals(channel)) {
-            PositionsWs.handleEvent(response, redisUtils);
-
-            String avgPxKey = PositionsWs.POSITIONSWS_CHANNEL + ":" + CoinEnums.HE_YUE.getCode() + ":avgPx";
-            String avgPx = (String) redisUtils.get(avgPxKey);
-            if (StrUtil.isBlank(avgPx)) {
-                log.error("未获取到持仓均价");
-                TradeOrderWs.orderEvent(webSocketClient, redisUtils, OrderParamEnums.INIT.getValue());
-                return;
-            }
-
-            String uplKey = PositionsWs.POSITIONSWS_CHANNEL + ":" + CoinEnums.HE_YUE.getCode() + ":upl";
-            String totalOrderUsdtKey = AccountWs.ACCOUNTWS_CHANNEL + ":" + CoinEnums.USDT.getCode() + ":totalOrderUsdt";
-
-            String upl = (String) redisUtils.get(uplKey);
-            String totalOrderUsdt = (String) redisUtils.get(totalOrderUsdtKey);
-            BigDecimal multiply = new BigDecimal(upl).multiply(new BigDecimal("-1"));
-
-            if (new BigDecimal(totalOrderUsdt).compareTo(multiply) < 0) {
-                log.error("持仓盈亏超过下单总保证金,止损冷静一天......");
-                TradeOrderWs.orderEvent(webSocketClient, redisUtils, OrderParamEnums.OUT.getValue());
-                return;
-            }
-
+            AccountWs.handleEvent(response);
             String side = caoZuoService.caoZuo();
-            if (StrUtil.isNotBlank(avgPx)) {
-                TradeOrderWs.orderEvent(webSocketClient, redisUtils, side);
-            }
+            TradeOrderWs.orderEvent(webSocketClient, side);
+        } else if (PositionsWs.POSITIONSWS_CHANNEL.equals(channel)) {
+            PositionsWs.handleEvent(response);
         } else if (BalanceAndPositionWs.CHANNEL_NAME.equals(channel)) {
             BalanceAndPositionWs.handleEvent(response);
         }
@@ -308,7 +344,27 @@
             return t;
         });
 
-        heartbeatExecutor.scheduleWithFixedDelay(this::checkHeartbeatTimeout, 25, 25, TimeUnit.SECONDS);
+        heartbeatExecutor.scheduleWithFixedDelay(this::checkHeartbeatTimeout,
+                HEARTBEAT_TIMEOUT, HEARTBEAT_TIMEOUT, TimeUnit.SECONDS);
+    }
+
+    /**
+     * 安排定期重连任务
+     * 每小时执行一次重连以保持连接新鲜度
+     */
+    private void schedulePeriodicReconnect() {
+        if (reconnectScheduler != null && !reconnectScheduler.isTerminated()) {
+            reconnectScheduler.shutdownNow();
+        }
+
+        reconnectScheduler = Executors.newSingleThreadScheduledExecutor(r -> {
+            Thread t = new Thread(r, "okx-scheduled-reconnect");
+            t.setDaemon(true);
+            return t;
+        });
+
+        // 每小时执行一次重连
+        reconnectScheduler.scheduleWithFixedDelay(this::performScheduledReconnect, 60, 60, TimeUnit.MINUTES);
     }
 
     /**
@@ -318,10 +374,25 @@
     private void resetHeartbeatTimer() {
         cancelPongTimeout();
 
-        if (heartbeatExecutor != null) {
+        // 检查线程池状态,避免在关闭过程中提交任务
+        if (heartbeatExecutor != null && !heartbeatExecutor.isShutdown()) {
             pongTimeoutFuture = heartbeatExecutor.schedule(this::checkHeartbeatTimeout,
                     HEARTBEAT_TIMEOUT, TimeUnit.SECONDS);
         }
+    }
+
+    /**
+     * 执行定时重连任务
+     * 每小时强制重连一次以确保连接的新鲜度
+     */
+    private void performScheduledReconnect() {
+        log.info("执行定时重连任务");
+        if (webSocketClient != null && webSocketClient.isOpen()) {
+            log.info("关闭当前连接准备重连");
+            webSocketClient.close();
+        }
+        // 更新最后重连时间
+        lastReconnectTime.set(System.currentTimeMillis());
     }
 
     /**
@@ -329,6 +400,11 @@
      * 若长时间未收到任何消息则主动发送 ping 请求保持连接活跃。
      */
     private void checkHeartbeatTimeout() {
+        // 只有在连接状态下才检查心跳
+        if (!isConnected.get()) {
+            return;
+        }
+        
         long currentTime = System.currentTimeMillis();
         long lastTime = lastMessageTime.get();
 
@@ -367,15 +443,30 @@
      * 在连接意外中断后尝试重新建立连接。
      */
     private void reconnectWithBackoff() throws InterruptedException {
+        // 如果正在连接,则不重复发起重连
+        if (isConnecting.get()) {
+            log.info("连接已在进行中,跳过重连请求");
+            return;
+        }
+        
         int attempt = 0;
         int maxAttempts = 5;
         long delayMs = 1000;
 
-        while (attempt < maxAttempts) {
+        while (attempt < maxAttempts && !isConnected.get()) {
             try {
                 Thread.sleep(delayMs);
                 connect();
-                return;
+                
+                // 等待连接建立
+                for (int i = 0; i < 10 && isConnecting.get(); i++) {
+                    Thread.sleep(500);
+                }
+                
+                if (isConnected.get()) {
+                    log.info("重连成功");
+                    return;
+                }
             } catch (Exception e) {
                 log.warn("第{}次重连失败", attempt + 1, e);
                 delayMs *= 2;
@@ -385,4 +476,4 @@
 
         log.error("超过最大重试次数({})仍未连接成功", maxAttempts);
     }
-}
+}
\ No newline at end of file

--
Gitblit v1.9.1