From 9ff37e00205c776f62a9b07e1f9d8736a14d4d8c Mon Sep 17 00:00:00 2001
From: Administrator <15274802129@163.com>
Date: Wed, 08 Jul 2026 11:50:41 +0800
Subject: [PATCH] feat(gate): 添加 Gate 策略控制面板和配置管理功能

---
 src/main/java/com/xcong/excoin/modules/gateApi/GateWebSocketClientManager.java |  173 +++++++++++++++++++++++++++++++++++++++++++--------------
 1 files changed, 131 insertions(+), 42 deletions(-)

diff --git a/src/main/java/com/xcong/excoin/modules/gateApi/GateWebSocketClientManager.java b/src/main/java/com/xcong/excoin/modules/gateApi/GateWebSocketClientManager.java
index 58deb72..f68a09e 100644
--- a/src/main/java/com/xcong/excoin/modules/gateApi/GateWebSocketClientManager.java
+++ b/src/main/java/com/xcong/excoin/modules/gateApi/GateWebSocketClientManager.java
@@ -1,9 +1,11 @@
 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 lombok.extern.slf4j.Slf4j;
+import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Component;
 
 import javax.annotation.PostConstruct;
@@ -11,30 +13,58 @@
 import java.math.BigDecimal;
 
 /**
- * Gate 模块 Spring 入口,组装所有组件并管理生命周期。
+ * Gate 模块 Spring 容器入口 — 组件组装 + 生命周期管理。
  *
- * <h3>启动流程 ({@code @PostConstruct})</h3>
+ * <h3>组装顺序({@code @PostConstruct})</h3>
  * <ol>
- *   <li>构建 {@link GateConfig}(Builder 模式,含 API 密钥、合约、策略参数)</li>
- *   <li>创建 {@link GateGridTradeService} → init():切持仓模式、清旧条件单、设杠杆</li>
- *   <li>创建 {@link GateKlineWebSocketClient} → 注册 3 个 Handler → init():建立 WS 连接</li>
- *   <li>gridTradeService.startGrid():激活策略,等待 K 线触发首次双开</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>gridTradeService.stopGrid():取消条件单 → 关闭交易线程池</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>销毁顺序({@code @PreDestroy})</h3>
+ * <ol>
+ *   <li>gridTradeService.stopGrid():取消所有条件单 → 关闭交易线程池</li>
  *   <li>wsClient.destroy():取消订阅 → 断开 WS → 关闭线程池</li>
  * </ol>
- *
- * <h3>配置</h3>
- * 当前在代码中硬编码测试网参数。切换到生产网只需改为 {@code .isProduction(true)}。
  *
  * @author Administrator
  */
 @Slf4j
 @Component
 public class GateWebSocketClientManager {
+
+    // ==================== 修改配置只需改这里 ===================
+    static final String DEFAULT_API_KEY = "08d6108fc29378fe64d6f0e89745b3fb";
+    static final String DEFAULT_API_SECRET = "4d16e40117477a2344f6fd5872c5b075dc9664a599e65422d3e6791ff64379f1";
+    // 不可调参数
+    static final String DEFAULT_CONTRACT = "ETH_USDT";
+    static final String DEFAULT_LEVERAGE = "100";
+    static final String DEFAULT_MARGIN_MODE = "CROSS";
+    static final String DEFAULT_POSITION_MODE = "dual";
+    static final int DEFAULT_RESTART_GRID_SPAN = 2;
+    static final int DEFAULT_PRICE_SCALE = 2;
+    static final String DEFAULT_CONTRACT_MULTIPLIER = "0.01";
+    static final String DEFAULT_PNL_MODE = "LAST_PRICE";
+    static final int DEFAULT_REOPEN_MAX_RETRIES = 3;
+    // ==========================================================
+
+    @Autowired
+    private GateConfigPersistenceService persistenceService;
+
+    @Autowired
+    private GateLogBuffer logBuffer;
 
     /** WebSocket 连接管理器 */
     private GateKlineWebSocketClient wsClient;
@@ -45,44 +75,41 @@
 
     @PostConstruct
     public void init() {
-        log.info("[管理器] 开始初始化...");
+        logBuffer.info("[管理器] 加载配置,策略待手动启动...");
 
         try {
-            config = GateConfig.builder()
-                    .apiKey("d90ca272391992b8e74f8f92cedb21ec")
-                    .apiSecret("1861e4f52de4bb53369ea3208d9ede38ece4777368030f96c77d27934c46c274")
-                    .contract("ETH_USDT")
-                    .leverage("100")
-                    .marginMode("cross")
-                    .positionMode("dual")
-                    .gridRate(new BigDecimal("0.004"))
-                    .overallTp(new BigDecimal("0.5"))
-                    .maxLoss(new BigDecimal("7.5"))
-                    .quantity("1")
-                    .contractMultiplier(new BigDecimal("0.01"))
-                    .unrealizedPnlPriceMode(GateConfig.PnLPriceMode.LAST_PRICE)
-                    .isProduction(false)
-                    .reopenMaxRetries(3)
-                    .build();
+            // 优先读取持久化配置,不存在则用默认值创建文件
+            GateConfigDTO saved = persistenceService.load(DEFAULT_API_KEY);
+            if (saved != null) {
+                config = persistenceService.buildFromDTO(saved, DEFAULT_API_KEY);
+                logBuffer.info("[管理器] 已加载持久化配置, apiKey=" + mask(DEFAULT_API_KEY));
+            } else {
+                // 构建默认配置 DTO(只有 7 项可调参数)
+                GateConfigDTO defaults = GateConfigDTO.builder()
+                        .apiKey(DEFAULT_API_KEY)
+                        .gridRate(new BigDecimal("0.005"))
+                        .expectedProfit(new BigDecimal("0.15"))
+                        .maxLoss(new BigDecimal("1.5"))
+                        .baseQuantity("2")
+                        .quantity("2")
+                        .maxPositionSize(4)
+                        .stopLossCount(0)
+                        .build();
+                // 确保配置文件存在
+                persistenceService.ensureExists(DEFAULT_API_KEY, defaults);
+                config = persistenceService.buildFromDTO(defaults, DEFAULT_API_KEY);
+                logBuffer.info("[管理器] 已创建默认配置文件");
+            }
 
-            gridTradeService = new GateGridTradeService(config);
-            gridTradeService.init();
-
-            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.init();
-            log.info("[管理器] WS已连接, 已注册 3 个频道处理器");
-
-            gridTradeService.startGrid();
+            logBuffer.info("[管理器] 配置就绪,请通过控制面板点击「重启策略」启动");
         } catch (Exception e) {
-            log.error("[管理器] 初始化失败", e);
+            logBuffer.error("[管理器] 配置加载失败: " + e.getMessage());
         }
     }
 
+    /**
+     * 销毁:停止策略 → 关闭交易线程池 → 取消 WS 订阅 → 断开连接 → 关闭 WS 线程池。
+     */
     @PreDestroy
     public void destroy() {
         log.info("[管理器] 开始销毁...");
@@ -95,6 +122,68 @@
         log.info("[管理器] 销毁完成");
     }
 
+    /**
+     * @return WebSocket 连接管理器实例
+     */
     public GateKlineWebSocketClient getKlineWebSocketClient() { return wsClient; }
+    /**
+     * @return 网格交易策略服务实例
+     */
     public GateGridTradeService getGridTradeService() { return gridTradeService; }
+    /**
+     * @return 当前生效的配置
+     */
+    public GateConfig getConfig() { return config; }
+
+    /**
+     * 使用新配置重启策略:停止旧策略 → 重建 WS → 重新 init + startGrid。
+     */
+    public void restartWithConfig(GateConfig newConfig) {
+        logBuffer.info("[管理器] 开始策略重启...");
+        // 1. 停止旧策略 + 断开旧 WS
+        if (gridTradeService != null) {
+            gridTradeService.stopGrid();
+        }
+        if (wsClient != null) {
+            wsClient.destroy();
+        }
+        // 2. 使用新配置重建
+        this.config = newConfig;
+        this.gridTradeService = new GateGridTradeService(config);
+        gridTradeService.init();
+        // 3. 重建 WS 客户端并重新注册 Handler
+        this.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();
+        logBuffer.info("[管理器] WS已重连, Handler 已重新绑定");
+        // 4. 启动策略
+        gridTradeService.startGrid();
+        logBuffer.info("[管理器] 策略已使用新配置重启完成");
+    }
+
+    /**
+     * 停止策略:取消所有条件单 → 停止网格 → 断开 WS。
+     */
+    public void stopStrategy() {
+        logBuffer.info("[管理器] 停止策略...");
+        if (gridTradeService != null) {
+            gridTradeService.stopGrid();
+        }
+        if (wsClient != null) {
+            wsClient.destroy();
+        }
+        logBuffer.info("[管理器] 策略已停止");
+    }
+
+    private static String mask(String key) {
+        if (key == null || key.length() <= 8) return "***";
+        return key.substring(0, 4) + "****" + key.substring(key.length() - 4);
+    }
 }

--
Gitblit v1.9.1