策略模式的优缺点及优化方案

策略模式是一种非常常用的设计模式，它的核心思想是将算法或行为封装成独立的类，使得它们可以相互替换。这样，客户端可以根据需要动态选择不同的策略，而不需要修改原有的代码结构。下面我们来聊聊策略模式的优点、缺点，以及如何通过结合简单工厂模式来优化它。

策略模式的优点

算法自由切换
策略模式允许高层模块（客户端）根据需要传入不同的策略实现类，从而动态切换算法或行为。比如，你可以根据不同的条件选择不同的策略，而不需要修改客户端代码。
良好的扩展性
策略模式符合开闭原则（Open/Closed Principle）。当需要新增策略时，只需要添加一个新的策略类，而不需要修改现有的代码。这使得系统更容易扩展和维护。
避免多重条件判断
在没有使用策略模式的情况下，我们可能会通过大量的 if-else 或 switch-case 来判断不同的条件。而策略模式将这些条件判断转移到了策略类的选择上，使得代码更加清晰和简洁。

策略模式的缺点

策略类需要暴露给高层模块
在传统的策略模式中，所有具体的策略类都需要在高层模块中实例化，然后传递给上下文类。这可能会导致高层模块与具体策略类之间的耦合度增加。
策略膨胀问题
如果策略类过多，可能会导致“策略膨胀”问题。比如，在一个复杂的业务场景中，可能会有成百上千个策略类，这会增加系统的复杂性，同时也增加了维护成本。

优化方案：策略模式 + 简单工厂模式

为了解决策略模式的缺点，我们可以将策略模式与简单工厂模式结合使用。通过工厂类来管理策略类的创建，从而减少高层模块与具体策略类之间的耦合。

优化后的代码实现

1. 上下文类（Context）

public class AgeContext {

    private AgeStrategy ageStrategy;

    public AgeContext(int age) {
        this.ageStrategy = AgeStrategyFactory.getAgeStrategy(age);
    }

    public String executeAgeStrategy() {
        return ageStrategy.execute();
    }
}

在上下文类中，我们通过工厂类 AgeStrategyFactory 来获取具体的策略对象，而不是直接在高层模块中实例化策略类。

2. 策略接口（Strategy）

public interface AgeStrategy {
    String execute();
}

@Service
public class ChildStrategy implements AgeStrategy {
    @Override
    public String execute() {
        return "child";
    }
}

@Service
public class AdultStrategy implements AgeStrategy {
    @Override
    public String execute() {
        return "adult";
    }
}

@Service
public class SeniorStrategy implements AgeStrategy {
    @Override
    public String execute() {
        return "senior";
    }
}

策略接口定义了所有策略类的通用行为，具体的策略类实现这个接口并提供自己的逻辑。

3. 工厂类（Factory）

@Configuration
public class AgeStrategyFactory {
    private static final Map<Integer, AgeStrategy> ageStrategyMap = new HashMap<>();

    static {
        ageStrategyMap.put(0, new ChildStrategy());
        ageStrategyMap.put(1, new AdultStrategy());
        ageStrategyMap.put(2, new SeniorStrategy());
    }

    public static AgeStrategy getAgeStrategy(int age) {
        if (age < 18) {
            return ageStrategyMap.get(0);
        } else if (age < 65) {
            return ageStrategyMap.get(1);
        } else {
            return ageStrategyMap.get(2);
        }
    }
}

工厂类负责根据传入的参数（如年龄）来选择合适的策略类。通过工厂类，我们可以将策略类的创建逻辑集中管理，减少高层模块的负担。

4. 测试类（Test）

@Test
public void testAgeContext() {
    AgeContext ageContext = new AgeContext(10);
    String result = ageContext.executeAgeStrategy();
    System.out.println(result);
    assertEquals("child", result);
}

在测试类中，我们只需要创建上下文对象并传入参数，剩下的策略选择和执行逻辑都由上下文类和工厂类来完成。

总结

通过将策略模式与简单工厂模式结合，我们可以有效解决策略模式的缺点，比如策略类的暴露问题和策略膨胀问题。工厂类负责管理策略类的创建，使得高层模块与具体策略类之间的耦合度降低，同时也使得系统更加灵活和易于扩展。

如果你在实际开发中遇到类似的需求，不妨试试这种优化方案。希望这篇文章对你有帮助！如果你有任何问题或想法，欢迎在评论区留言讨论！

Yufelix

Yufelix

策略模式的优缺点及优化方案