feat(gate): 添加策略运行轮数限制功能

Administrator

2026-07-22 a5e1637255f17436b3147d805ee2e9a4192c1a16

 src/main/java/com/xcong/excoin/modules/gateApi/GateGridTradeService.java

@@ -15,6 +15,7 @@
import java.math.BigDecimal;
import java.math.RoundingMode;
import java.util.*;
import java.util.concurrent.atomic.AtomicInteger;

import com.xcong.excoin.modules.gateApi.wsHandler.handler.CandlestickChannelHandler;
import com.xcong.excoin.modules.gateApi.wsHandler.handler.PositionClosesChannelHandler;
@@ -133,6 +134,9 @@
    private volatile boolean shortActive = false;
    /** 多头是否活跃（有仓位） */
    private volatile boolean longActive = false;

    /** 当前已完成轮数，每次盈利重启时+1 */
    private volatile int currentRound = 0;

    /** 多头累计止损次数（始终递增，加仓订单成交后归零） */
    private volatile int accumulatedLongLossCount = 0;
@@ -321,6 +325,7 @@
        currentShortOrderIds.clear();
        // 每次重启重新获取当前本金
        refreshInitialPrincipal();
        currentRound = 0;

        log.info("[Gate] 网格策略已启动, 当前本金: {} USDT", initialPrincipal);
    }
