Spring State Machine状态持久化及高并发EntryAction丢失问题问询

问题1：统一配置状态持久化的简洁方案 #

当前为每个状态单独配置entryAction的方式重复繁琐，Spring State Machine提供了默认入口动作的配置方式，只需一次配置即可让所有状态进入时自动执行持久化逻辑：

实现方式 #

使用configureStates().withStates().defaultEntryAction()方法，替换原来逐个状态配置entryAction的代码：

StateMachineBuilder.Builder<OperationState, OperationEvent> builder = StateMachineBuilder.builder();
builder.configureStates().withStates()
        .states(EnumSet.allOf(OperationState.class))
        .initial(genericOp.getOpState())
        // 统一配置所有状态的默认入口动作
        .defaultEntryAction(updateOperationState(), null);

所有状态在被进入时，都会自动触发updateOperationState()动作，无需再为每个state()单独配置。

额外优化：线程安全获取业务ID #

原代码中updateOperationState()直接使用外部共享的opId变量，高并发下会出现线程安全问题，建议从状态机上下文的消息头中获取opId：

@Bean
private static Action<OperationState, OperationEvent> updateOperationState() {
    return context -> {
        OperationState tgtOpState = context.getStateMachine().getState().getId();
        Message<OperationEvent> message = context.getMessage();
        if (message == null) {
            return;
        }
        // 从消息头中获取opId，保证线程安全
        String opId = message.getHeaders().get("opId", String.class);
        opInfoRepository.updateOpState(opId, tgtOpState);
        log.debug("Updating the operation [{}] state to [{}]", opId, tgtOpState);
    };
}

触发事件时，将opId放入消息头：

stateMachine.sendEvent(MessageBuilder.withPayload(OperationEvent.TRIGGER)
        .setHeader("opId", opInfo.getId())
        .build());

问题2：高并发下entryAction调用丢失的根因与调优 #

常见根因 #

1. 线程安全问题：原代码依赖外部共享的opId变量，高并发下多个线程覆盖该变量，导致部分更新逻辑执行错误或丢失。
2. 事务未正确管理：持久化操作updateOpState()未加事务，数据库更新失败时无回滚或重试，表现为动作"丢失"。
3. 状态机实例非线程安全：复用的状态机实例未做线程隔离，或每次请求创建新实例，导致状态变更逻辑混乱。
4. 并发更新冲突：多个线程同时更新同一条OpInfo记录，未加锁机制导致更新被覆盖。

调优配置与解决方案 #

1. 修复线程安全问题：按上述优化，从消息头获取opId，避免使用共享变量。
2. 添加事务管理：在updateOpState()方法或对应服务层添加@Transactional注解，保证持久化操作的原子性：

@Transactional
public void updateOpState(String opId, OperationState state) {
    opInfoRepository.updateOpState(opId, state);
}

3. 配置线程安全的状态机：使用StateMachineFactory创建线程安全的状态机实例，或配置同步执行策略与线程池：

@Bean
public StateMachineFactory<OperationState, OperationEvent> stateMachineFactory() {
    StateMachineConfigurationConfigurer<OperationState, OperationEvent> config = configurer -> {
        config
            .withConfiguration()
            .autoStartup(false)
            .taskExecutor(Executors.newFixedThreadPool(10)) // 配置业务线程池
            .lockProvider(new DefaultStateMachineLockProvider<>()); // 加锁保证状态机线程安全
    };
    // 其他状态、转换配置...
    return new StateMachineFactoryBuilder().withConfiguration(config).build();
}

4. 添加乐观锁防并发覆盖：在OpInfo实体中添加版本字段，避免并发更新丢失：

@Entity
@Builder
public class OpInfo {
    // 原有字段...
    @Version
    private Integer version; // 乐观锁字段
}

5. 启用状态机持久化：使用Spring State Machine的StateMachinePersister，将状态机状态持久化到数据库，保证内存状态与数据库一致：

@Bean
public StateMachinePersister<OperationState, OperationEvent, OpInfo> stateMachinePersister(OpInfoRepository repository) {
    return new DefaultStateMachinePersister<>(new RepositoryStateMachinePersist<>(repository));
}

内容的提问来源于stack exchange，提问作者Krutik