增加航线状态机

liusuyi

2025-12-18 073239cedaab68465e109b1957144f5fca17958d

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
package com.ard.work.config;
 
import com.ard.work.domain.enums.RouteEvent;
import com.ard.work.domain.enums.RouteState;
import lombok.RequiredArgsConstructor;
import org.springframework.messaging.Message;
import org.springframework.messaging.support.MessageBuilder;
import org.springframework.statemachine.StateMachine;
import org.springframework.statemachine.service.StateMachineService;
import org.springframework.statemachine.support.DefaultExtendedState;
import org.springframework.statemachine.support.DefaultStateMachineContext;
import org.springframework.stereotype.Service;
 
import java.util.concurrent.ConcurrentHashMap;
 
@Service
@RequiredArgsConstructor
public class RouteStateMachineService {
 
    private final StateMachineService<RouteState, RouteEvent> stateMachineService;
    private final RouteStateMachineListener listener;
    private final ConcurrentHashMap<String, Boolean> listenerRegisteredMap = new ConcurrentHashMap<>();
 
    private StateMachine<RouteState, RouteEvent> getStateMachine(String boxSn) {
        StateMachine<RouteState, RouteEvent> stateMachine =
                stateMachineService.acquireStateMachine(boxSn);
 
        // 使用原子操作确保 listener 只添加一次
        listenerRegisteredMap.computeIfAbsent(boxSn, k -> {
            stateMachine.addStateListener(listener);
            return true;
        });
        stateMachine.start();
        return stateMachine;
    }
 
    /**
     * 发送事件到指定boxSn的状态机
     */
    public boolean sendEvent(String boxSn, RouteEvent event) {
        try {
            // 获取/创建指定boxSn的状态机实例
            StateMachine<RouteState, RouteEvent> stateMachine = getStateMachine(boxSn);
 
            // 构建事件消息（携带boxSn作为头信息）
            Message<RouteEvent> message = MessageBuilder
                    .withPayload(event)
                    .setHeader("boxSn", boxSn)
                    .build();
 
            // 发送事件并返回执行结果
            return stateMachine.sendEvent(message);
        } catch (Exception e) {
            // 异常处理：日志记录、状态机重置等
            throw new RuntimeException("发送航线状态事件失败，boxSn=" + boxSn + ", event=" + event, e);
        }
    }
 
    /**
     * 获取指定boxSn的当前状态
     */
    public RouteState getCurrentState(String boxSn) {
        try {
            StateMachine<RouteState, RouteEvent> stateMachine =
                    stateMachineService.acquireStateMachine(boxSn);
            return stateMachine.getState().getId();
        } catch (Exception e) {
            // 未找到状态机实例时返回初始状态
            return RouteState.IDLE;
        }
    }
 
    /**
     * 释放指定boxSn的状态机实例（可选，节省资源）
     */
    public void releaseStateMachine(String boxSn) {
        stateMachineService.releaseStateMachine(boxSn);
    }
 
    /**
     * 重置指定boxSn的状态机（恢复到初始状态）
     */
    public void resetStateMachine(String boxSn) {
        try {
            StateMachine<RouteState, RouteEvent> stateMachine =
                    stateMachineService.acquireStateMachine(boxSn);
 
            // 1️⃣ 停止状态机
            stateMachine.stop();
 
            // 2️⃣ 构造 ExtendedState（可选，但强烈推荐）
            DefaultExtendedState extendedState = new DefaultExtendedState();
            extendedState.getVariables().put("boxSn", boxSn);
 
            // 3️⃣ 重置状态机到 IDLE
            stateMachine.getStateMachineAccessor()
                    .doWithAllRegions(accessor -> {
                        accessor.resetStateMachine(
                                new DefaultStateMachineContext<>(
                                        RouteState.IDLE, // 初始状态
                                        null,            // event
                                        null,            // event headers
                                        extendedState    // 扩展状态
                                )
                        );
                    });
 
            // 4️⃣ 重新启动
            stateMachine.start();
 
        } catch (Exception e) {
            throw new RuntimeException("重置航线状态机失败，boxSn=" + boxSn, e);
        }
    }
}