Spring状态机深度解析：从入门到生产实战-育师

Spring State Machine是Spring生态系统中一个强大的状态机框架，它让复杂的状态流转变得优雅而简单。本文将带你从基础概念出发，逐步深入理解并掌握Spring状态机在实际生产环境中的应用。

一、状态机是什么？为什么要用它？

想象一下订单系统：用户下单后，订单会经历"待支付→已支付→待发货→已发货→已完成"等一系列状态变化。如果在代码里用if-else来处理这些状态流转，很快就会变成一团乱麻。

状态机（State Machine）就是解决这类问题的利器！它明确定义了：

状态（State）：系统可能处于的状态
事件（Event）：触发状态变化的动作
转换（Transition）：状态之间的流转规则

二、Spring状态机核心概念

2.1 三大核心组件

java // 1. 定义状态枚举 public enum OrderStatus { WAIT_PAYMENT, // 待支付 PAID, // 已支付 WAIT_DELIVER, // 待发货 DELIVERED, // 已发货 COMPLETED, // 已完成 CANCELLED // 已取消 } // 2. 定义事件枚举 public enum OrderEvent { PAY, // 支付 DELIVER, // 发货 RECEIVE, // 收货 CANCEL // 取消 } // 3. 配置状态机 @Configuration @EnableStateMachine public class OrderStateMachineConfig extends StateMachineConfigurerAdapter<OrderStatus, OrderEvent> { @Override public void configure(StateMachineStateConfigurer<OrderStatus, OrderEvent> states) throws Exception { states.withStates() .initial(OrderStatus.WAIT_PAYMENT) .states(EnumSet.allOf(OrderStatus.class)); } @Override public void configure(StateMachineTransitionConfigurer<OrderStatus, OrderEvent> transitions) throws Exception { transitions .withExternal() .source(OrderStatus.WAIT_PAYMENT) .target(OrderStatus.PAID) .event(OrderEvent.PAY) .and() .withExternal() .source(OrderStatus.PAID) .target(OrderStatus.WAIT_DELIVER) .event(OrderEvent.DELIVER) .and() .withExternal() .source(OrderStatus.WAIT_DELIVER) .target(OrderStatus.DELIVERED) .event(OrderEvent.RECEIVE) .and() .withExternal() .source(OrderStatus.DELIVERED) .target(OrderStatus.COMPLETED) .event(OrderEvent.RECEIVE); } }

2.2 状态持久化

生产环境中，状态必须持久化。Spring状态机支持多种持久化方式：

