| | |
| | | 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; |
| | |
| | | @Component |
| | | public class GateWebSocketClientManager { |
| | | |
| | | // ==================== 修改配置只需改这里 =================== |
| | | static final String DEFAULT_API_KEY = "b698ffb0cabdb575d04f42539b4b5acf"; |
| | | static final String DEFAULT_API_SECRET = "6bd6d436abd5c05ebf382bf4494cf6f9444ec1b074eadbba4ee239aaffb0fa09"; |
| | | // 不可调参数 |
| | | 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; |
| | | /** 网格交易策略服务 */ |
| | |
| | | |
| | | @PostConstruct |
| | | public void init() { |
| | | log.info("[管理器] 开始初始化..."); |
| | | logBuffer.info("[管理器] 加载配置,策略待手动启动..."); |
| | | |
| | | try { |
| | | //实盘 |
| | | config = GateConfig.builder() |
| | | .apiKey("b698ffb0cabdb575d04f42539b4b5acf") |
| | | .apiSecret("6bd6d436abd5c05ebf382bf4494cf6f9444ec1b074eadbba4ee239aaffb0fa09") |
| | | .contract("ETH_USDT") |
| | | .leverage("100") |
| | | .marginMode("CROSS") |
| | | .positionMode("dual") |
| | | .gridRate(new BigDecimal("0.0035")) |
| | | .expectedProfit(new BigDecimal("2.5")) |
| | | .maxLoss(new BigDecimal("1.5")) |
| | | .baseQuantity("75") |
| | | .quantity("25") |
| | | .restartGridSpan(8) |
| | | .maxPositionSize(2) |
| | | .priceScale(2) |
| | | .contractMultiplier(new BigDecimal("0.01")) |
| | | .unrealizedPnlPriceMode(GateConfig.PnLPriceMode.LAST_PRICE) |
| | | .isProduction(true) |
| | | .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("[管理器] 已创建默认配置文件"); |
| | | } |
| | | |
| | | // 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(); |
| | | logBuffer.info("[管理器] 配置就绪,请通过控制面板点击「重启策略」启动"); |
| | | } catch (Exception e) { |
| | | log.error("[管理器] 初始化失败", e); |
| | | logBuffer.error("[管理器] 配置加载失败: " + e.getMessage()); |
| | | } |
| | | } |
| | | |
| | |
| | | * @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); |
| | | } |
| | | } |