From d07e0c1a8961b61557de3812450ddfd227f773de Mon Sep 17 00:00:00 2001
From: Junjie <fallin.jie@qq.com>
Date: 星期六, 21 三月 2026 22:36:19 +0800
Subject: [PATCH] #
---
src/main/java/com/zy/core/trace/StationTaskTraceRegistry.java | 377 +++++++++++++++++++++++++++++++++++++++++++++++++++++
1 files changed, 375 insertions(+), 2 deletions(-)
diff --git a/src/main/java/com/zy/core/trace/StationTaskTraceRegistry.java b/src/main/java/com/zy/core/trace/StationTaskTraceRegistry.java
index 4c8e76b..1751be6 100644
--- a/src/main/java/com/zy/core/trace/StationTaskTraceRegistry.java
+++ b/src/main/java/com/zy/core/trace/StationTaskTraceRegistry.java
@@ -1,10 +1,17 @@
package com.zy.core.trace;
+import com.alibaba.fastjson.JSON;
import com.zy.asrs.domain.vo.StationTaskTraceEventVo;
import com.zy.asrs.domain.vo.StationTaskTraceSegmentVo;
import com.zy.asrs.domain.vo.StationTaskTraceVo;
+import com.zy.asrs.entity.WrkMast;
+import com.zy.asrs.service.WrkMastService;
+import com.zy.common.utils.RedisUtil;
+import com.zy.core.enums.WrkStsType;
import com.zy.core.model.command.StationCommand;
+import com.zy.core.enums.RedisKeyType;
import lombok.Data;
+import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
import java.util.ArrayList;
@@ -20,6 +27,7 @@
private static final int MAX_EVENT_COUNT = 200;
private static final long TERMINAL_KEEP_MS = 30L * 60L * 1000L;
+ private static final String TRACE_CACHE_KEY = RedisKeyType.STATION_TASK_TRACE_REGISTRY.key;
public static final String STATUS_WAITING = "WAITING";
public static final String STATUS_RUNNING = "RUNNING";
@@ -29,7 +37,14 @@
public static final String STATUS_FINISHED = "FINISHED";
public static final String STATUS_REROUTED = "REROUTED";
+ @Autowired
+ private RedisUtil redisUtil;
+
+ @Autowired
+ private WrkMastService wrkMastService;
+
private final Map<Integer, TraceTaskState> taskStateMap = new ConcurrentHashMap<>();
+ private volatile boolean loadedFromRedis = false;
public TraceRegistration registerPlan(Integer taskNo,
String threadImpl,
@@ -42,10 +57,13 @@
return new TraceRegistration();
}
+ ensureCacheLoaded();
cleanupExpired();
TraceTaskState taskState = taskStateMap.computeIfAbsent(taskNo, TraceTaskState::new);
- return taskState.registerPlan(threadImpl, startStationId, currentStationId, finalTargetStationId,
+ TraceRegistration registration = taskState.registerPlan(threadImpl, startStationId, currentStationId, finalTargetStationId,
copyIntegerList(localPathStationIds), copySegmentList(localSegmentList));
+ persistState(taskState);
+ return registration;
}
public void markSegmentIssued(Integer taskNo,
@@ -54,11 +72,13 @@
String eventType,
String message,
Map<String, Object> details) {
+ ensureCacheLoaded();
TraceTaskState state = taskStateMap.get(taskNo);
if (state == null) {
return;
}
state.markSegmentIssued(traceVersion, command, eventType, message, details);
+ persistState(state);
}
public void updateProgress(Integer taskNo,
@@ -67,63 +87,90 @@
String eventType,
String message,
Map<String, Object> details) {
+ ensureCacheLoaded();
TraceTaskState state = taskStateMap.get(taskNo);
if (state == null) {
return;
}
state.updateProgress(traceVersion, currentStationId, eventType, message, details);
+ persistState(state);
}
public void markBlocked(Integer taskNo,
Integer traceVersion,
Integer currentStationId,
Map<String, Object> details) {
+ ensureCacheLoaded();
TraceTaskState state = taskStateMap.get(taskNo);
if (state == null) {
return;
}
state.markTerminal(traceVersion, STATUS_BLOCKED, currentStationId, currentStationId,
"BLOCKED", "杈撻�佷换鍔¤繍琛屽牭濉�", details);
+ persistState(state);
}
public void markCancelled(Integer taskNo,
Integer traceVersion,
Integer currentStationId,
Map<String, Object> details) {
+ ensureCacheLoaded();
TraceTaskState state = taskStateMap.get(taskNo);
if (state == null) {
return;
}
state.markTerminal(traceVersion, STATUS_CANCELLED, currentStationId, null,
"CANCELLED", "杈撻�佷换鍔℃敹鍒板彇娑堜俊鍙�", details);
+ persistState(state);
}
public void markTimeout(Integer taskNo,
Integer traceVersion,
Integer currentStationId,
Map<String, Object> details) {
+ ensureCacheLoaded();
TraceTaskState state = taskStateMap.get(taskNo);
if (state == null) {
return;
}
state.markTerminal(traceVersion, STATUS_TIMEOUT, currentStationId, null,
"TIMEOUT", "杈撻�佷换鍔¢暱鏃堕棿鏃犳硶瀹氫綅褰撳墠浣嶇疆", details);
+ persistState(state);
}
public void markFinished(Integer taskNo,
Integer traceVersion,
Integer currentStationId,
Map<String, Object> details) {
+ ensureCacheLoaded();
TraceTaskState state = taskStateMap.get(taskNo);
if (state == null) {
return;
}
state.markTerminal(traceVersion, STATUS_FINISHED, currentStationId, null,
"FINISHED", "杈撻�佷换鍔¤建杩瑰畬鎴�", details);
+ persistState(state);
+ }
+
+ public void updateLoopHint(Integer taskNo,
+ Boolean loopAlertActive,
+ String loopAlertType,
+ String loopAlertText,
+ Integer loopAlertCount,
+ Map<String, Object> details) {
+ ensureCacheLoaded();
+ TraceTaskState state = taskStateMap.get(taskNo);
+ if (state == null) {
+ return;
+ }
+ state.updateLoopHint(loopAlertActive, loopAlertType, loopAlertText, loopAlertCount, details);
+ persistState(state);
}
public List<StationTaskTraceVo> listLatestTraces() {
+ ensureCacheLoaded();
cleanupExpired();
+ reconcileInactiveBusinessTasks();
List<StationTaskTraceVo> result = new ArrayList<>();
for (TraceTaskState state : taskStateMap.values()) {
if (state != null) {
@@ -141,13 +188,155 @@
return result;
}
+ public List<StationTaskTraceVo> listActiveTraces() {
+ List<StationTaskTraceVo> latestTraces = listLatestTraces();
+ List<StationTaskTraceVo> result = new ArrayList<>();
+ for (StationTaskTraceVo traceVo : latestTraces) {
+ if (traceVo == null) {
+ continue;
+ }
+ String status = traceVo.getStatus();
+ if (STATUS_WAITING.equals(status)
+ || STATUS_RUNNING.equals(status)
+ || STATUS_REROUTED.equals(status)) {
+ result.add(traceVo);
+ }
+ }
+ return result;
+ }
+
private void cleanupExpired() {
long now = System.currentTimeMillis();
for (Map.Entry<Integer, TraceTaskState> entry : taskStateMap.entrySet()) {
TraceTaskState state = entry.getValue();
if (state != null && state.shouldRemove(now)) {
taskStateMap.remove(entry.getKey(), state);
+ removePersistedState(entry.getKey());
}
+ }
+ }
+
+ private void reconcileInactiveBusinessTasks() {
+ if (wrkMastService == null) {
+ return;
+ }
+ for (TraceTaskState state : taskStateMap.values()) {
+ if (state == null || state.isTerminal()) {
+ continue;
+ }
+ WrkMast wrkMast;
+ try {
+ wrkMast = wrkMastService.selectByWorkNo(state.taskNo);
+ } catch (Exception ignore) {
+ continue;
+ }
+ if (wrkMast == null) {
+ Integer currentStationId = state.currentStationId != null ? state.currentStationId : state.finalTargetStationId;
+ Map<String, Object> details = new LinkedHashMap<>();
+ details.put("reason", "wrk_missing");
+ state.markTerminal(state.traceVersion, STATUS_FINISHED, currentStationId, null,
+ "AUTO_FINISHED", "杈撻�佷换鍔℃。宸蹭笉瀛樺湪锛岃建杩硅嚜鍔ㄧ粨鏉�", details);
+ persistState(state);
+ continue;
+ }
+ if (isStationTraceActiveWrkStatus(wrkMast.getWrkSts())) {
+ continue;
+ }
+ Integer currentStationId = state.currentStationId != null ? state.currentStationId : state.finalTargetStationId;
+ Map<String, Object> details = new LinkedHashMap<>();
+ details.put("reason", "wrk_status_transition");
+ details.put("wrkSts", wrkMast.getWrkSts());
+ details.put("wrkStsDesc", wrkMast.getWrkSts$());
+ if (isManualWrkStatus(wrkMast.getWrkSts())) {
+ state.markTerminal(state.traceVersion, STATUS_CANCELLED, currentStationId, null,
+ "AUTO_CANCELLED", "杈撻�佷换鍔″凡閫�鍑鸿繍琛岋紝杞ㄨ抗鑷姩缁撴潫", details);
+ } else {
+ state.markTerminal(state.traceVersion, STATUS_FINISHED, currentStationId, null,
+ "AUTO_FINISHED", "杈撻�佷换鍔″凡缁撴潫锛岃建杩硅嚜鍔ㄧ粨鏉�", details);
+ }
+ persistState(state);
+ }
+ }
+
+ private boolean isStationTraceActiveWrkStatus(Long wrkSts) {
+ return Objects.equals(wrkSts, WrkStsType.INBOUND_DEVICE_RUN.sts)
+ || Objects.equals(wrkSts, WrkStsType.STATION_RUN.sts);
+ }
+
+ private boolean isManualWrkStatus(Long wrkSts) {
+ return Objects.equals(wrkSts, WrkStsType.INBOUND_MANUAL.sts)
+ || Objects.equals(wrkSts, WrkStsType.OUTBOUND_MANUAL.sts)
+ || Objects.equals(wrkSts, WrkStsType.LOC_MOVE_MANUAL.sts);
+ }
+
+ private void ensureCacheLoaded() {
+ if (loadedFromRedis) {
+ return;
+ }
+ synchronized (this) {
+ if (loadedFromRedis) {
+ return;
+ }
+ Map<Object, Object> cacheMap = redisUtil == null ? null : redisUtil.hmget(TRACE_CACHE_KEY);
+ if (cacheMap != null && !cacheMap.isEmpty()) {
+ long now = System.currentTimeMillis();
+ for (Map.Entry<Object, Object> entry : cacheMap.entrySet()) {
+ Integer taskNo = parseTaskNo(entry.getKey());
+ TraceSnapshot snapshot = parseSnapshot(entry.getValue(), taskNo);
+ if (snapshot == null || snapshot.getTaskNo() == null) {
+ if (taskNo != null) {
+ removePersistedState(taskNo);
+ }
+ continue;
+ }
+ if (snapshot.isExpired(now)) {
+ removePersistedState(snapshot.getTaskNo());
+ continue;
+ }
+ taskStateMap.put(snapshot.getTaskNo(), TraceTaskState.fromSnapshot(snapshot));
+ }
+ }
+ loadedFromRedis = true;
+ }
+ }
+
+ private void persistState(TraceTaskState state) {
+ if (state == null || state.taskNo == null || redisUtil == null) {
+ return;
+ }
+ redisUtil.hset(TRACE_CACHE_KEY, String.valueOf(state.taskNo), JSON.toJSONString(state.toSnapshot()));
+ }
+
+ private void removePersistedState(Integer taskNo) {
+ if (taskNo == null || redisUtil == null) {
+ return;
+ }
+ redisUtil.hdel(TRACE_CACHE_KEY, String.valueOf(taskNo));
+ }
+
+ private Integer parseTaskNo(Object value) {
+ if (value == null) {
+ return null;
+ }
+ try {
+ return Integer.parseInt(String.valueOf(value));
+ } catch (Exception e) {
+ return null;
+ }
+ }
+
+ private TraceSnapshot parseSnapshot(Object cacheValue, Integer fallbackTaskNo) {
+ if (cacheValue == null) {
+ return null;
+ }
+ try {
+ TraceSnapshot snapshot = JSON.parseObject(String.valueOf(cacheValue), TraceSnapshot.class);
+ if (snapshot != null && snapshot.getTaskNo() == null) {
+ snapshot.setTaskNo(fallbackTaskNo);
+ }
+ return snapshot;
+ } catch (Exception e) {
+ return null;
}
}
@@ -205,6 +394,29 @@
return result;
}
+ private static List<StationTaskTraceEventVo> copyEventList(List<StationTaskTraceEventVo> source) {
+ List<StationTaskTraceEventVo> result = new ArrayList<>();
+ if (source == null) {
+ return result;
+ }
+ for (StationTaskTraceEventVo item : source) {
+ if (item == null) {
+ continue;
+ }
+ StationTaskTraceEventVo copy = new StationTaskTraceEventVo();
+ copy.setTimestamp(item.getTimestamp());
+ copy.setEventType(item.getEventType());
+ copy.setMessage(item.getMessage());
+ copy.setStatus(item.getStatus());
+ copy.setCurrentStationId(item.getCurrentStationId());
+ copy.setTargetStationId(item.getTargetStationId());
+ copy.setTraceVersion(item.getTraceVersion());
+ copy.setDetails(copyDetails(item.getDetails()));
+ result.add(copy);
+ }
+ return result;
+ }
+
private static boolean isTerminalStatus(String status) {
return STATUS_BLOCKED.equals(status)
|| STATUS_CANCELLED.equals(status)
@@ -218,6 +430,37 @@
private Integer pathOffset;
private List<Integer> fullPathStationIds = new ArrayList<>();
private boolean rerouted;
+ }
+
+ @Data
+ private static class TraceSnapshot {
+ private Integer taskNo;
+ private String threadImpl;
+ private String status;
+ private Integer traceVersion;
+ private Integer startStationId;
+ private Integer currentStationId;
+ private Integer finalTargetStationId;
+ private Integer blockedStationId;
+ private List<Integer> fullPathStationIds = new ArrayList<>();
+ private List<Integer> issuedStationIds = new ArrayList<>();
+ private List<Integer> passedStationIds = new ArrayList<>();
+ private List<Integer> pendingStationIds = new ArrayList<>();
+ private List<Integer> latestIssuedSegmentPath = new ArrayList<>();
+ private List<StationTaskTraceSegmentVo> segmentList = new ArrayList<>();
+ private Integer issuedSegmentCount;
+ private Integer totalSegmentCount;
+ private Boolean loopAlertActive;
+ private String loopAlertType;
+ private String loopAlertText;
+ private Integer loopAlertCount;
+ private Long updatedAt;
+ private Long terminalExpireAt;
+ private List<StationTaskTraceEventVo> events = new ArrayList<>();
+
+ private boolean isExpired(long now) {
+ return terminalExpireAt != null && terminalExpireAt <= now;
+ }
}
private static class TraceTaskState {
@@ -238,6 +481,10 @@
private List<StationTaskTraceSegmentVo> segmentList = new ArrayList<>();
private Integer issuedSegmentCount = 0;
private Integer totalSegmentCount = 0;
+ private Boolean loopAlertActive = Boolean.FALSE;
+ private String loopAlertType;
+ private String loopAlertText;
+ private Integer loopAlertCount;
private final List<StationTaskTraceEventVo> events = new ArrayList<>();
private Long updatedAt = System.currentTimeMillis();
private Long terminalExpireAt;
@@ -253,6 +500,7 @@
List<Integer> localPathStationIds,
List<StationTaskTraceSegmentVo> localSegmentList) {
TraceRegistration registration = new TraceRegistration();
+ boolean firstPlan = this.traceVersion == null || this.traceVersion <= 0;
boolean rerouted = !isTerminalStatus(this.status) && this.traceVersion != null && this.traceVersion > 0;
int nextTraceVersion = rerouted ? this.traceVersion + 1 : 1;
int pathOffset = rerouted ? this.passedStationIds.size() : 0;
@@ -274,6 +522,9 @@
this.latestIssuedSegmentPath = new ArrayList<>();
this.status = rerouted ? STATUS_REROUTED : STATUS_WAITING;
this.terminalExpireAt = null;
+ if (firstPlan) {
+ clearLoopHintState();
+ }
rebuildProgress(planCurrentStationId);
this.updatedAt = System.currentTimeMillis();
@@ -283,6 +534,7 @@
details.put("segmentCount", this.totalSegmentCount);
details.put("pathOffset", pathOffset);
details.put("currentStationId", this.currentStationId);
+ appendLoopHintDetails(details);
appendEvent(rerouted ? "REROUTED" : "PLAN_READY",
rerouted ? "杈撻�佷换鍔¤矾寰勫凡閲嶇畻骞剁画鎺ヨ建杩�" : "杈撻�佷换鍔″垎娈佃鍒掑凡寤虹珛",
details);
@@ -378,8 +630,50 @@
appendEvent(eventType, message, nextDetails);
}
+ private synchronized void updateLoopHint(Boolean loopAlertActive,
+ String loopAlertType,
+ String loopAlertText,
+ Integer loopAlertCount,
+ Map<String, Object> details) {
+ boolean active = Boolean.TRUE.equals(loopAlertActive)
+ && loopAlertCount != null
+ && loopAlertCount > 2
+ && loopAlertText != null
+ && !loopAlertText.trim().isEmpty();
+ String nextType = active ? loopAlertType : null;
+ String nextText = active ? loopAlertText.trim() : null;
+ Integer nextCount = loopAlertCount != null && loopAlertCount > 0 ? loopAlertCount : null;
+ boolean changed = !Objects.equals(this.loopAlertActive, active)
+ || !Objects.equals(this.loopAlertType, nextType)
+ || !Objects.equals(this.loopAlertText, nextText)
+ || !Objects.equals(this.loopAlertCount, nextCount);
+ if (!changed) {
+ return;
+ }
+
+ this.loopAlertActive = active;
+ this.loopAlertType = nextType;
+ this.loopAlertText = nextText;
+ this.loopAlertCount = nextCount;
+ this.updatedAt = System.currentTimeMillis();
+ if (!active) {
+ return;
+ }
+
+ Map<String, Object> nextDetails = copyDetails(details);
+ nextDetails.put("loopAlertActive", Boolean.TRUE);
+ nextDetails.put("loopAlertType", this.loopAlertType);
+ nextDetails.put("loopAlertText", this.loopAlertText);
+ nextDetails.put("loopAlertCount", this.loopAlertCount);
+ appendEvent("LOOP_REPEAT_ALERT", this.loopAlertText, nextDetails);
+ }
+
private synchronized boolean shouldRemove(long now) {
return terminalExpireAt != null && terminalExpireAt <= now;
+ }
+
+ private synchronized boolean isTerminal() {
+ return isTerminalStatus(this.status);
}
private synchronized StationTaskTraceVo toVo() {
@@ -400,9 +694,69 @@
vo.setSegmentList(copySegmentListWithIssued(segmentList, issuedSegmentCount));
vo.setIssuedSegmentCount(issuedSegmentCount);
vo.setTotalSegmentCount(totalSegmentCount);
+ vo.setLoopAlertActive(loopAlertActive);
+ vo.setLoopAlertType(loopAlertType);
+ vo.setLoopAlertText(loopAlertText);
+ vo.setLoopAlertCount(loopAlertCount);
vo.setUpdatedAt(updatedAt);
- vo.setEvents(new ArrayList<>(events));
+ vo.setEvents(copyEventList(events));
return vo;
+ }
+
+ private synchronized TraceSnapshot toSnapshot() {
+ TraceSnapshot snapshot = new TraceSnapshot();
+ snapshot.setTaskNo(taskNo);
+ snapshot.setThreadImpl(threadImpl);
+ snapshot.setStatus(status);
+ snapshot.setTraceVersion(traceVersion);
+ snapshot.setStartStationId(startStationId);
+ snapshot.setCurrentStationId(currentStationId);
+ snapshot.setFinalTargetStationId(finalTargetStationId);
+ snapshot.setBlockedStationId(blockedStationId);
+ snapshot.setFullPathStationIds(copyIntegerList(fullPathStationIds));
+ snapshot.setIssuedStationIds(copyIntegerList(issuedStationIds));
+ snapshot.setPassedStationIds(copyIntegerList(passedStationIds));
+ snapshot.setPendingStationIds(copyIntegerList(pendingStationIds));
+ snapshot.setLatestIssuedSegmentPath(copyIntegerList(latestIssuedSegmentPath));
+ snapshot.setSegmentList(copySegmentList(segmentList));
+ snapshot.setIssuedSegmentCount(issuedSegmentCount);
+ snapshot.setTotalSegmentCount(totalSegmentCount);
+ snapshot.setLoopAlertActive(loopAlertActive);
+ snapshot.setLoopAlertType(loopAlertType);
+ snapshot.setLoopAlertText(loopAlertText);
+ snapshot.setLoopAlertCount(loopAlertCount);
+ snapshot.setUpdatedAt(updatedAt);
+ snapshot.setTerminalExpireAt(terminalExpireAt);
+ snapshot.setEvents(copyEventList(events));
+ return snapshot;
+ }
+
+ private static TraceTaskState fromSnapshot(TraceSnapshot snapshot) {
+ TraceTaskState state = new TraceTaskState(snapshot.getTaskNo());
+ state.threadImpl = snapshot.getThreadImpl();
+ state.status = snapshot.getStatus() == null ? STATUS_WAITING : snapshot.getStatus();
+ state.traceVersion = snapshot.getTraceVersion() == null ? 0 : snapshot.getTraceVersion();
+ state.startStationId = snapshot.getStartStationId();
+ state.currentStationId = snapshot.getCurrentStationId();
+ state.finalTargetStationId = snapshot.getFinalTargetStationId();
+ state.blockedStationId = snapshot.getBlockedStationId();
+ state.fullPathStationIds = copyIntegerList(snapshot.getFullPathStationIds());
+ state.issuedStationIds = copyIntegerList(snapshot.getIssuedStationIds());
+ state.passedStationIds = copyIntegerList(snapshot.getPassedStationIds());
+ state.pendingStationIds = copyIntegerList(snapshot.getPendingStationIds());
+ state.latestIssuedSegmentPath = copyIntegerList(snapshot.getLatestIssuedSegmentPath());
+ state.segmentList = copySegmentList(snapshot.getSegmentList());
+ state.issuedSegmentCount = snapshot.getIssuedSegmentCount() == null ? 0 : snapshot.getIssuedSegmentCount();
+ state.totalSegmentCount = snapshot.getTotalSegmentCount() == null ? state.segmentList.size() : snapshot.getTotalSegmentCount();
+ state.loopAlertActive = Boolean.TRUE.equals(snapshot.getLoopAlertActive());
+ state.loopAlertType = snapshot.getLoopAlertType();
+ state.loopAlertText = snapshot.getLoopAlertText();
+ state.loopAlertCount = snapshot.getLoopAlertCount();
+ state.updatedAt = snapshot.getUpdatedAt() == null ? System.currentTimeMillis() : snapshot.getUpdatedAt();
+ state.terminalExpireAt = snapshot.getTerminalExpireAt();
+ state.events.clear();
+ state.events.addAll(copyEventList(snapshot.getEvents()));
+ return state;
}
private List<Integer> buildFullPathForRegistration(boolean rerouted,
@@ -448,6 +802,25 @@
this.pendingStationIds = copyIntegerList(fullPath.subList(currentIndex + 1, fullPath.size()));
}
+ private void clearLoopHintState() {
+ this.loopAlertActive = Boolean.FALSE;
+ this.loopAlertType = null;
+ this.loopAlertText = null;
+ this.loopAlertCount = null;
+ }
+
+ private void appendLoopHintDetails(Map<String, Object> details) {
+ if (details == null || this.loopAlertCount == null || this.loopAlertCount <= 0) {
+ return;
+ }
+ details.put("loopAlertCount", this.loopAlertCount);
+ if (Boolean.TRUE.equals(this.loopAlertActive)) {
+ details.put("loopAlertActive", Boolean.TRUE);
+ details.put("loopAlertType", this.loopAlertType);
+ details.put("loopAlertText", this.loopAlertText);
+ }
+ }
+
private boolean acceptTraceVersion(Integer incomingTraceVersion) {
return incomingTraceVersion != null
&& this.traceVersion != null
--
Gitblit v1.9.1