策略模式是一种常用的行为设计模式,它允许在运行时选择一系列算法中的一个来执行,而无需修改客户端代码。策略模式将算法封装起来,使算法的变化不会影响到客户端。
策略模式最常见的作用就是解决代码中冗长的 if-else 或 switch 分支判断语句。
问题引出:
// 根据 type 的不同,执行不同分支的代码逻辑,
private void process(String type){
if(type == "A"){
// 执行一大堆 A 类型的代码逻辑
} else if (type == "B") {
// 执行一大堆 B 类型的代码逻辑
} else if (type == "C") {
// 执行一大堆 C 类型的代码逻辑
} else {
// 上述都没有匹配成功,执行默认的代码逻辑,
// 或者抛出异常打印错误日志等
}
}
一、策略模式的定义
策略模式(Strategy Pattern)属于对象的行为模式,其定义了算法族,分别封装起来,让它们之间可以互相转换,此模式让算法的变化独立于使用算法的用户。
二、策略模式的结构
策略模式通常包含以下角色:
- Strategy(策略接口):定义了一系列的策略算法的接口。
- ConcreteStrategy(具体策略类):实现了策略接口,提供具体的算法实现。
- StrategyFactory(策略工厂类):用来根据具体算法的关键key来生成具体算法实现类
三、策略模式的实现
// 定义策略接口类
public interface Strategy {
void algorithmInterface();
}
// 定义算法实现类A
public class ConcreteStrategyA implements Strategy {
@Override
public void algorithmInterface() {
//ConcreteStrategyA具体的算法...
}
}
// 定义算法实现类B
public class ConcreteStrategyB implements Strategy {
@Override
public void algorithmInterface() {
//ConcreteStrategyB具体的算法...
}
}
策略的创建重点在于使用查表法代替了大量的分支判断,即每次根据 type 去 Map 中获取,省略了大量的 if-else。
public class StrategyFactory {
// Map 的 key 为策略类型,value 为具体的策略实现类
private static final Map<String, Strategy> strategies = new HashMap<>();
// 提前创建好所有策略类,缓存到 Map 中
static {
strategies.put("A", new ConcreteStrategyA());
strategies.put("B", new ConcreteStrategyB());
}
// 需要使用策略时根据 type 去 Map 中 get 即可获取到相应的策略实现类
public static Strategy getStrategy(String type) {
if (type == null || type.isEmpty()) {
throw new IllegalArgumentException("type should not be empty.");
}
return strategies.get(type);
}
}
如何使用呢?
// 根据 type 的不同,执行不同分支的代码逻辑,
private void process(String type){
Strategy strategy = StrategyFactory.getStrategy(type); // type 是A就用A的逻辑,是B就用B的逻辑
// 调用策略类的 algorithmInterface 方法去执行代码逻辑
strategy.algorithmInterface();
}
四、策略模式的优点
- 算法可以自由切换:策略模式提供了一种可以替换算法的方式,而不必修改客户端的代码。
- 代码易于扩展:新的算法可以很容易地添加进系统,而不用修改现有代码。
- 算法可封装:每个算法都是一个独立的类,符合单一职责原则。
五、策略模式的缺点
- 客户端必须知道所有的策略:客户端需要知道有哪些策略可供选择,这可能会使得客户端代码更加复杂。
- 存在多余的对象:每个策略都会产生一个具体的对象,这可能会导致系统资源的浪费。
六、策略模式适用场景
- 当一个系统需要多种算法变体时。
- 当算法的实现可能经常变化时。
- 当需要避免使用条件语句来在多种算法间进行选择时。
通过策略模式,我们可以将算法的选择和使用从客户端逻辑中分离出来,使得系统更加灵活和易于维护。