巧用状态机思想，让业务代码更简洁清晰

前端

2023-11-20 22:04:10

状态机：让复杂状态流转轻松搞定

在软件开发中，我们时常会碰见状态流转的场景，比如订单从下单到发货，需经历一系列状态变化，如待付款、待发货、已发货、已收货等等。倘若我们采用直接的 if-else 语句判断状态，代码会变得繁琐且难以维护。

何为状态机？

状态机是一种用于处理复杂状态流转的利器。它抽象出状态与状态间转换关系，助我们更清晰地理解和管理状态。

状态机的优点

将状态机思想融入业务代码，可带来诸多好处：

代码简洁易懂： 状态机抽象出状态及转换关系，使得我们对状态的理解和管理更为清晰。这大大提升了代码的可读性和可维护性。
复用性更强： 状态机是一种通用工具，适用于多种场景。这增强了代码的可复用性。
拓展性更好： 状态机支持轻松扩展，以应对新增状态或转换关系。这使得代码更易适应需求变化。

业务场景示例

以下是一个订单管理系统的业务场景：订单从下单到发货，需要经历多个状态，如待付款、待发货、已发货、已收货等。直接使用 if-else 语句判断状态会让代码变得复杂难懂。

状态机设计

我们可以使用状态机来抽象订单的状态流转。状态机如下所示：

                  +----------------+
                  |                |
             +-----|   待付款   |-----+
             |     |                |     |
             |     +----------------+     |
             |                               |
             |                               |
             |                               |
+-----------+                               +-----------+
|           |                               |           |
|  待发货  +-----------------------------+  已发货  |
|           |                               |           |
+-----------+                               +-----------+
             |                               |
             |                               |
             |                               |
             |                               |
             |                               |
             +-------------------------------+
                  |                |
                  |   已收货   |
                  +----------------+

代码实现

Python 代码实现状态机如下：

class Order:
    def __init__(self):
        self.state = "待付款"

    def pay(self):
        if self.state == "待付款":
            self.state = "待发货"

    def deliver(self):
        if self.state == "待发货":
            self.state = "已发货"

    def receive(self):
        if self.state == "已发货":
            self.state = "已收货"

order = Order()
order.pay()
order.deliver()
order.receive()

print(order.state)  # 已收货