From 9ad9a7d87f0a3ddfc3c6ba5268437c260ea30634 Mon Sep 17 00:00:00 2001
From: Administrator <15274802129@163.com>
Date: Mon, 22 Jun 2026 14:22:39 +0800
Subject: [PATCH] refactor(gateApi): 优化网格交易持仓查询逻辑

---
 src/main/java/com/xcong/excoin/modules/gateApi/GateGridTradeService.java |  134 ++++++++++++++++++++++++++++++++++++++++----
 1 files changed, 120 insertions(+), 14 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 01bea3a..f787a58 100644
--- a/src/main/java/com/xcong/excoin/modules/gateApi/GateGridTradeService.java
+++ b/src/main/java/com/xcong/excoin/modules/gateApi/GateGridTradeService.java
@@ -309,6 +309,8 @@
         accumulatedShortLossCount = 0;
         shortPriceQueue.clear();
         longPriceQueue.clear();
+        totalShortPriceQueue.clear();
+        totalLongPriceQueue.clear();
         currentLongOrderIds.clear();
         currentShortOrderIds.clear();
         // 每次重启重新获取当前本金
@@ -597,20 +599,22 @@
             return;
         }
 
-        // [Gate-需求1] 多仓止盈触发:清空止盈状态 + 取消最远多仓止损
+        // [Gate-需求1] 多仓止盈触发:清空止盈状态 + 取消最远多仓止损 + 检查是否最后一个止盈
         GridElement longTpElem = GridElement.findByLongTakeProfitOrderId(orderId);
         if (longTpElem != null && StrUtil.isNotEmpty(tradeId) && !tradeId.equals("0")) {
             longTakeProfitTraderIdParam(longTpElem, null, false);
             log.info("[Gate] 多仓止盈触发 gridId:{}, orderId:{}", longTpElem.getId(), orderId);
             cancelFarthestLongStopLoss();
+            checkLastTakeProfitAndRestart();
             return;
         }
-        // [Gate-需求1] 空仓止盈触发:清空止盈状态 + 取消最远空仓止损
+        // [Gate-需求1] 空仓止盈触发:清空止盈状态 + 取消最远空仓止损 + 检查是否最后一个止盈
         GridElement shortTpElem = GridElement.findByShortTakeProfitOrderId(orderId);
         if (shortTpElem != null && StrUtil.isNotEmpty(tradeId) && !tradeId.equals("0")) {
             shortTakeProfitTraderIdParam(shortTpElem, null, false);
             log.info("[Gate] 空仓止盈触发 gridId:{}, orderId:{}", shortTpElem.getId(), orderId);
             cancelFarthestShortStopLoss();
+            checkLastTakeProfitAndRestart();
             return;
         }
 
