feat(gateApi): 添加用户ID获取和WebSocket私有频道订阅功能

Administrator

4 days ago 6d213ed49218afbf8b83e440eba41f642a4695f2

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

@@ -1,24 +1,42 @@
package com.xcong.excoin.modules.gateApi;

import com.xcong.excoin.modules.okxNewPrice.OkxKlineWebSocketClient;
import com.xcong.excoin.modules.okxNewPrice.OkxQuantWebSocketClient;
import com.xcong.excoin.modules.okxNewPrice.celue.CaoZuoService;
import com.xcong.excoin.modules.okxNewPrice.okxWs.enums.ExchangeInfoEnum;
import com.xcong.excoin.modules.okxNewPrice.okxWs.wanggeList.WangGeListService;
import com.xcong.excoin.utils.RedisUtils;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty;
import org.springframework.stereotype.Component;

import javax.annotation.PostConstruct;
import javax.annotation.PreDestroy;
import java.util.Collection;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
import java.math.BigDecimal;

/**
 * 管理多个OKX WebSocket客户端实例，每个实例对应一个账号
 * Gate 模块入口管理类，作为 Spring Bean 负责组件的生命周期。
 *
 * <h3>启动流程</h3>
 * <pre>
 *   @PostConstruct init():
 *     1. new GateGridTradeService(参数) → init()     // 设杠杆、查余额、切双向持仓
 *     2. new GateKlineWebSocketClient → init()        // 连接 WebSocket，订阅 K 线 + 仓位
 *     3. gridTradeService.startGrid()                 // 激活策略，等待 K 线触发首次双开
 * </pre>
 *
 * <h3>销毁流程</h3>
 * <pre>
 *   @PreDestroy destroy():
 *     1. gridTradeService.stopGrid()                  // 停止策略
 *     2. klinePriceClient.destroy()                   // 取消订阅 → 关闭 WS → 关闭线程池
 * </pre>
 *
 * <h3>配置参数</h3>
 * 当前配置在代码中硬编码：
 * <ul>
 *   <li>合约: XAU_USDT, 杠杆: 30x, 全仓, 双向持仓</li>
 *   <li>网格间距: 0.35%, 整体止盈: 0.5 USDT, 最大亏损: 7.5 USDT</li>
 *   <li>数量: 10 张</li>
 * </ul>
 *
 * @author Administrator
 */
@Slf4j
@Component
@@ -26,58 +44,77 @@
    @Autowired
    private CaoZuoService caoZuoService;
    @Autowired
    private RedisUtils redisUtils;
    @Autowired
    private WangGeListService wangGeListService;

    /** K 线 WebSocket 客户端 */
    private GateKlineWebSocketClient klinePriceClient;
    /** 网格交易服务 */
    private GateGridTradeService gridTradeService;

    /** Gate 测试网 API Key */
    private static final String API_KEY = "d90ca272391992b8e74f8f92cedb21ec";
    /** Gate 测试网 API Secret */
    private static final String API_SECRET = "1861e4f52de4bb53369ea3208d9ede38ece4777368030f96c77d27934c46c274";

    /**
     * 初始化方法，在Spring Bean构造完成后执行
     * 创建并初始化所有账号的WebSocket客户端实例
     * Spring 容器启动后自动调用。初始化网格交易服务和 WebSocket 客户端。
     */
    @PostConstruct
    public void init() {
        log.info("开始初始化OkxWebSocketClientManager");
        
        // 初始化价格WebSocket客户端
        log.info("开始初始化GateWebSocketClientManager");

        try {
            klinePriceClient = new GateKlineWebSocketClient(caoZuoService, this, wangGeListService);
            gridTradeService = new GateGridTradeService(
                    API_KEY, API_SECRET,
                    "XAUT_USDT",
                    "30",
                    "cross",
                    "dual",
                    new BigDecimal("0.0035"),
                    new BigDecimal("0.5"),
                    3,
                    new BigDecimal("7.5"),
                    "10"
            );
            gridTradeService.init();

            klinePriceClient = new GateKlineWebSocketClient(caoZuoService, this, wangGeListService, gridTradeService, API_KEY, API_SECRET);
            klinePriceClient.init();
            log.info("已初始化OkxNewPriceWebSocketClient");
            log.info("已初始化GateKlineWebSocketClient");

            gridTradeService.startGrid();
        } catch (Exception e) {
            log.error("初始化OkxNewPriceWebSocketClient失败", e);
            log.error("初始化GateWebSocketClientManager失败", e);
        }
        
    }

    /**
     * 销毁方法，在Spring Bean销毁前执行
     * 关闭所有WebSocket客户端连接和相关资源
     * Spring 容器销毁前自动调用。停止策略并关闭所有连接。
     */
    @PreDestroy
    public void destroy() {
        log.info("开始销毁OkxWebSocketClientManager");
        
        // 关闭价格WebSocket客户端
        log.info("开始销毁GateWebSocketClientManager");

        if (gridTradeService != null) {
            gridTradeService.stopGrid();
        }
        if (klinePriceClient != null) {
            try {
                klinePriceClient.destroy();
                log.info("已销毁OkxNewPriceWebSocketClient");
                log.info("已销毁GateKlineWebSocketClient");
            } catch (Exception e) {
                log.error("销毁OkxNewPriceWebSocketClient失败", e);
                log.error("销毁GateKlineWebSocketClient失败", e);
            }
        }
        
        log.info("OkxWebSocketClientManager销毁完成");

        log.info("GateWebSocketClientManager销毁完成");
    }

    /**
     * 获取OkxNewPriceWebSocketClient实例
     * @return 价格WebSocket客户端实例
     */
    public GateKlineWebSocketClient getKlineWebSocketClient() {
        return klinePriceClient;
    }
}

    public GateGridTradeService getGridTradeService() {
        return gridTradeService;
    }
}