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

---
 src/main/java/com/xcong/excoin/modules/gateApi/GateGridTradeService.java |   80 +++++++++++++++++++++++++++++++++-------
 1 files changed, 66 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 dc78d98..3bf8290 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,6 +599,23 @@
             return;
         }
 
+        // [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();
+            return;
+        }
+        // [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();
+            return;
+        }
+
         GridElement longStopLossElem = GridElement.findByLongStopLossOrderId(orderId);
 //        if (longStopLossElem != null && longPositionSize.compareTo(BigDecimal.ZERO) > 0 && StrUtil.isNotEmpty(tradeId) && !tradeId.equals("0")) {
         if (longStopLossElem != null && StrUtil.isNotEmpty(tradeId) && !tradeId.equals("0")) {
@@ -615,15 +634,18 @@
             if (shortGridElement.isHasShortOrder() && StrUtil.isNotEmpty(tradeId) && !tradeId.equals("0") ){
                 int filledQty = Integer.parseInt(shortGridElement.getShortTraderParam().getQuantity());
                 shortEntryTraderIdParam(shortGridElement, null, false);
-                extendShortStopLoss(filledQty,shortGridElement.getId());
+                // [Gate-需求2] 加仓后先撤空仓所有止盈+止损,再查交易所持仓后重挂
+                cancelAllShortTakeProfitsAndStopLosses();
+                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;
@@ -654,15 +676,18 @@
 
                 int filledQty = Integer.parseInt(longGridElement.getLongTraderParam().getQuantity());
                 longEntryTraderIdParam(longGridElement, null, false);
-                extendLongStopLoss(filledQty,longGridElement.getId());
+                // [Gate-需求2] 加仓后先撤多仓所有止盈+止损,再查交易所持仓后重挂
+                cancelAllLongTakeProfitsAndStopLosses();
+                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;
@@ -689,6 +714,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;
+    }
 
     // ---- 网格队列处理 ----
 
@@ -857,6 +905,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);
@@ -969,8 +1019,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());
@@ -1246,8 +1296,9 @@
             furthestSlId = gridId;
             interval = 2;
         }
-        log.info("[Gate] 多仓追挂止损, 当前最远止损gridId:{}, 追加{}张", furthestSlId, filledQty);
-        for (int i = 0; i < filledQty; i++) {
+        int stopLossCount = filledQty / Integer.parseInt(config.getQuantity());
+        log.info("[Gate] 多仓追挂止损, 当前最远止损gridId:{}, 成交{}张, 追加{}个止损单", furthestSlId, filledQty, stopLossCount);
+        for (int i = 0; i < stopLossCount; i++) {
             int newSlId = furthestSlId - i - interval;
             GridElement elem = GridElement.findById(newSlId);
             if (elem == null) {
@@ -1282,8 +1333,9 @@
             furthestSlId = gridId;
             interval = 2;
         }
-        log.info("[Gate] 空仓追挂止损, 当前最远止损gridId:{}, 追加{}张", furthestSlId, filledQty);
-        for (int i = 0; i < filledQty; i++) {
+        int stopLossCount = filledQty / Integer.parseInt(config.getQuantity());
+        log.info("[Gate] 空仓追挂止损, 当前最远止损gridId:{}, 成交{}张, 追加{}个止损单", furthestSlId, filledQty, stopLossCount);
+        for (int i = 0; i < stopLossCount; i++) {
             int newSlId = furthestSlId + i + interval;
             GridElement elem = GridElement.findById(newSlId);
             if (elem == null) {

--
Gitblit v1.9.1