From b6deca01c4d6be9f732e0feb976b44b9bd822925 Mon Sep 17 00:00:00 2001
From: Administrator <15274802129@163.com>
Date: Tue, 02 Jun 2026 17:54:10 +0800
Subject: [PATCH] fix(gateapi): 解决WebSocket连接超时问题

---
 src/main/java/com/xcong/excoin/modules/gateApi/GateGridTradeService.java |  127 ++++++++++++++++++++++++-----------------
 1 files changed, 74 insertions(+), 53 deletions(-)

diff --git a/src/main/java/com/xcong/excoin/modules/gateApi/GateGridTradeService.java b/src/main/java/com/xcong/excoin/modules/gateApi/GateGridTradeService.java
index a51fb12..e277173 100644
--- a/src/main/java/com/xcong/excoin/modules/gateApi/GateGridTradeService.java
+++ b/src/main/java/com/xcong/excoin/modules/gateApi/GateGridTradeService.java
@@ -305,7 +305,23 @@
         longPriceQueue.clear();
         currentLongOrderIds.clear();
         currentShortOrderIds.clear();
-        log.info("[Gate] 网格策略已启动");
+
+        // 每次重启重新获取当前本金
+        refreshInitialPrincipal();
+
+        log.info("[Gate] 网格策略已启动, 当前本金: {} USDT", initialPrincipal);
+    }
+
+    /**
+     * 重新获取当前账户权益作为初始本金。
+     */
+    private void refreshInitialPrincipal() {
+        try {
+            FuturesAccount account = futuresApi.listFuturesAccounts(SETTLE);
+            this.initialPrincipal = new BigDecimal(account.getTotal());
+        } catch (Exception e) {
+            log.warn("[Gate] 获取初始化本金失败,使用旧值: {}", initialPrincipal);
+        }
     }
 
     /**
@@ -353,6 +369,8 @@
             BigDecimal totalPnl = cumulativePnl.add(unrealizedPnl);
             log.info("[Gate] 已实现:{}, 未实现:{}, 合计:{}",
                     cumulativePnl, unrealizedPnl, totalPnl);
+
+            startGrid();
             return;
         }
 
@@ -427,8 +445,8 @@
                     tryGenerateQueues();
                 }else {
                     longPositionSize = size;
-                    checkShortEntryOrderToCancel();
-                    checkLongEntryOrderToCancel();
+//                    checkShortEntryOrderToCancel();
+//                    checkLongEntryOrderToCancel();
                 }
             } else {
                 if (longActive && state == StrategyState.ACTIVE) {
@@ -450,8 +468,8 @@
                     tryGenerateQueues();
                 }else {
                     shortPositionSize = size.abs();
-                    checkShortEntryOrderToCancel();
-                    checkLongEntryOrderToCancel();
+//                    checkShortEntryOrderToCancel();
+//                    checkLongEntryOrderToCancel();
                 }
             } else {
                 if (shortActive && state == StrategyState.ACTIVE) {
@@ -559,19 +577,12 @@
         BigDecimal totalPnl = cumulativePnl.add(unrealizedPnl);
         log.info("[Gate] 已实现:{}, 未实现:{}, 合计:{}",
                 cumulativePnl, unrealizedPnl, totalPnl);
-
-        if (totalPnl.compareTo(config.getOverallTp()) >= 0) {
-            log.info("[Gate] 已达止盈目标(合计{})→已停止, 已实现:{}, 未实现:{}",
-                    totalPnl, cumulativePnl, unrealizedPnl);
-            state = StrategyState.STOPPED;
-        } else if (totalPnl.compareTo(config.getMaxLoss().negate()) <= 0) {
+        if(totalPnl.compareTo(config.getMaxLoss().negate()) <= 0) {
             String logMessage = StrUtil.format("[Gate] 已达亏损风险值(合计{}), 已实现:{}, 未实现:{}",
                     totalPnl, cumulativePnl, unrealizedPnl);
             log.info(logMessage);
 
-
             DingTalkUtils.getDefault().sendActionCard("风险提醒", logMessage, config.getApiKey(), "");
-//            state = StrategyState.STOPPED;
         }
     }
 
@@ -734,34 +745,34 @@
             return;
         }
 
-        GridElement byShortTakeProfitOrderId = GridElement.findByShortTakeProfitOrderId(orderId);
-        if (byShortTakeProfitOrderId != null){
-            shortTakeProfitTraderIdParam(
-                    byShortTakeProfitOrderId,
-                    null,
-                    false
-            );
-            shortEntryTraderIdParam(
-                    byShortTakeProfitOrderId,
-                    null,
-                    false
-            );
-            TPonUserTradeShortEntry(byShortTakeProfitOrderId);
-        }
-        GridElement byLongTakeProfitOrderId = GridElement.findByLongTakeProfitOrderId(orderId);
-        if (byLongTakeProfitOrderId != null){
-            longTakeProfitTraderIdParam(
-                    byLongTakeProfitOrderId,
-                    null,
-                    false
-            );
-            longEntryTraderIdParam(
-                    byLongTakeProfitOrderId,
-                    null,
-                    false
-            );
-            TPonUserTradeLongEntry(byLongTakeProfitOrderId);
-        }
+//        GridElement byShortTakeProfitOrderId = GridElement.findByShortTakeProfitOrderId(orderId);
+//        if (byShortTakeProfitOrderId != null){
+//            shortTakeProfitTraderIdParam(
+//                    byShortTakeProfitOrderId,
+//                    null,
+//                    false
+//            );
+//            shortEntryTraderIdParam(
+//                    byShortTakeProfitOrderId,
+//                    null,
+//                    false
+//            );
+//            TPonUserTradeShortEntry(byShortTakeProfitOrderId);
+//        }
+//        GridElement byLongTakeProfitOrderId = GridElement.findByLongTakeProfitOrderId(orderId);
+//        if (byLongTakeProfitOrderId != null){
+//            longTakeProfitTraderIdParam(
+//                    byLongTakeProfitOrderId,
+//                    null,
+//                    false
+//            );
+//            longEntryTraderIdParam(
+//                    byLongTakeProfitOrderId,
+//                    null,
+//                    false
+//            );
+//            TPonUserTradeLongEntry(byLongTakeProfitOrderId);
+//        }
 
         GridElement shortGridElement = GridElement.findByShortOrderId(orderId);
         if (shortGridElement != null) {
@@ -913,7 +924,7 @@
                         triggerPrice,
                         FuturesPriceTrigger.RuleEnum.NUMBER_1,
                         ORDER_TYPE_CLOSE_SHORT,
-                        "1",
+                        config.getQuantity(),
                         profitId -> {
                             elem.setShortStopLossOrderId(profitId);
                             GridElement.refreshIndices();
@@ -933,7 +944,7 @@
                         triggerPrice,
                         FuturesPriceTrigger.RuleEnum.NUMBER_2,
                         ORDER_TYPE_CLOSE_LONG,
-                        "-1",
+                        negate(config.getQuantity()),
                         profitId -> {
                             elem.setLongStopLossOrderId(profitId);
                             GridElement.refreshIndices();
@@ -1357,7 +1368,6 @@
         log.info("[Gate] 多仓止损触发 gridId:{}, 开始追单", gridId);
 
         int newEntryGridId = -(N - 1);
-        int entryQty = N - 1;
 
         GridElement newEntryGrid = GridElement.findById(newEntryGridId);
         if (newEntryGrid == null) {
@@ -1377,12 +1387,16 @@
             }
         }
 
-        String size = String.valueOf(entryQty);
         BigDecimal triggerPrice = newEntryGrid.getGridPrice();
-        log.info("[Gate] 多仓止损触发 gridId:{}, 在gridId:{}挂{}张多单", gridId, newEntryGridId, entryQty);
+        BigDecimal priceDiff = longEntryPrice.subtract(triggerPrice).abs();
+        int entryQty = priceDiff.divide(config.getStep(), 0, RoundingMode.DOWN).intValue();
+        entryQty = Math.max(1, entryQty);
+        String size = new BigDecimal(String.valueOf(entryQty)).multiply(new BigDecimal(config.getQuantity())).toString();
+        log.info("[Gate] 多仓止损触发 gridId:{}, 在gridId:{}挂{}张多单, 均价:{}, 价差:{}, 步长:{}",
+                gridId, newEntryGridId, entryQty, longEntryPrice, priceDiff, config.getStep());
         newEntryGrid.getLongTraderParam().setQuantity(size);
         placeEntryOrderWithPreFlag(newEntryGrid, true, triggerPrice,
-                FuturesPriceTrigger.RuleEnum.NUMBER_2, size);
+                FuturesPriceTrigger.RuleEnum.NUMBER_1, size);
     }
 
     private void handleShortStopLossTriggered(GridElement gridElement) {
@@ -1392,7 +1406,6 @@
         log.info("[Gate] 空仓止损触发 gridId:{}, 开始追单", gridId);
 
         int newEntryGridId = N - 1;
-        int entryQty = N - 1;
 
         GridElement newEntryGrid = GridElement.findById(newEntryGridId);
         if (newEntryGrid == null) {
@@ -1412,12 +1425,18 @@
             }
         }
 
-        String size = String.valueOf(entryQty);
+
+
         BigDecimal triggerPrice = newEntryGrid.getGridPrice();
-        log.info("[Gate] 空仓止损触发 gridId:{}, 在gridId:{}挂{}张空单", gridId, newEntryGridId, entryQty);
+        BigDecimal priceDiff = shortEntryPrice.subtract(triggerPrice).abs();
+        int entryQty = priceDiff.divide(config.getStep(), 0, RoundingMode.DOWN).intValue();
+        entryQty = Math.max(1, entryQty);
+        String size = new BigDecimal(String.valueOf(entryQty)).multiply(new BigDecimal(config.getQuantity())).toString();
+        log.info("[Gate] 空仓止损触发 gridId:{}, 在gridId:{}挂{}张空单, 均价:{}, 价差:{}, 步长:{}",
+                gridId, newEntryGridId, entryQty, shortEntryPrice, priceDiff, config.getStep());
         newEntryGrid.getShortTraderParam().setQuantity(size);
         placeEntryOrderWithPreFlag(newEntryGrid, false, triggerPrice,
-                FuturesPriceTrigger.RuleEnum.NUMBER_1, negate(size));
+                FuturesPriceTrigger.RuleEnum.NUMBER_2, negate(size));
     }
 
     private void extendLongStopLoss(int filledQty) {
@@ -1443,7 +1462,7 @@
                     triggerPrice,
                     FuturesPriceTrigger.RuleEnum.NUMBER_2,
                     ORDER_TYPE_CLOSE_LONG,
-                    "-1",
+                    negate(config.getQuantity()),
                     profitId -> {
                         elem.setLongStopLossOrderId(profitId);
                         GridElement.refreshIndices();
@@ -1476,7 +1495,7 @@
                     triggerPrice,
                     FuturesPriceTrigger.RuleEnum.NUMBER_1,
                     ORDER_TYPE_CLOSE_SHORT,
-                    "1",
+                    config.getQuantity(),
                     profitId -> {
                         elem.setShortStopLossOrderId(profitId);
                         GridElement.refreshIndices();
@@ -1497,6 +1516,7 @@
                     unrealisedPnl, available, totalEquity, target);
             if (totalEquity.compareTo(target) > 0) {
                 log.info("[Gate] 盈亏达标({}>{}),重置策略", totalEquity, target);
+                state = StrategyState.STOPPED;
                 closeExistingPositions();
                 futuresApi.cancelPriceTriggeredOrderList(SETTLE, config.getContract());
                 startGrid();
@@ -1507,6 +1527,7 @@
     }
 
     private void handlePositionZeroAndReset(String direction) {
+        state = StrategyState.STOPPED;
         try {
             futuresApi.cancelPriceTriggeredOrderList(SETTLE, config.getContract());
         } catch (Exception e) {

--
Gitblit v1.9.1