fix(gateApi): 修复网格交易盈亏计算逻辑

Administrator

2026-06-08 4c2987e3566f34087aa6251006b78326b4973dd9

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

@@ -11,9 +11,6 @@
import io.gate.gateapi.models.*;
import lombok.extern.slf4j.Slf4j;

import okhttp3.ConnectionPool;
import okhttp3.OkHttpClient;

import java.io.IOException;
import java.math.BigDecimal;
import java.math.RoundingMode;
@@ -22,7 +19,6 @@
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;
import java.util.concurrent.TimeUnit;

import com.xcong.excoin.modules.gateApi.wsHandler.handler.CandlestickChannelHandler;
import com.xcong.excoin.modules.gateApi.wsHandler.handler.PositionClosesChannelHandler;
@@ -150,27 +146,20 @@
    private volatile BigDecimal shortPositionSize = BigDecimal.ZERO;
    private Long userId;
    private volatile BigDecimal initialPrincipal = BigDecimal.ZERO;
    private volatile GateKlineWebSocketClient wsClient;

    /** 多仓挂单张数计数器：止损触发时用当前值挂单，随后+1；挂单成交后重置为1 */
    private volatile int longEntryQty = 1;
    /** 空仓挂单张数计数器：止损触发时用当前值挂单，随后+1；挂单成交后重置为1 */
    private volatile int shortEntryQty = 1;

    public GateGridTradeService(GateConfig config) {
        this.config = config;
        ApiClient apiClient = createApiClient(config);
        ApiClient apiClient = new ApiClient();
        apiClient.setBasePath(config.getRestBasePath());
        apiClient.setApiKeySecret(config.getApiKey(), config.getApiSecret());
        this.futuresApi = new FuturesApi(apiClient);
        this.executor = new GateTradeExecutor(apiClient, config.getContract());
    }

    private static ApiClient createApiClient(GateConfig config) {
        OkHttpClient httpClient = new OkHttpClient.Builder()
                .connectTimeout(30, TimeUnit.SECONDS)
                .readTimeout(60, TimeUnit.SECONDS)
                .writeTimeout(20, TimeUnit.SECONDS)
                .connectionPool(new ConnectionPool(5, 5, TimeUnit.SECONDS))
                .retryOnConnectionFailure(true)
                .build();
        ApiClient client = new ApiClient();
        client.setBasePath(config.getRestBasePath());
        client.setApiKeySecret(config.getApiKey(), config.getApiSecret());
        client.setHttpClient(httpClient);
        return client;
    }

    // ---- 初始化 ----
@@ -191,7 +180,9 @@
     */
    public void init() {
        try {
            ApiClient detailClient = createApiClient(config);
            ApiClient detailClient = new ApiClient();
            detailClient.setBasePath(config.getRestBasePath());
            detailClient.setApiKeySecret(config.getApiKey(), config.getApiSecret());
            AccountDetail detail = new AccountApi(detailClient).getAccountDetail();
            this.userId = detail.getUserId();
            log.info("[Gate] 用户ID: {}", userId);
@@ -320,6 +311,8 @@
        longPriceQueue.clear();
        currentLongOrderIds.clear();
        currentShortOrderIds.clear();
        longEntryQty = 1;
        shortEntryQty = 1;

        // 每次重启重新获取当前本金
        refreshInitialPrincipal();
@@ -391,6 +384,9 @@

        //初始化0位置的开仓，并且用空的开仓价格，作为价格基准来划分网格
        if (state == StrategyState.WAITING_KLINE) {
            if (wsClient == null || !wsClient.areAllSubscribed()) {
                return;
            }
            state = StrategyState.OPENING;
            log.info("[Gate] 首根K线到达，开基底仓位 多空各{}张...", config.getBaseQuantity());
            executor.openLong(config.getBaseQuantity(), (orderId) -> {
@@ -794,8 +790,10 @@
            if (shortGridElement.isHasShortOrder() && !tradeId.equals("0")){
                int filledQty = Integer.parseInt(shortGridElement.getShortTraderParam().getQuantity());
                shortEntryTraderIdParam(shortGridElement, null, false);
                shortEntryQty = 1;
                extendShortStopLoss(filledQty);
                log.info("[Gate] 空单成交 gridId:{}, qty:{}, 追挂止损", shortGridElement.getId(), filledQty);
//                checkMaxPositionAndPlaceTakeProfit(false, shortGridElement.getId());
            }
        }
        GridElement longGridElement = GridElement.findByLongOrderId(orderId);
@@ -803,8 +801,10 @@
            if (longGridElement.isHasLongOrder() && !tradeId.equals("0")){
                int filledQty = Integer.parseInt(longGridElement.getLongTraderParam().getQuantity());
                longEntryTraderIdParam(longGridElement, null, false);
                longEntryQty = 1;
                extendLongStopLoss(filledQty);
                log.info("[Gate] 多单成交 gridId:{}, qty:{}, 追挂止损", longGridElement.getId(), filledQty);
//                checkMaxPositionAndPlaceTakeProfit(true, longGridElement.getId());
            }
        }
    }
@@ -928,7 +928,8 @@
            baseGridElement.setShortOrderId(baseShortTraderParam.getEntryOrderId());
            baseGridElement.setHasShortOrder(true);

            for (int id = 2; id <= 11; id++) {
            int shortTime = Integer.parseInt(config.getBaseQuantity()) + 1;
            for (int id = 2; id <= shortTime; id++) {
                GridElement elem = GridElement.findById(id);
                if (elem == null) {
                    continue;
@@ -948,7 +949,9 @@
                );
            }

            for (int id = -2; id >= -11; id--) {

            int longTime = Integer.parseInt(config.getBaseQuantity()) + 1;
            for (int id = -2; id >= -longTime; id--) {
                GridElement elem = GridElement.findById(id);
                if (elem == null) {
                    continue;
@@ -968,7 +971,7 @@
                );
            }

            log.info("[Gate] 止损单已全部挂完, 空仓止损: 2~11, 多仓止损: -2~-11");
            log.info("[Gate] 止损单已全部挂完, 空仓止损: 2~{}, 多仓止损: -2~-{}", shortTime, longTime);
            state = StrategyState.ACTIVE;
        }
    }
@@ -1403,12 +1406,11 @@
        }

        BigDecimal triggerPrice = newEntryGrid.getGridPrice();
        BigDecimal priceDiff = longEntryPrice.subtract(triggerPrice).abs();
        int entryQty = priceDiff.divide(config.getStep(), 0, RoundingMode.DOWN).intValue();
        entryQty = Math.max(1, entryQty);
        longEntryQty++;
        int entryQty = longEntryQty;
        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());
        log.info("[Gate] 多仓止损触发 gridId:{}, 在gridId:{}挂{}基础张多单(计数器:{}, size:{})",
                gridId, newEntryGridId, entryQty, longEntryQty, size);
        newEntryGrid.getLongTraderParam().setQuantity(size);
        placeEntryOrderWithPreFlag(newEntryGrid, true, triggerPrice,
                FuturesPriceTrigger.RuleEnum.NUMBER_1, size);
