策略模式是一种行为设计模式,它允许在运行时选择算法的行为。它使得可以定义一系列算法,并将每个算法封装到独立的类中,使它们可以相互替换,而不影响客户端使用这些算法的方式。这种模式有助于将算法的实现与使用算法的客户端代码分离开来。
策略模式的主要角色如下:
-
抽象策略(Strategy)类:这是一个抽象角色,通常由一个接口或抽象类实现。此角色给出所有 的具体策略类所需的接口。
-
具体策略(Concrete Strategy)类:实现了抽象策略定义的接口,提供具体的算法实现或行为。
-
环境(Context)类:持有一个策略类的引用,最终给客户端调用。
那么哪些场景下适合使用策略模式呢?
- 订单的支付策略(支付宝、微信、银行卡...)
- 解析不同类型excel (xls格式、xlsx格式)
- 打折促销(满300元9折、满500元8折、满1000元7折...)
- 物流运费阶梯计算(5kg以下、5-10kg、10-2kg、20kg以上)
一句话总结:只要代码中有冗长的 if-else 或 switch 分支判断都可以采用策略模式优化
废话不多说,直接上代码!
枚举类(定义策略的唯一标识)
/**
* @fileName: StrategyE
* @author: 四面神
* @createTime:2023/12/1 20:07
* 描述:策略标识
*/
@AllArgsConstructor
public enum StrategyE {
STRATEGY9("九折"),
STRATEGY95("九五折");
private String discount;
}
抽象策略接口
/**
* @fileName: DiscountStrategy
* @author: 四面神
* @createTime:2023/11/18 19:02
* 描述:抽象策略接口
*/
public interface DiscountStrategy {
//定义抽象策略接口
Double discount(Double price);
//策略的唯一标识
StrategyE Mark();
}
具体策略类
/**
* @fileName: Discount9Strategy
* @author: 四面神
* @createTime:2023/11/18 19:04
* 描述:具体策略1
*/
@Component
public class Discount9Strategy implements DiscountStrategy{
@Override
public Double discount(Double price) {
return price*0.9;
}
@Override
public StrategyE Mark() {
return StrategyE.STRATEGY9;
}
}
/**
* @fileName: Discount95Strategy
* @author: 四面神
* @createTime:2023/11/18 19:05
* 描述: 具体策略2
*/
@Component
public class Discount95Strategy implements DiscountStrategy{
@Override
public Double discount(Double price) {
return price*0.95;
}
@Override
public StrategyE Mark() {
return StrategyE.STRATEGY95;
}
}
环境类
/**
* @fileName: DiscountStrategyFactory
* @author: 四面神
* @createTime:2023/11/18 19:08
* 描述:具体环境上下文类
*/
@Component
public class DiscountStrategyFactory implements ApplicationContextAware {
private final Map<StrategyE,DiscountStrategy> discountStrategies = new HashMap<>();
public DiscountStrategy chooseStrategy(StrategyE type){
return discountStrategies.get(type);
}
@Override
public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {
//从IOC容器中获取 DiscountStrategy 类型的所有Bean对象,存入map中方便获取
Map<String, DiscountStrategy> discountStrategyMap = applicationContext.getBeansOfType(DiscountStrategy.class);
//将具体的策略算法标识以及本身添加到我们自定义的MAP中,方便屏蔽客户端感知调用
discountStrategyMap.forEach((key,val)->
discountStrategies.put(val.Mark(),val)
);
}
}
测试类
@RequiredArgsConstructor
public class TestStrategy {
private final DiscountStrategyFactory discountStrategyFactory;
public void testStrategy(StrategyE type){
//获取具体策略类
DiscountStrategy discountStrategy = discountStrategyFactory.chooseStrategy(type);
//调用具体方法
System.out.println(discountStrategy.discount(100d));
}
}
优点:
-
灵活性和可维护性:策略模式允许在运行时动态选择算法,使得系统更加灵活。新增或修改算法不影响客户端代码。
-
降低耦合度:策略模式将每个算法封装到独立的类中,使得各个算法之间的耦合度降低,易于理解、维护和扩展。
-
避免使用条件语句:避免大量的条件语句判断不同的算法,提高了代码的可读性和可维护性。
-
单一职责原则:每个具体策略类都只关注自己的算法逻辑,符合单一职责原则。