From 8e232144615734d62e6dd0293956aca8bcc44e40 Mon Sep 17 00:00:00 2001
From: Administrator <15274802129@163.com>
Date: Mon, 24 Aug 2026 14:13:39 +0800
Subject: [PATCH] fix(gate): 修复心跳调度器中服务器端口获取问题
---
src/main/java/com/xcong/excoin/modules/gateApi/HeartbeatScheduler.java | 96 ++++++++++++++++++++++++++++++++++++++++++++++++
1 files changed, 96 insertions(+), 0 deletions(-)
diff --git a/src/main/java/com/xcong/excoin/modules/gateApi/HeartbeatScheduler.java b/src/main/java/com/xcong/excoin/modules/gateApi/HeartbeatScheduler.java
new file mode 100644
index 0000000..7423b29
--- /dev/null
+++ b/src/main/java/com/xcong/excoin/modules/gateApi/HeartbeatScheduler.java
@@ -0,0 +1,96 @@
+package com.xcong.excoin.modules.gateApi;
+
+import com.xcong.excoin.modules.station.model.GateStatsEvent;
+import com.xcong.excoin.modules.station.model.HeartbeatMsg;
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.core.env.Environment;
+import org.springframework.scheduling.annotation.EnableScheduling;
+import org.springframework.scheduling.annotation.Scheduled;
+import org.springframework.stereotype.Component;
+
+import javax.annotation.Resource;
+import java.net.InetAddress;
+import java.security.MessageDigest;
+
+/**
+ * JAR 侧 — 心跳定时发送（每 30s）
+ */
+@Slf4j
+@Component
+@EnableScheduling
+public class HeartbeatScheduler {
+
+ @Resource
+ private StatsEventProducer statsProducer;
+
+ @Resource
+ private GateWebSocketClientManager manager;
+
+ @Resource
+ private Environment environment;
+
+ private String apiKeyMd5;
+ private String hostPort;
+
+ // initialDelay = 0：启动后立即发第一次心跳，避免 Station 要等 30s 才能发现实例
+ @Scheduled(fixedRate = 30_000, initialDelay = 1_000)
+ public void heartbeat() {
+ try {
+ if (apiKeyMd5 == null) {
+ apiKeyMd5 = md5(manager.getConfig().getApiKey());
+ hostPort = resolveHostPort();
+ }
+
+ // 策略未启动（gridTradeService == null）时也发送心跳，
+ // 否则 Station 无法发现该 JAR 实例，也就无法对它下发 START 指令（死锁）。
+ GateGridTradeService strategy = manager.getGridTradeService();
+ String state = (strategy != null && strategy.getState() != null)
+ ? strategy.getState().name() : "STOPPED";
+ int currentRound = strategy != null ? strategy.getCurrentRound() : 0;
+ String cumulativePnl = (strategy != null && strategy.getCumulativePnl() != null)
+ ? strategy.getCumulativePnl().toPlainString() : "0";
+ String principal = (strategy != null && strategy.getInitialPrincipal() != null)
+ ? strategy.getInitialPrincipal().toPlainString() : "0";
+
+ HeartbeatMsg hb = HeartbeatMsg.builder()
+ .contract(manager.getConfig().getContract())
+ .state(state)
+ .leverage(manager.getConfig().getLeverage())
+ .currentRound(currentRound)
+ .cumulativePnl(cumulativePnl)
+ .principal(principal)
+ .hostPort(hostPort)
+ .build();
+
+ GateStatsEvent event = statsProducer.newHeartbeat(apiKeyMd5, hb);
+ statsProducer.sendHeartbeat(event);
+
+ log.info("[Gate] 心跳已发送, state={}, apiKeyMd5={}", state, apiKeyMd5);
+ } catch (Exception e) {
+ log.warn("[Gate] 心跳发送失败", e);
+ }
+ }
+
+ private String resolveHostPort() {
+ try {
+ String host = InetAddress.getLocalHost().getHostAddress();
+ // port 从 Spring Environment 获取（兼容 --server.port=xxx 命令行参数），默认 8888
+ String port = environment.getProperty("server.port", "8888");
+ return host + ":" + port;
+ } catch (Exception e) {
+ return "unknown";
+ }
+ }
+
+ private static String md5(String input) {
+ try {
+ MessageDigest md = MessageDigest.getInstance("MD5");
+ byte[] digest = md.digest(input.getBytes(java.nio.charset.StandardCharsets.UTF_8));
+ StringBuilder sb = new StringBuilder();
+ for (byte b : digest) sb.append(String.format("%02x", b));
+ return sb.toString();
+ } catch (Exception e) {
+ return Integer.toHexString(input.hashCode());
+ }
+ }
+}
--
Gitblit v1.9.1