@@ -1443,12 +1445,11 @@


        BigDecimal triggerPrice = newEntryGrid.getGridPrice();
        BigDecimal priceDiff = shortEntryPrice.subtract(triggerPrice).abs();
        int entryQty = priceDiff.divide(config.getStep(), 0, RoundingMode.DOWN).intValue();
        entryQty = Math.max(1, entryQty);
        shortEntryQty++;
        int entryQty = shortEntryQty;
        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());
        log.info("[Gate] 空仓止损触发 gridId:{}, 在gridId:{}挂{}基础张空单(计数器:{}, size:{})",
                gridId, newEntryGridId, entryQty, shortEntryQty, size);
        newEntryGrid.getShortTraderParam().setQuantity(size);
        placeEntryOrderWithPreFlag(newEntryGrid, false, triggerPrice,
                FuturesPriceTrigger.RuleEnum.NUMBER_2, negate(size));
@@ -1520,21 +1521,103 @@
        }
    }

    /**
     * 通过 REST API 查询实时持仓，超限则在下一网格挂止盈单。
     * 使用异步执行避免阻塞 WS 回调线程。
     *
     * @param isLong 是否为多仓方向
     * @param gridId 当前挂单成交的网格 ID
     */
    private void checkMaxPositionAndPlaceTakeProfit(boolean isLong, int gridId) {
        int maxPos = config.getMaxPositionSize();
        if (maxPos <= 0) return;

        executor.submitTask(() -> {
            try {
                List<Position> positions = futuresApi.listPositions(SETTLE).execute();
                if (positions == null) return;

                long actualPosSize = 0;
                String targetMode = isLong ? "dual_long" : "dual_short";
                for (Position pos : positions) {
                    if (!config.getContract().equals(pos.getContract())) continue;
                    Position.ModeEnum mode = pos.getMode();
                    if (mode != null && mode.getValue() != null && targetMode.equals(mode.getValue())) {
                        actualPosSize = Math.abs(Long.parseLong(pos.getSize()));
                        break;
                    }
                }

                if (actualPosSize <= maxPos) {
                    log.debug("[Gate] 持仓检查 {}方向 实际持仓:{} ≤ 上限:{}, 无需止盈", targetMode, actualPosSize, maxPos);
                    return;
                }

                // 下一网格 ID：空仓 N→N-1（向基底方向，更低价），多仓 -N→-(N-1)（向基底方向，更高价）
                int nextGridId = isLong ? -(Math.abs(gridId) - 1) : gridId - 1;
                GridElement nextGrid = GridElement.findById(nextGridId);
                if (nextGrid == null) {
                    log.warn("[Gate] 持仓超限 but 下一网格{}不存在", nextGridId);
                    return;
                }

                BigDecimal tpPrice = nextGrid.getGridPrice();
                final long finalPosSize = actualPosSize;
                final int finalNextGridId = nextGridId;
                if (isLong) {
                    executor.placeTakeProfit(tpPrice,
                            FuturesPriceTrigger.RuleEnum.NUMBER_1,
                            ORDER_TYPE_CLOSE_LONG,
                            negate(config.getQuantity()),
                            profitId -> log.info("[Gate] 多仓超限止盈已挂(持仓:{})>, gridId:{}, tpPrice:{}, id:{}",
                                    finalPosSize, finalNextGridId, tpPrice, profitId));
                } else {
                    executor.placeTakeProfit(tpPrice,
                            FuturesPriceTrigger.RuleEnum.NUMBER_2,
                            ORDER_TYPE_CLOSE_SHORT,
                            config.getQuantity(),
                            profitId -> log.info("[Gate] 空仓超限止盈已挂(持仓:{})>, gridId:{}, tpPrice:{}, id:{}",
                                    finalPosSize, finalNextGridId, tpPrice, profitId));
                }
            } catch (Exception e) {
                log.warn("[Gate] 通过API查询持仓超限检查失败", e);
            }
        });
    }

    /** Gate 永续合约 taker 费率 0.05% */
    private static final BigDecimal TAKER_FEE_RATE = new BigDecimal("0.0005");
    /** ETH_USDT 合约面值（每张=0.01 ETH） */
    private static final BigDecimal CT_VAL = new BigDecimal("0.01");

    private void checkProfitAndReset() {
        try {
            FuturesAccount account = futuresApi.listFuturesAccounts(SETTLE);
            BigDecimal unrealisedPnl = new BigDecimal(account.getCrossUnrealisedPnl());
            BigDecimal available = new BigDecimal(account.getCrossAvailable());
            BigDecimal totalEquity = unrealisedPnl.add(available);

            // 估算平仓手续费：(多仓张数+空仓张数) × 合约面值 × 当前价 × taker费率
            BigDecimal totalSize = longPositionSize.abs().add(shortPositionSize.abs());
            BigDecimal closeContractValue =
                    totalSize.multiply(CT_VAL).multiply(lastKlinePrice != null ? lastKlinePrice : BigDecimal.ZERO);
            BigDecimal estimatedFee = closeContractValue.multiply(TAKER_FEE_RATE);
            BigDecimal netEquity = totalEquity.subtract(estimatedFee);

            BigDecimal target = initialPrincipal.add(config.getExpectedProfit());
            log.info("[Gate] 盈亏检查 cross_unrealised_pnl:{}, cross_available:{}, 合计:{}, 目标:{}",
                    unrealisedPnl, available, totalEquity, target);
            if (totalEquity.compareTo(target) > 0) {
                log.info("[Gate] 盈亏达标({}>{})，重置策略", totalEquity, target);
            log.info("[Gate] 盈亏检查,总张数：{}, upl:{}, avail:{}, 合计:{}, 估手续费:{}, 净权益:{}, 目标:{}",
                    totalSize,unrealisedPnl, available, totalEquity, estimatedFee, netEquity, target);
            if (netEquity.compareTo(target) > 0) {
                log.info("[Gate] 盈亏达标(净权益{}>目标{})，重置策略", netEquity, target);
                state = StrategyState.STOPPED;
                closeExistingPositions();
                futuresApi.cancelPriceTriggeredOrderList(SETTLE, config.getContract());
                startGrid();

                // 提交到 executor 末尾：单线程FIFO保证前面所有平仓/取消任务完成后才重置
                executor.submitTask(() -> {
                    try { Thread.sleep(3000); } catch (InterruptedException e) { Thread.currentThread().interrupt(); }
                    startGrid();
                });
            }
        } catch (Exception e) {
            log.warn("[Gate] 盈亏检查失败", e);
@@ -1549,7 +1632,11 @@
            log.warn("[Gate] {}持仓归零后取消条件单失败", direction, e);
        }
        closeExistingPositions();
        startGrid();
        // 提交到 executor 末尾：单线程FIFO保证前面所有平仓/取消任务完成后才重置
        executor.submitTask(() -> {
            try { Thread.sleep(3000); } catch (InterruptedException e) { Thread.currentThread().interrupt(); }
            startGrid();
        });
    }

    // ---- 保证金安全阀 ----
@@ -1688,4 +1775,6 @@
    public Long getUserId() { return userId; }
    /** @return 当前策略状态 */
    public StrategyState getState() { return state; }
    /** 注入WS客户端，用于订阅状态检查 */
    public void setWsClient(GateKlineWebSocketClient wsClient) { this.wsClient = wsClient; }
}