| | |
| | | package com.xcong.excoin.modules.gateApi; |
| | | |
| | | import com.xcong.excoin.modules.gateApi.wsHandler.handler.AutoOrdersChannelHandler; |
| | | import com.xcong.excoin.modules.gateApi.wsHandler.handler.CandlestickChannelHandler; |
| | | import com.xcong.excoin.modules.gateApi.wsHandler.handler.PositionClosesChannelHandler; |
| | | import com.xcong.excoin.modules.gateApi.wsHandler.handler.PositionsChannelHandler; |
| | |
| | | import javax.annotation.PostConstruct; |
| | | import javax.annotation.PreDestroy; |
| | | import java.math.BigDecimal; |
| | | import java.util.ArrayList; |
| | | import java.util.List; |
| | | |
| | | /** |
| | | * Gate 模块 Spring 入口,管理多个账号实例的生命周期。 |
| | | * Gate 模块 Spring 容器入口 — 组件组装 + 生命周期管理。 |
| | | * |
| | | * <h3>启动流程 ({@code @PostConstruct})</h3> |
| | | * <h3>组装顺序({@code @PostConstruct})</h3> |
| | | * <ol> |
| | | * <li>遍历 {@link #buildAccountConfigs()} 构建每个账号的 {@link GateConfig}</li> |
| | | * <li>每个账号创建独立的 {@link AccountInstance}(含 service + wsClient)</li> |
| | | * <li>逐个 init → startGrid</li> |
| | | * <li>{@link GateConfig} — 构建配置(API 密钥、合约、策略参数)</li> |
| | | * <li>{@link GateGridTradeService} — init():获取用户 ID → 切双向持仓 → 清旧条件单 → 平仓 → 设杠杆</li> |
| | | * <li>{@link GateKlineWebSocketClient} — 注册 6 个频道处理器 → init():建立 WS 连接并订阅</li> |
| | | * <li>{@code gridTradeService.startGrid()} — 状态重置,等待首根 K 线</li> |
| | | * </ol> |
| | | * |
| | | * <h3>销毁流程 ({@code @PreDestroy})</h3> |
| | | * <h3>6 个频道处理器</h3> |
| | | * <ol> |
| | | * <li>遍历 instances:stopGrid + wsClient.destroy()</li> |
| | | * <li>CandlestickChannelHandler — 公开频道,K线 → onKline()</li> |
| | | * <li>PositionsChannelHandler — 私有频道,仓位 → onPositionUpdate()</li> |
| | | * <li>PositionClosesChannelHandler — 私有频道,平仓 → onPositionClose()</li> |
| | | * <li>OrdersChannelHandler — 私有频道,订单成交 → onOrderUpdate()</li> |
| | | * <li>UserTradesChannelHandler — 私有频道,用户成交 → onUserTrade()</li> |
| | | * <li>AutoOrdersChannelHandler — 私有频道,条件单状态 → onAutoOrder()</li> |
| | | * </ol> |
| | | * |
| | | * <h3>添加新账号</h3> |
| | | * 在 {@link #buildAccountConfigs()} 方法中添加新的 {@link GateConfig} 即可。 |
| | | * 通过 {@link GateConfig#getLabel()} 区分日志输出, |
| | | * 通过 {@link GateConfig#isProduction()} 控制模拟盘/实盘环境。 |
| | | * <h3>销毁顺序({@code @PreDestroy})</h3> |
| | | * <ol> |
| | | * <li>gridTradeService.stopGrid():取消所有条件单 → 关闭交易线程池</li> |
| | | * <li>wsClient.destroy():取消订阅 → 断开 WS → 关闭线程池</li> |
| | | * </ol> |
| | | * |
| | | * @author Administrator |
| | | */ |
| | |
| | | @Component |
| | | public class GateWebSocketClientManager { |
| | | |
| | | /** 所有账号实例列表 */ |
| | | private final List<AccountInstance> instances = new ArrayList<>(); |
| | | |
| | | /** |
| | | * 封装一个账号的完整运行时组件。 |
| | | */ |
| | | private static class AccountInstance { |
| | | final String label; |
| | | final GateConfig config; |
| | | final GateGridTradeService service; |
| | | final GateKlineWebSocketClient wsClient; |
| | | |
| | | AccountInstance(GateConfig config) { |
| | | this.label = config.getLabel(); |
| | | this.config = config; |
| | | this.service = new GateGridTradeService(config); |
| | | this.wsClient = new GateKlineWebSocketClient(config.getWsUrl()); |
| | | } |
| | | |
| | | void init() { |
| | | service.init(); |
| | | wsClient.addChannelHandler(new CandlestickChannelHandler(config.getContract(), service)); |
| | | wsClient.addChannelHandler(new PositionsChannelHandler( |
| | | config.getApiKey(), config.getApiSecret(), config.getContract(), service)); |
| | | wsClient.addChannelHandler(new PositionClosesChannelHandler( |
| | | config.getApiKey(), config.getApiSecret(), config.getContract(), service)); |
| | | wsClient.init(); |
| | | log.info("[{}] WS已连接, 已注册 3 个频道处理器", label); |
| | | service.startGrid(); |
| | | } |
| | | |
| | | void destroy() { |
| | | log.info("[{}] 开始销毁...", label); |
| | | service.stopGrid(); |
| | | wsClient.destroy(); |
| | | log.info("[{}] 销毁完成", label); |
| | | } |
| | | } |
| | | |
| | | /** |
| | | * 在此方法中配置所有账号。 |
| | | * <p>每个 {@link GateConfig} 对应一个账号,label 用于日志区分, |
| | | * isProduction 控制连接测试网(true=实盘)还是测试网(false=模拟盘)。 |
| | | */ |
| | | private static GateConfig[] buildAccountConfigs() { |
| | | return new GateConfig[]{ |
| | | GateConfig.builder() |
| | | .label("模拟盘1") |
| | | .apiKey("d90ca272391992b8e74f8f92cedb21ec") |
| | | .apiSecret("1861e4f52de4bb53369ea3208d9ede38ece4777368030f96c77d27934c46c274") |
| | | .contract("ETH_USDT") |
| | | .leverage("100") |
| | | .marginMode("CROSS") |
| | | .positionMode("dual") |
| | | .gridRate(new BigDecimal("0.0015")) |
| | | .overallTp(new BigDecimal("5")) |
| | | .maxLoss(new BigDecimal("15")) |
| | | .quantity("1") |
| | | .contractMultiplier(new BigDecimal("0.01")) |
| | | .unrealizedPnlPriceMode(GateConfig.PnLPriceMode.LAST_PRICE) |
| | | .isProduction(false) |
| | | .reopenMaxRetries(3) |
| | | .build(), |
| | | }; |
| | | } |
| | | /** WebSocket 连接管理器 */ |
| | | private GateKlineWebSocketClient wsClient; |
| | | /** 网格交易策略服务 */ |
| | | private GateGridTradeService gridTradeService; |
| | | /** 统一配置 */ |
| | | private GateConfig config; |
| | | |
| | | @PostConstruct |
| | | public void init() { |
| | | GateConfig[] configs = buildAccountConfigs(); |
| | | log.info("[管理器] 开始初始化 {} 个账号...", configs.length); |
| | | for (GateConfig config : configs) { |
| | | try { |
| | | AccountInstance instance = new AccountInstance(config); |
| | | instance.init(); |
| | | instances.add(instance); |
| | | } catch (Exception e) { |
| | | log.error("[管理器] 账号 {} 初始化失败", config.getLabel(), e); |
| | | } |
| | | } |
| | | log.info("[管理器] 开始初始化..."); |
| | | |
| | | log.info("[管理器] 初始化完成, 成功启动 {}/{} 个账号", instances.size(), configs.length); |
| | | try { |
| | | //实盘 |
| | | config = GateConfig.builder() |
| | | .apiKey("08d6108fc29378fe64d6f0e89745b3fb") |
| | | .apiSecret("4d16e40117477a2344f6fd5872c5b075dc9664a599e65422d3e6791ff64379f1") |
| | | .contract("ETH_USDT") |
| | | .leverage("100") |
| | | .marginMode("CROSS") |
| | | .positionMode("dual") |
| | | .gridRate(new BigDecimal("0.002")) |
| | | .expectedProfit(new BigDecimal("2.5")) |
| | | .maxLoss(new BigDecimal("1.5")) |
| | | .baseQuantity("1") |
| | | .quantity("1") |
| | | .maxPositionSize(2) |
| | | .priceScale(2) |
| | | .contractMultiplier(new BigDecimal("0.01")) |
| | | .unrealizedPnlPriceMode(GateConfig.PnLPriceMode.LAST_PRICE) |
| | | .isProduction(true) |
| | | .reopenMaxRetries(3) |
| | | .build(); |
| | | |
| | | // 1. 初始化交易服务:查用户ID → 切持仓模式 → 清条件单 → 平已有仓位 → 设杠杆 |
| | | gridTradeService = new GateGridTradeService(config); |
| | | gridTradeService.init(); |
| | | |
| | | // 2. 创建 WS 客户端并注册频道处理器 |
| | | wsClient = new GateKlineWebSocketClient(config.getWsUrl()); |
| | | wsClient.addChannelHandler(new CandlestickChannelHandler(config.getContract(), gridTradeService)); |
| | | wsClient.addChannelHandler(new PositionsChannelHandler( |
| | | config.getApiKey(), config.getApiSecret(), config.getContract(), gridTradeService)); |
| | | wsClient.addChannelHandler(new PositionClosesChannelHandler( |
| | | config.getApiKey(), config.getApiSecret(), config.getContract(), gridTradeService)); |
| | | wsClient.addChannelHandler(new AutoOrdersChannelHandler( |
| | | config.getApiKey(), config.getApiSecret(), config.getContract(), gridTradeService)); |
| | | gridTradeService.setWsClient(wsClient); |
| | | wsClient.init(); |
| | | log.info("[管理器] WS已连接, 已注册 4 个频道处理器"); |
| | | |
| | | // 3. 激活策略,等待首根 K 线触发基底双开 |
| | | gridTradeService.startGrid(); |
| | | } catch (Exception e) { |
| | | log.error("[管理器] 初始化失败", e); |
| | | } |
| | | } |
| | | |
| | | /** |
| | | * 销毁:停止策略 → 关闭交易线程池 → 取消 WS 订阅 → 断开连接 → 关闭 WS 线程池。 |
| | | */ |
| | | @PreDestroy |
| | | public void destroy() { |
| | | log.info("[管理器] 开始销毁 {} 个账号...", instances.size()); |
| | | for (AccountInstance instance : instances) { |
| | | try { |
| | | instance.destroy(); |
| | | } catch (Exception e) { |
| | | log.error("[管理器] 账号 {} 销毁失败", instance.label, e); |
| | | } |
| | | log.info("[管理器] 开始销毁..."); |
| | | if (gridTradeService != null) { |
| | | gridTradeService.stopGrid(); |
| | | } |
| | | if (wsClient != null) { |
| | | wsClient.destroy(); |
| | | } |
| | | log.info("[管理器] 销毁完成"); |
| | | } |
| | | |
| | | /** 获取账号数量 */ |
| | | public int getInstanceCount() { |
| | | return instances.size(); |
| | | } |
| | | |
| | | /** 获取指定索引的 GridTradeService(用于外部监控/查询) */ |
| | | public GateGridTradeService getGridTradeService(int index) { |
| | | return instances.get(index).service; |
| | | } |
| | | /** |
| | | * @return WebSocket 连接管理器实例 |
| | | */ |
| | | public GateKlineWebSocketClient getKlineWebSocketClient() { return wsClient; } |
| | | /** |
| | | * @return 网格交易策略服务实例 |
| | | */ |
| | | public GateGridTradeService getGridTradeService() { return gridTradeService; } |
| | | } |