@@ -402,6 +407,7 @@
        executor.submitTask(this::checkProfitAndReset);

        if (state == StrategyState.ACTIVE &&
                config.isPriceDriveEnabled() &&
                longActive == false &&
                    longPositionSize.compareTo(BigDecimal.ZERO) == 0){
            processShortGrid(closePrice);
@@ -409,6 +415,7 @@


        if (state == StrategyState.ACTIVE &&
                config.isPriceDriveEnabled() &&
                shortActive == false &&
                        shortPositionSize.compareTo(BigDecimal.ZERO) == 0){
            processLongGrid(closePrice);
@@ -429,7 +436,17 @@
            BigDecimal totalEquity = new BigDecimal(account.getTotal()).add(new BigDecimal(account.getUnrealisedPnl()));

            if (totalEquity.compareTo(target) > 0) {
                log.info("[Gate] 盈亏达标(净权益{}>目标{})，重置策略", totalEquity, target);
                currentRound++;
                int maxRounds = config.getRounds();
                log.info("[Gate] 盈亏达标(净权益{}>目标{})，第{}轮完成", totalEquity, target, currentRound);

                if (maxRounds > 0 && currentRound >= maxRounds) {
                    log.info("[Gate] 已达到运行轮数上限({})，策略停止", maxRounds);
                    stopGrid();
                    return;
                }

                log.info("[Gate] 重置策略，开始第{}轮...", currentRound);
                state = StrategyState.STOPPED;
                try {
                    futuresApi.cancelPriceTriggeredOrderList(SETTLE, config.getContract());
@@ -440,7 +457,10 @@
                // 提交到 executor 末尾：单线程FIFO保证前面所有平仓/取消任务完成后才重置
                executor.submitTask(() -> {
                    try { Thread.sleep(3000); } catch (InterruptedException e) { Thread.currentThread().interrupt(); }
                    // 注意：startGrid() 会将 currentRound 归零，这里需要保留
                    int savedRound = currentRound;
                    startGrid();
                    currentRound = savedRound;
                });
            }
        } catch (Exception e) {
@@ -539,7 +559,9 @@
            // 提交到 executor 末尾：单线程FIFO保证前面所有平仓/取消任务完成后才重置
            executor.submitTask(() -> {
                try { Thread.sleep(3000); } catch (InterruptedException e) { Thread.currentThread().interrupt(); }
                int savedRound = currentRound;
                startGrid();
                currentRound = savedRound;
            });
            log.info("[Gate] 重置策略");
            return;
@@ -634,6 +656,17 @@
            if (StrUtil.isNotEmpty(tradeId) && !tradeId.equals("0") ){
                int filledQty = Integer.parseInt(shortGridElement.getShortTraderParam().getQuantity());
                shortEntryTraderIdParam(shortGridElement, orderId, false);

                // 防重入：同一网格存在多个入场单且相近时间成交时，只处理第一次 extend，
                // 后续成交打标 pendingReExtend，延展完成后自动用最新仓位重挂一次。
                if (shortGridElement.isExtendStopLossInProgress()) {
                    shortGridElement.setPendingStopLossReExtend(true);
                    log.info("[Gate] 空单成交 gridId:{}, 止损追单已在进行中，标记待重挂, orderId:{}",
                            shortGridElement.getId(), orderId);
                    return;
                }
                shortGridElement.setExtendStopLossInProgress(true);

                accumulatedShortLossCount = 0;
                lastShortStopLossGridId = Integer.MAX_VALUE;
                // [Gate-需求2] 加仓后先撤空仓所有止盈+止损，再查交易所持仓后重挂
@@ -672,8 +705,9 @@
                int shortGridQty = Integer.parseInt(config.getQuantity());
                int shortTpCount = posSize > shortBaseQty ? (posSize - shortBaseQty) / shortGridQty : 0;

                int tpSpan = config.getTakeProfitGridSpan();
                for (int i = 0; i < shortTpCount; i++) {
                    int tpGridId = shortGridElement.getId() - 2 * (i + 1);
                    int tpGridId = shortGridElement.getId() - tpSpan * (i + 1);
                    GridElement tpElem = GridElement.findById(tpGridId);
                    // 用 takeProfitPlaced 做同步标记，避免异步回调未执行时重复挂单
                    if (tpElem == null || tpElem.getShortTraderParam().isTakeProfitPlaced()) {
@@ -702,6 +736,17 @@

                int filledQty = Integer.parseInt(longGridElement.getLongTraderParam().getQuantity());
                longEntryTraderIdParam(longGridElement, orderId, false);

                // 防重入：同一网格存在多个入场单且相近时间成交时，只处理第一次 extend，
                // 后续成交打标 pendingReExtend，延展完成后自动用最新仓位重挂一次。
                if (longGridElement.isExtendStopLossInProgress()) {
                    longGridElement.setPendingStopLossReExtend(true);
                    log.info("[Gate] 多单成交 gridId:{}, 止损追单已在进行中，标记待重挂, orderId:{}",
                            longGridElement.getId(), orderId);
                    return;
                }
                longGridElement.setExtendStopLossInProgress(true);

                accumulatedLongLossCount = 0;
                lastLongStopLossGridId = Integer.MAX_VALUE;
                // [Gate-需求2] 加仓后先撤多仓所有止盈+止损，再查交易所持仓后重挂
@@ -740,8 +785,9 @@
                int longGridQty = Integer.parseInt(config.getQuantity());
                int longTpCount = posSize > longBaseQty ? (posSize - longBaseQty) / longGridQty : 0;

                int tpSpan = config.getTakeProfitGridSpan();
                for (int i = 0; i < longTpCount; i++) {
                    int tpGridId = longGridElement.getId() + 2 * (i + 1);
                    int tpGridId = longGridElement.getId() + tpSpan * (i + 1);
                    GridElement tpElem = GridElement.findById(tpGridId);
                    // 用 takeProfitPlaced 做同步标记，避免异步回调未执行时重复挂单
                    if (tpElem == null || tpElem.getLongTraderParam().isTakeProfitPlaced()) {
@@ -1664,7 +1710,10 @@
        int qty = Integer.parseInt(config.getQuantity());
        int baseQty = Integer.parseInt(config.getBaseQuantity());
        int gridCount = baseQty / qty;
        GridElement entryElem = GridElement.findById(gridId);
        if (gridCount <= 0 || filledQty < qty) {
            // 早期返回需重置防重入标记
            if (entryElem != null) { entryElem.setExtendStopLossInProgress(false); }
            return;
        }

@@ -1672,31 +1721,38 @@
        int[] orders = new int[gridCount];

        if (gridCount == 1) {
            // 只有1个网格，全量放在上面
            // 只有1个网格，全量挂在上面
            orders[0] = totalOrders;
        } else {
            int aCount = Math.max(0, totalOrders - 2);
            if (aCount == 0) {
                orders[0] = 2;
            } else if (aCount == 1) {
                orders[0] = 2;
                orders[1] = 1;
            // 最近网格(orders[0])挂剩余张数，其余每个网格各挂1张
            orders[0] = totalOrders - (gridCount - 1);
            if (orders[0] < 1) {
                // 订单数不够覆盖所有网格(理论上不会出现)，全量挂在最近网格
                orders[0] = totalOrders;
            } else {
                orders[0] = aCount;
                int remainder = totalOrders - aCount;
                for (int i = 1; i < gridCount && remainder > 0; i++) {
                for (int i = 1; i < gridCount; i++) {
                    orders[i] = 1;
                    remainder--;
                }
            }
        }

        // 计算总订单数，用于异步回调完成时重置防重入标记
        int totalOrdersToPlace = 0;
        for (int i = 0; i < gridCount; i++) {
            if (orders[i] > 0) { totalOrdersToPlace += orders[i]; }
        }
        if (totalOrdersToPlace == 0) {
            if (entryElem != null) { entryElem.setExtendStopLossInProgress(false); }
            return;
        }
        AtomicInteger remainCount = new AtomicInteger(totalOrdersToPlace);

        StringBuilder detail = new StringBuilder();
        for (int i = 0; i < gridCount; i++) {
            if (orders[i] > 0) {
                int sgId = gridId - 2 - i;
                placeStopLossOrders(sgId, orders[i], qty, FuturesPriceTrigger.RuleEnum.NUMBER_2,
                        ORDER_TYPE_CLOSE_LONG, true);
                        ORDER_TYPE_CLOSE_LONG, true, remainCount, entryElem);
                if (detail.length() > 0) {
                    detail.append("  ");
                }
@@ -1716,7 +1772,10 @@
        int qty = Integer.parseInt(config.getQuantity());
        int baseQty = Integer.parseInt(config.getBaseQuantity());
        int gridCount = baseQty / qty;
        GridElement entryElem = GridElement.findById(gridId);
        if (gridCount <= 0 || filledQty < qty) {
            // 早期返回需重置防重入标记
            if (entryElem != null) { entryElem.setExtendStopLossInProgress(false); }
            return;
        }

@@ -1724,31 +1783,38 @@
        int[] orders = new int[gridCount];

        if (gridCount == 1) {
            // 只有1个网格，全量放在上面
            // 只有1个网格，全量挂在上面
            orders[0] = totalOrders;
        } else {
            int aCount = Math.max(0, totalOrders - 2);
            if (aCount == 0) {
                orders[0] = 2;
            } else if (aCount == 1) {
                orders[0] = 2;
                orders[1] = 1;
            // 最近网格(orders[0])挂剩余张数，其余每个网格各挂1张
            orders[0] = totalOrders - (gridCount - 1);
            if (orders[0] < 1) {
                // 订单数不够覆盖所有网格(理论上不会出现)，全量挂在最近网格
                orders[0] = totalOrders;
            } else {
                orders[0] = aCount;
                int remainder = totalOrders - aCount;
                for (int i = 1; i < gridCount && remainder > 0; i++) {
                for (int i = 1; i < gridCount; i++) {
                    orders[i] = 1;
                    remainder--;
                }
            }
        }

        // 计算总订单数，用于异步回调完成时重置防重入标记
        int totalOrdersToPlace = 0;
        for (int i = 0; i < gridCount; i++) {
            if (orders[i] > 0) { totalOrdersToPlace += orders[i]; }
        }
        if (totalOrdersToPlace == 0) {
            if (entryElem != null) { entryElem.setExtendStopLossInProgress(false); }
            return;
        }
        AtomicInteger remainCount = new AtomicInteger(totalOrdersToPlace);

        StringBuilder detail = new StringBuilder();
        for (int i = 0; i < gridCount; i++) {
            if (orders[i] > 0) {
                int sgId = gridId + 2 + i;
                placeStopLossOrders(sgId, orders[i], qty, FuturesPriceTrigger.RuleEnum.NUMBER_1,
                        ORDER_TYPE_CLOSE_SHORT, false);
                        ORDER_TYPE_CLOSE_SHORT, false, remainCount, entryElem);
                if (detail.length() > 0) {
                    detail.append("  ");
                }
@@ -1761,17 +1827,66 @@
    }

    /**
     * 延展完成后重挂止损（处理被跳过的入场单成交）。
     * 取消已有止损单并用最新仓位重新挂单，确保止损覆盖最新持仓数。
     */
    private void reExtendLongStopLoss(GridElement entryElem) {
        if (entryElem.isExtendStopLossInProgress()) {
            log.info("[Gate] 多仓重挂止损跳过, entryGridId:{}, 仍在进行中", entryElem.getId());
            return;
        }
        entryElem.setExtendStopLossInProgress(true);
        cancelAllLongTakeProfitsAndStopLosses();
        int latestPos = Math.max(queryPositionSize(Position.ModeEnum.DUAL_LONG), longPositionSize.intValue());
        log.info("[Gate] 多仓重挂止损, entryGridId:{}, 最新仓位:{}张", entryElem.getId(), latestPos);
        extendLongStopLoss(latestPos, entryElem.getId());
    }

    private void reExtendShortStopLoss(GridElement entryElem) {
        if (entryElem.isExtendStopLossInProgress()) {
            log.info("[Gate] 空仓重挂止损跳过, entryGridId:{}, 仍在进行中", entryElem.getId());
            return;
        }
        entryElem.setExtendStopLossInProgress(true);
        cancelAllShortTakeProfitsAndStopLosses();
        int latestPos = Math.max(queryPositionSize(Position.ModeEnum.DUAL_SHORT), shortPositionSize.intValue());
        log.info("[Gate] 空仓重挂止损, entryGridId:{}, 最新仓位:{}张", entryElem.getId(), latestPos);
        extendShortStopLoss(latestPos, entryElem.getId());
    }

    /**
     * 在指定网格位置挂 count 个独立止损单，每个 size 张。
     */
    private void placeStopLossOrders(int gridId, int count, int qty,
                                      FuturesPriceTrigger.RuleEnum rule,
                                      String orderType, boolean isLong) {
                                      String orderType, boolean isLong,
                                      AtomicInteger remainCount,
                                      GridElement entryElement) {
        if (count <= 0) {
            return;
        }
        GridElement elem = GridElement.findById(gridId);
        if (elem == null) {
            log.warn("[Gate] 止损挂单位置不存在, gridId:{}", gridId);
            // 即使挂单位置不存在也需递减计数器，避免标志永不重置
            if (remainCount != null && entryElement != null) {
                for (int i = 0; i < count; i++) {
                    if (remainCount.decrementAndGet() == 0) {
                        entryElement.setExtendStopLossInProgress(false);
                        log.info("[Gate] {}止损追单全部完成(部分位置缺失), entryGridId:{}, 防重入标记已重置",
                                isLong ? "多仓" : "空仓", entryElement.getId());
                        // 检查待重挂请求
                        if (entryElement.isPendingStopLossReExtend()) {
                            entryElement.setPendingStopLossReExtend(false);
                            if (isLong) {
                                reExtendLongStopLoss(entryElement);
                            } else {
                                reExtendShortStopLoss(entryElement);
                            }
                        }
                    }
                }
            }
            return;
        }
        BigDecimal triggerPrice = elem.getGridPrice();
@@ -1790,6 +1905,21 @@
                        GridElement.refreshIndices();
                        log.info("[Gate] {}止损追加, gridId:{}, 触发价:{}, 第{}单, stopLossId:{}",
                                isLong ? "多仓" : "空仓", finalGridId, triggerPrice, finalI + 1, profitId);
                        // 计数器归零时重置防重入标记，并检查是否有待重挂请求
                        if (remainCount != null && remainCount.decrementAndGet() == 0 && entryElement != null) {
                            entryElement.setExtendStopLossInProgress(false);
                            log.info("[Gate] {}止损追单全部完成, entryGridId:{}, 防重入标记已重置",
                                    isLong ? "多仓" : "空仓", entryElement.getId());
                            // 如果有被跳过的入场单成交，用最新仓位重挂一次止损
                            if (entryElement.isPendingStopLossReExtend()) {
                                entryElement.setPendingStopLossReExtend(false);
                                if (isLong) {
                                    reExtendLongStopLoss(entryElement);
                                } else {
                                    reExtendShortStopLoss(entryElement);
                                }
                            }
                        }
                    }
            );
        }