@@ -632,17 +636,18 @@
             if (shortGridElement.isHasShortOrder() && StrUtil.isNotEmpty(tradeId) && !tradeId.equals("0") ){
                 int filledQty = Integer.parseInt(shortGridElement.getShortTraderParam().getQuantity());
                 shortEntryTraderIdParam(shortGridElement, null, false);
-                // [Gate-需求2] 加仓后先撤空仓所有止盈+止损,再重挂
+                // [Gate-需求2] 加仓后先撤空仓所有止盈+止损,再查交易所持仓后重挂
                 cancelAllShortTakeProfitsAndStopLosses();
-                extendShortStopLoss(filledQty,shortGridElement.getId());
+                int posSize = queryPositionSize(Position.ModeEnum.DUAL_SHORT);
+                extendShortStopLoss(posSize, shortGridElement.getId());
                 accumulatedShortLossCount = 0; // 加仓订单成交,重置止损累计
-                log.info("[Gate] 空单成交 gridId:{}", filledQty);
+                log.info("[Gate] 空单成交 gridId:{}, 当前持仓:{}张", filledQty, posSize);
 
                 // 空仓持仓超过baseQuantity时,从gridId-2开始向外追挂止盈
                 BigDecimal shortBaseQty = new BigDecimal(config.getBaseQuantity());
                 BigDecimal shortGridQty = new BigDecimal(config.getQuantity());
-                if (shortPositionSize.compareTo(shortBaseQty) > 0) {
-                    BigDecimal shortExcess = shortPositionSize.subtract(shortBaseQty);
+                if (BigDecimal.valueOf(posSize).compareTo(shortBaseQty) > 0) {
+                    BigDecimal shortExcess = BigDecimal.valueOf(posSize).subtract(shortBaseQty);
                     int shortExcessCount = shortExcess.divide(shortGridQty, 0, RoundingMode.DOWN).intValue();
                     for (int i = 0; i < shortExcessCount; i++) {
                         int tpGridId = shortGridElement.getId() - 2 - i;
@@ -673,17 +678,18 @@
 
                 int filledQty = Integer.parseInt(longGridElement.getLongTraderParam().getQuantity());
                 longEntryTraderIdParam(longGridElement, null, false);
-                // [Gate-需求2] 加仓后先撤多仓所有止盈+止损,再重挂
+                // [Gate-需求2] 加仓后先撤多仓所有止盈+止损,再查交易所持仓后重挂
                 cancelAllLongTakeProfitsAndStopLosses();
-                extendLongStopLoss(filledQty,longGridElement.getId());
+                int posSize = queryPositionSize(Position.ModeEnum.DUAL_LONG);
+                extendLongStopLoss(posSize, longGridElement.getId());
                 accumulatedLongLossCount = 0; // 加仓订单成交,重置止损累计
-                log.info("[Gate] 多单成交 gridId:{}", filledQty);
+                log.info("[Gate] 多单成交 gridId:{}, 当前持仓:{}张", filledQty, posSize);
 
                 // 多仓持仓超过baseQuantity时,从gridId+2开始向外追挂止盈
                 BigDecimal longBaseQty = new BigDecimal(config.getBaseQuantity());
                 BigDecimal longGridQty = new BigDecimal(config.getQuantity());
-                if (longPositionSize.compareTo(longBaseQty) > 0) {
-                    BigDecimal longExcess = longPositionSize.subtract(longBaseQty);
+                if (BigDecimal.valueOf(posSize).compareTo(longBaseQty) > 0) {
+                    BigDecimal longExcess = BigDecimal.valueOf(posSize).subtract(longBaseQty);
                     int longExcessCount = longExcess.divide(longGridQty, 0, RoundingMode.DOWN).intValue();
                     for (int i = 0; i < longExcessCount; i++) {
                         int tpGridId = longGridElement.getId() + 2 + i;
@@ -710,6 +716,29 @@
         }
     }
 
+
+    /**
+     * 查询交易所当前持仓张数(绝对值)。加仓/开仓后重挂止盈止损时调用,
+     * 绕过本地 WS 推送缓存避免时序竞态,直接拿到交易所权威数据。
+     *
+     * @param mode 持仓模式(DUAL_LONG / DUAL_SHORT)
+     * @return 持仓张数(绝对值),查询失败返回 0
+     */
+    private int queryPositionSize(Position.ModeEnum mode) {
+        try {
+            List<Position> positions = futuresApi.listPositions(SETTLE).execute();
+            if (positions != null) {
+                for (Position p : positions) {
+                    if (mode == p.getMode() && config.getContract().equals(p.getContract())) {
+                        return new BigDecimal(p.getSize()).abs().intValue();
+                    }
+                }
+            }
+        } catch (Exception e) {
+            log.warn("[Gate] 查询{}持仓失败", mode, e);
+        }
+        return 0;
+    }
 
     // ---- 网格队列处理 ----
 
@@ -878,6 +907,8 @@
      */
     private void generateShortQueue() {
         shortPriceQueue.clear();
+        totalShortPriceQueue.clear();
+        totalLongPriceQueue.clear();
         int prec = config.getPriceScale();
         BigDecimal step = shortBaseEntryPrice.multiply(config.getGridRate()).setScale(prec, RoundingMode.HALF_UP);
         config.setStep(step);
@@ -990,8 +1021,8 @@
             elements.add(GridElement.builder()
                     .id(0)
                     .gridPrice(price)
-                    .upId(shortSize > 0 ? 1 : null)
-                    .downId(longSize > 0 ? -1 : null)
+                    .upId(longSize > 0 ? 1 : null)
+                    .downId(shortSize > 0 ? -1 : null)
                     .longTraderParam(longParam)
                     .shortTraderParam(shortParam)
                     .build());
@@ -1257,6 +1288,81 @@
     // ========== 止盈/止损取消辅助方法 ==========
 
     /**
+     * 止盈触发后检查跨度是否达至要求,满足条件则重启策略。
+     *
+     * <h3>跨度定义</h3>
+     * {@code restartGridSpan} 表示多少倍的绝对步长 step(= 短基价 × gridRate)。
+     *
+     * <h3>判断逻辑</h3>
+     * <ol>
+     *   <li>多空双边均有持仓:longEntryPrice − shortEntryPrice > span × step</li>
+     *   <li>仅持多仓:currentPrice − longEntryPrice > span × step</li>
+     *   <li>仅持空仓:shortEntryPrice − currentPrice > span × step</li>
+     * </ol>
+     * restartGridSpan=0 时禁用此功能。重启复用仓位归零模式:取消全部条件单 → 平仓 → 延迟启动。
+     */
+    private void checkLastTakeProfitAndRestart() {
+        int span = config.getRestartGridSpan();
+        if (span <= 0) {
+            return;
+        }
+        BigDecimal step = config.getStep();
+        if (step == null || step.compareTo(BigDecimal.ZERO) == 0) {
+            return;
+        }
+        BigDecimal threshold = step.multiply(new BigDecimal(span));
+
+        BigDecimal currentPrice = lastKlinePrice;
+        if (currentPrice == null || currentPrice.compareTo(BigDecimal.ZERO) == 0) {
+            return;
+        }
+
+        boolean shouldRestart = false;
+        String reason = "";
+
+        if (longActive && shortActive) {
+            // 多空双边持仓:多均价 − 空均价 > span × step
+            BigDecimal gap = longEntryPrice.subtract(shortEntryPrice);
+            if (gap.compareTo(threshold) > 0) {
+                shouldRestart = true;
+                reason = StrUtil.format("双边跨度 多均价:{} − 空均价:{} = {} > {} (span:{}×step:{})",
+                        longEntryPrice, shortEntryPrice, gap, threshold, span, step);
+            }
+        } else if (longActive) {
+            // 仅持多仓:当前价 − 多均价 > span × step
+            BigDecimal gap = currentPrice.subtract(longEntryPrice);
+            if (gap.compareTo(threshold) > 0) {
+                shouldRestart = true;
+                reason = StrUtil.format("多仓跨度 当前价:{} − 多均价:{} = {} > {} (span:{}×step:{})",
+                        currentPrice, longEntryPrice, gap, threshold, span, step);
+            }
+        } else if (shortActive) {
+            // 仅持空仓:空均价 − 当前价 > span × step
+            BigDecimal gap = shortEntryPrice.subtract(currentPrice);
+            if (gap.compareTo(threshold) > 0) {
+                shouldRestart = true;
+                reason = StrUtil.format("空仓跨度 空均价:{} − 当前价:{} = {} > {} (span:{}×step:{})",
+                        shortEntryPrice, currentPrice, gap, threshold, span, step);
+            }
+        }
+
+        if (shouldRestart) {
+            log.info("[Gate] 跨度已达要求 → {},最后一个止盈触发策略重启", reason);
+            try {
+                futuresApi.cancelPriceTriggeredOrderList(SETTLE, config.getContract());
+            } catch (ApiException ex) {
+                log.warn("[Gate] 重启前清理条件单失败", ex);
+            }
+            closeExistingPositions();
+            state = StrategyState.STOPPED;
+            executor.submitTask(() -> {
+                try { Thread.sleep(3000); } catch (InterruptedException ex) { Thread.currentThread().interrupt(); }
+                startGrid();
+            });
+        }
+    }
+
+    /**
      * 取消最远的多仓止损订单。
      * 多仓止损在 gridId 负方向,最远 = id 最小。
      */

--
Gitblit v1.9.1