重构复杂简单变量之状态与策略模式

news2024/10/24 7:02:36

状态与策略模式 主要用于消除复杂的类型代码，并将其替换为更清晰、可维护的状态或策略对象。这个方法通常用于以下情况：

类型代码问题：当我们在类中使用整数或字符串来表示对象的状态或行为时，这可能会导致代码变得难以理解和维护。每当状态变化时，我们都需要在代码中进行大量的条件判断。
状态/策略模式：
- 状态模式：允许对象在内部状态改变时改变其行为。通过将每种状态抽象为一个类，我们可以避免复杂的条件语句。
- 策略模式：定义一系列算法，将每个算法封装起来，并使它们可以互换。这种模式让算法独立于使用它的客户端。

实施步骤

识别类型代码：首先，识别出需要重构的类型代码。例如，某个类中的整数常量用来表示不同的状态。
创建状态/策略类：
- 对于状态模式，创建一个抽象状态类，并为每种状态实现一个子类。
- 对于策略模式，创建一个接口，并为每种策略实现相应的类。
替换条件语句：用状态/策略对象替代原有的条件判断逻辑。这意味着在类中不再直接使用类型代码，而是使用状态/策略对象来处理行为。
更新客户端代码：确保使用新状态/策略对象的代码能正常工作。

示例

假设我们有一个简单的订单类，使用整数表示订单状态：

public class Order {
    public const int NEW = 0;
    public const int PROCESSING = 1;
    public const int COMPLETED = 2;
    
    private int status;

    public void Process() {
        if (status == NEW) {
            // 处理新订单
            status = PROCESSING;
        } else if (status == PROCESSING) {
            // 完成订单处理
            status = COMPLETED;
        }
    }
}

我们可以将其重构为状态模式：

public interface IOrderState {
    void Process(Order order);
}

public class NewOrderState : IOrderState {
    public void Process(Order order) {
        // 处理新订单
        order.SetState(new ProcessingOrderState());
    }
}

public class ProcessingOrderState : IOrderState {
    public void Process(Order order) {
        // 完成订单处理
        order.SetState(new CompletedOrderState());
    }
}

public class CompletedOrderState : IOrderState {
    public void Process(Order order) {
        // 已完成，无法再次处理
    }
}

public class Order {
    private IOrderState state;

    public Order() {
        state = new NewOrderState();
    }

    public void SetState(IOrderState newState) {
        state = newState;
    }

    public void Process() {
        state.Process(this);
    }
}