package com.xcong.excoin.modules.station.registry;
|
|
import com.xcong.excoin.modules.station.model.HeartbeatMsg;
|
import com.xcong.excoin.modules.station.model.InstanceInfo;
|
import lombok.extern.slf4j.Slf4j;
|
import org.springframework.scheduling.annotation.Scheduled;
|
import org.springframework.stereotype.Component;
|
|
import java.util.Collection;
|
import java.util.concurrent.ConcurrentHashMap;
|
|
/**
|
* JAR 实例注册表 — 心跳更新，超时自动清理。
|
*/
|
@Slf4j
|
@Component
|
public class InstanceRegistry {
|
|
/** 30s 心跳 × 2 = 60s 超时下线 */
|
private static final long TIMEOUT_MS = 60_000;
|
|
private final ConcurrentHashMap<String, InstanceInfo> registry = new ConcurrentHashMap<>();
|
|
/**
|
* 心跳更新（不存在则新增）
|
*/
|
public void update(String apiKeyMd5, HeartbeatMsg msg) {
|
InstanceInfo info = InstanceInfo.builder()
|
.apiKeyMd5(apiKeyMd5)
|
.contract(msg.getContract())
|
.state(msg.getState())
|
.leverage(msg.getLeverage())
|
.currentRound(msg.getCurrentRound())
|
.cumulativePnl(msg.getCumulativePnl())
|
.principal(msg.getPrincipal())
|
.hostPort(msg.getHostPort())
|
.lastSeen(System.currentTimeMillis())
|
.build();
|
InstanceInfo old = registry.put(apiKeyMd5, info);
|
if (old == null) {
|
log.info("[Station] 实例上线, apiKeyMd5={}, contract={}", apiKeyMd5, msg.getContract());
|
}
|
}
|
|
/**
|
* 更新状态（用于 ACK）
|
*/
|
public void updateState(String apiKeyMd5, String newState) {
|
InstanceInfo info = registry.get(apiKeyMd5);
|
if (info != null) {
|
info.setState(newState);
|
info.setLastSeen(System.currentTimeMillis());
|
}
|
}
|
|
/**
|
* 获取单个实例
|
*/
|
public InstanceInfo get(String apiKeyMd5) {
|
return registry.get(apiKeyMd5);
|
}
|
|
/**
|
* 所有在线实例
|
*/
|
public Collection<InstanceInfo> list() {
|
return registry.values();
|
}
|
|
/**
|
* 每 30s 清理超时实例
|
*/
|
@Scheduled(fixedRate = 30_000)
|
public void cleanDead() {
|
long now = System.currentTimeMillis();
|
registry.entrySet().removeIf(entry -> {
|
boolean dead = now - entry.getValue().getLastSeen() > TIMEOUT_MS;
|
if (dead) {
|
log.info("[Station] 实例离线, apiKeyMd5={}, 最后心跳:{}ms前",
|
entry.getKey(), now - entry.getValue().getLastSeen());
|
}
|
return dead;
|
});
|
}
|
}
|
