SerLiunx/state-management
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: 状态机支持发布、订阅事件.

Browse Source
This commit is contained in:
SerLiunx-ctrl
2025-01-24 09:43:34 +08:00
parent 54e6f56bc6
commit a1bdb96638
5 changed files with 52 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
src/main/java/com/serliunx/statemanagement/machine/AbstractStateMachine.java
View File

@@ -9,6 +9,7 @@ import java.util.List;
import java.util.Map;
import java.util.concurrent.Executor;
import java.util.concurrent.ExecutorService;
import java.util.function.Consumer;
/**
* 状态机抽象实现, 实现最基本功能
@@ -31,6 +32,10 @@ public abstract class AbstractStateMachine<S> extends AbstractStateManager<S> im
* 交换事件集合
*/
protected final Map<String, List<StateHandlerWrapper<S>>> exchangeHandlers;
/**
* 事件注册集合
*/
protected final Map<Object, List<Consumer<StateMachine<S>>>> eventRegistries;
/**
* 异步执行器
*/
@@ -55,6 +60,7 @@ public abstract class AbstractStateMachine<S> extends AbstractStateManager<S> im
Map<S, List<StateHandlerWrapper<S>>> entryHandlers,
Map<S, List<StateHandlerWrapper<S>>> leaveHandlers,
Map<String, List<StateHandlerWrapper<S>>> exchangeHandlers,
Map<Object, List<Consumer<StateMachine<S>>>> eventRegistries,
Executor executor,
Boolean async
) {
@@ -64,6 +70,7 @@ public abstract class AbstractStateMachine<S> extends AbstractStateManager<S> im
this.exchangeHandlers = exchangeHandlers;
this.executor = executor;
this.async = async;
this.eventRegistries = eventRegistries;
}
@Override

9
src/main/java/com/serliunx/statemanagement/machine/StandardStateMachine.java
View File

@@ -5,6 +5,7 @@ import com.serliunx.statemanagement.machine.handler.StateHandlerWrapper;
import java.util.List;
import java.util.Map;
import java.util.concurrent.Executor;
import java.util.function.Consumer;
/**
* 状态机的标准实现
@@ -28,14 +29,18 @@ public class StandardStateMachine<S> extends AbstractStateMachine<S> implements
Map<S, List<StateHandlerWrapper<S>>> entryHandlers,
Map<S, List<StateHandlerWrapper<S>>> leaveHandlers,
Map<String, List<StateHandlerWrapper<S>>> exchangeHandlers,
Map<Object, List<Consumer<StateMachine<S>>>> eventRegistries,
Executor executor,
Boolean async
) {
super(stateList, entryHandlers, leaveHandlers, exchangeHandlers, executor, async);
super(stateList, entryHandlers, leaveHandlers, exchangeHandlers, eventRegistries, executor, async);
}
@Override
public void publish(Object event) {
List<Consumer<StateMachine<S>>> consumers = eventRegistries.get(event);
if (consumers != null) {
consumers.forEach(consumer -> consumer.accept(this));
}
}
}

16
src/main/java/com/serliunx/statemanagement/machine/StateMachineBuilder.java
View File

@@ -5,6 +5,7 @@ import com.serliunx.statemanagement.machine.handler.StateHandlerWrapper;
import java.util.*;
import java.util.concurrent.Executor;
import java.util.function.Consumer;
/**
* 状态机构建
@@ -38,6 +39,7 @@ public final class StateMachineBuilder<S> {
private final Map<S, List<StateHandlerWrapper<S>>> entryHandlers = new HashMap<>(64);
private final Map<S, List<StateHandlerWrapper<S>>> leaveHandlers = new HashMap<>(64);
private final Map<String, List<StateHandlerWrapper<S>>> exchangeHandlers = new HashMap<>(64);
private final Map<Object, List<Consumer<StateMachine<S>>>> eventRegistries = new HashMap<>(64);
private StateMachineBuilder(List<S> states) {
this.stateList = states;
@@ -169,6 +171,18 @@ public final class StateMachineBuilder<S> {
return whenEntry(state, handler, null);
}
/**
* 注册当前状态机感兴趣的事件
*
* @param event 事件
* @param logic 切换逻辑
*/
public StateMachineBuilder<S> whenHappened(Object event, Consumer<StateMachine<S>> logic) {
List<Consumer<StateMachine<S>>> consumers = eventRegistries.computeIfAbsent(event, k -> new ArrayList<>());
consumers.add(logic);
return this;
}
/**
* 指定状态机的执行器
* <p>
@@ -238,7 +252,7 @@ public final class StateMachineBuilder<S> {
}
if (type.equals(StateMachineType.STANDARD)) {
return (M)new StandardStateMachine<>(stateList, entryHandlers,
leaveHandlers, exchangeHandlers, executor, async);
leaveHandlers, exchangeHandlers, eventRegistries, executor, async);
}
throw new IllegalArgumentException("未知的状态机类型: " + type);
}