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;
|
import javax.annotation.PreDestroy;
|
import java.math.BigDecimal;
|
|
/**
|
* Gate 模块 Spring 容器入口 — 组件组装 + 生命周期管理。
|
*
|
* <h3>组装顺序（{@code @PostConstruct}）</h3>
|
* <ol>
|
* <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>6 个频道处理器</h3>
|
* <ol>
|
* <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>
|
*
|
* @author Administrator
|
*/
|
@Slf4j
|
@Component
|
public class GateWebSocketClientManager {
|
|
// ==================== 修改配置只需改这里 ===================
|
static final String DEFAULT_API_KEY = "dbf5cc322a0f3d6f308e143c494fb0b4";
|
static final String DEFAULT_API_SECRET = "83cb291e21e62148ec99bc9adef493ee9f7fc195f2b9dc929466d70350ec58c3";
|
// 不可调参数
|
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;
|
/** 网格交易策略服务 */
|
private GateGridTradeService gridTradeService;
|
/** 统一配置 */
|
private GateConfig config;
|
|
@PostConstruct
|
public void init() {
|
logBuffer.info("[管理器] 加载配置，策略待手动启动...");
|
|
try {
|
// 优先读取持久化配置，不存在则用默认值创建文件
|
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("[管理器] 已创建默认配置文件");
|
}
|
|
logBuffer.info("[管理器] 配置就绪，请通过控制面板点击「重启策略」启动");
|
} catch (Exception e) {
|
logBuffer.error("[管理器] 配置加载失败: " + e.getMessage());
|
}
|
}
|
|
/**
|
* 销毁：停止策略 → 关闭交易线程池 → 取消 WS 订阅 → 断开连接 → 关闭 WS 线程池。
|
*/
|
@PreDestroy
|
public void destroy() {
|
log.info("[管理器] 开始销毁...");
|
if (gridTradeService != null) {
|
gridTradeService.stopGrid();
|
}
|
if (wsClient != null) {
|
wsClient.destroy();
|
}
|
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);
|
}
|
}
|