java @Service @RequiredArgsConstructor public class OrderService { private final StateMachineFactory<OrderStatus, OrderEvent> factory; private final StateMachinePersist<OrderStatus, OrderEvent, String> persist; public boolean pay(String orderId) { StateMachine<OrderStatus, OrderEvent> sm = restoreStateMachine(orderId); boolean result = sm.sendEvent(OrderEvent.PAY); if (result) { persistStateMachine(orderId, sm); // 发送支付成功消息 publishPaymentSuccessEvent(orderId); } return result; } private StateMachine<OrderStatus, OrderEvent> restoreStateMachine(String orderId) { try { return persist.restore(factory.getStateMachine(), orderId); } catch (Exception e) { throw new RuntimeException("恢复状态机失败", e); } } private void persistStateMachine(String orderId, StateMachine<OrderStatus, OrderEvent> sm) { try { persist.persist(sm, orderId); } catch (Exception e) { throw new RuntimeException("保存状态机失败", e); } } }

三、生产实战：工作流引擎

让我们看一个更复杂的例子——审批工作流系统：

java // 支持并行审批的复杂状态机 @Configuration @EnableStateMachine(name = "workflowStateMachine") public class WorkflowStateMachineConfig extends EnumStateMachineConfigurerAdapter<WorkflowState, WorkflowEvent> { @Override public void configure(StateMachineStateConfigurer<WorkflowState, WorkflowEvent> states) throws Exception { states .withStates() .initial(WorkflowState.DRAFT) .fork(WorkflowState.FORK) .join(WorkflowState.JOIN) .state(WorkflowState.FINISHED) .and() .withStates() .parent(WorkflowState.FORK) .initial(WorkflowState.DEPT_APPROVAL) .state(WorkflowState.DEPT_APPROVED) .and() .withStates() .parent(WorkflowState.FORK) .initial(WorkflowState.FINANCE_APPROVAL) .state(WorkflowState.FINANCE_APPROVED); } @Override public void configure(StateMachineTransitionConfigurer<WorkflowState, WorkflowEvent> transitions) throws Exception { transitions // 提交到并行审批 .withExternal() .source(WorkflowState.DRAFT) .target(WorkflowState.FORK) .event(WorkflowEvent.SUBMIT) // 部门审批分支 .and() .withExternal() .source(WorkflowState.DEPT_APPROVAL) .target(WorkflowState.DEPT_APPROVED) .event(WorkflowEvent.DEPT_APPROVE) // 财务审批分支 .and() .withExternal() .source(WorkflowState.FINANCE_APPROVAL) .target(WorkflowState.FINANCE_APPROVED) .event(WorkflowEvent.FINANCE_APPROVE) // 合并后完成 .and() .withExternal() .source(WorkflowState.JOIN) .target(WorkflowState.FINISHED) .event(WorkflowEvent.COMPLETE); } }

四、状态监听器：记录每一次变化

javan@Component @WithStateMachine public class OrderStateListener { private static final Logger log = LoggerFactory.getLogger(OrderStateListener.class); @OnTransition(target = "PAID") public void onPay(Message<OrderEvent> message) { String orderId = getHeader(message, "orderId"); log.info("订单{}支付成功，状态流转到已支付", orderId); // 触发后续业务逻辑 paymentSuccessHandler.handle(orderId); } @OnTransition(target = "DELIVERED") public void onDeliver(Message<OrderEvent> message) { String orderId = getHeader(message, "orderId"); log.info("订单{}已发货，状态流转到已发货", orderId); // 发送短信通知 smsService.sendDeliverySms(orderId); } @OnTransitionEnd public void onTransitionEnd(StateContext<OrderStatus, OrderEvent> context) { log.info("状态转换完成：{} -> {}, 事件：{}", context.getSource().getId(), context.getTarget().getId(), context.getEvent() ); // 持久化状态转换记录 transitionLogService.log(context); } private String getHeader(Message<OrderEvent> message, String headerName) { return message.getHeaders().get(headerName, String.class); } }

五、Guards：智能的状态转换守卫

java @Component public class OrderGuard { @Bean public Guard<OrderStatus, OrderEvent> payGuard() { return context -> { String orderId = context.getMessageHeader("orderId"); BigDecimal amount = orderService.getOrderAmount(orderId); // 检查订单金额 if (amount.compareTo(BigDecimal.ZERO) <= 0) { log.warn("订单{}支付失败：金额为0", orderId); return false; } // 检查库存 boolean hasStock = inventoryService.checkStock(orderId); if (!hasStock) { log.warn("订单{}支付失败：库存不足", orderId); return false; } return true; }; } } // 在状态机配置中使用guard @Override public void configure(StateMachineTransitionConfigurer<OrderStatus, OrderEvent> transitions) throws Exception { transitions .withExternal() .source(OrderStatus.WAIT_PAYMENT) .target(OrderStatus.PAID) .event(OrderEvent.PAY) .guard(payGuard()); // 添加守卫条件 }

六、实战技巧与最佳实践

6.1 状态机可视化

javan@RestController @RequestMapping("/state-machine") public class StateMachineVisualController { @GetMapping("/diagram/{orderId}") public ResponseEntity<String> getStateDiagram(@PathVariable String orderId) { // 获取当前状态 OrderStatus currentStatus = orderService.getOrderStatus(orderId); // 生成PlantUML格式的状态图 String diagram = generatePlantUMLDiagram(currentStatus); return ResponseEntity.ok() .contentType(MediaType.TEXT_PLAIN) .body(diagram); } private String generatePlantUMLDiagram(OrderStatus currentStatus) { StringBuilder sb = new StringBuilder(); sb.append("@startuml\n"); sb.append("[*] --> WAIT_PAYMENT\n"); sb.append("WAIT_PAYMENT --> PAID : PAY\n"); sb.append("PAID --> WAIT_DELIVER : DELIVER\n"); sb.append("WAIT_DELIVER --> DELIVERED : RECEIVE\n"); sb.append("DELIVERED --> COMPLETED : RECEIVE\n"); // 高亮当前状态 sb.append("skinparam state {\n"); sb.append(" BackgroundColor<<Current>> LightBlue\n"); sb.append("}\n"); sb.append("state ").append(currentStatus).append(" <<Current>>\n"); sb.append("@enduml\n"); return sb.toString(); } }

6.2 分布式状态一致性

java // 使用分布式锁确保状态转换的原子性 @Service public class DistributedOrderService { private final RedissonClient redisson; private final StateMachineFactory<OrderStatus, OrderEvent> factory; public boolean transition(String orderId, OrderEvent event) { RLock lock = redisson.getLock("order:state:" + orderId); try { // 最多等待3秒，持锁10秒 if (lock.tryLock(3, 10, TimeUnit.SECONDS)) { // 恢复状态机 StateMachine<OrderStatus, OrderEvent> sm = restoreStateMachine(orderId); // 发送事件并处理结果 boolean result = sm.sendEvent(event); if (result) { // 持久化新状态 persistStateMachine(orderId, sm); // 发布领域事件 publishDomainEvent(orderId, event, sm.getState().getId()); } return result; } } catch (InterruptedException e) { Thread.currentThread().interrupt(); throw new RuntimeException("状态转换被中断", e); } finally { if (lock.isHeldByCurrentThread()) { lock.unlock(); } } return false; } }