策略模式是一种行为型设计模式,定义一族算法类,将每个算法分别封装起来,让它们可以互相替换。策略模式可以使算法的变化独立于使用它们的客户端。
策略类的定义比较简单,包含一个策略接口和一组实现这个接口的策略类。因为所有的策略类都实现相同的接口,所以客户端代码基于接口而非实现编程,可以灵活地替换不同的策略。
假如有如下需求:根据客户端的不同请求类型,服务端则执行不同的算法计算相应的结果,通常会按以下代码进行实现。
public class AlgorithmService {
public double algorithm(String type) {
double result = 0.0;
if ("A".equals(type)) { // A类型
//...省略A算法代码
} else if ("B".equals(type)) { // B类型
//...省略B算法代码
} else if ("C".equals(type)) { // C类型
//...省略C算法代码
}
return result;
}
}
这种方式实现简单,但会存在if-else或switch-case分支判断逻辑,后续如果增加算法必须对AlgorithmService进行修改,不符合开闭原则,如果要移除if-else或switch-case分支,策略模式就派上用场了。我们提供一个策略接口,然后将不同的算法实现封装到策略实现中,并提供一个策略工厂执行不同的策略算法,重构之后的代码就没有了 if-else 分支判断语句了。
// 定义策略接口
interface Strategy {
double algorithm();
}
// 定义策略A实现
class ConcreteStrategyA implements Strategy {
@Override
public double algorithm() {
//具体的算法...
return 0;
}
}
// 定义策略B实现
class ConcreteStrategyB implements Strategy {
@Override
public double algorithm() {
//具体的算法...
return 0;
}
}
// 定义策略工厂
class StrategyFactory {
private static final Map<String, Strategy> strategies = new HashMap<>();
static {
strategies.put("A", new ConcreteStrategyA());
strategies.put("B", new ConcreteStrategyB());
}
public static Strategy getStrategy(String type) {
if (!strategies.containsKey(type)) {
throw new IllegalArgumentException("type is ilegal.");
}
return strategies.get(type);
}
}
public class Application {
public static void main(String[] args) {
Strategy strategy = StrategyFactory.getStrategy("A");
System.out.println(strategy.algorithm());
}
}
综上所述,策略模式定义一族算法类,将每个算法分别封装起来,让它们可以互相替换。策略模式可以使算法的变化独立于使用它们的客户端,这里的客户端代指使用算法的代码。策略类的定义比较简单,包含一个策略接口和一组实现这个接口的策略类。策略的创建由工厂类来完成,封装策略创建的细节。
有人觉得策略模式的作用是避免 if-else 或switch-case分支判断逻辑,个人觉得如果分支判断数量不多,且判断中的逻辑不复杂,在一个方法中就可以解决, if-else这种写法没有任何问题,如果用策略模式,需要N+1(算法个数+策略工厂)个类,反倒是一种过度设计。实际上,策略模式主要的作用还是解耦策略的定义、创建和使用,控制代码的复杂度,满足开闭原则。