什么是策略模式?
-在策略模式(Strategy Pattern)中,一个类的行为或其算法可以在运行时更改。这种类型的设计模式属于行为型模式。
策略模式主要解决了什么问题?
在有多种算法相似的情况下,使用 if...else 所带来的复杂和难以维护。
策略模式实现的方式有哪几种?
- 可以基于自定义注解+spring启动监听+枚举实现策略。(个人推荐)
- 借助spring的生命周期,使用ApplicationContextAware接口,把对用的策略,初始化到map里面。
- 借助工厂模式,把对应的策略,初始化化到map里面。
基于自定义注解+spring启动监听+枚举实现策略具体实现:
业务需求:根据订单类型生成不同平台的订单,现假设有京东订单和网易严选2个平台的订单;
具体代码实现如下:
- 创建平台订单枚举
@Getter
public enum OmsOrderTypeEnum {
JD_ORDER("JD","京东订单"),
WYYX_ORDER("WYYX","网易严选");
private String code;
private String name;
OmsOrderTypeEnum(String code,String name){
this.code = code;
this.name = name;
}
/**
* 根据枚举code查询枚举
* @param code
* @return
*/
public static OmsOrderTypeEnum queryOmsOrderTypeEnumByCode(String code){
for (OmsOrderTypeEnum omsOrderTypeEnum:values()){
if (StringUtils.isNoneBlank(code) && omsOrderTypeEnum.code.equals(code)){
return omsOrderTypeEnum;
}
}
return null;
}
}
- 定义自定义注解返回枚举类
@Documented
@Retention(RetentionPolicy.RUNTIME)
@Target(ElementType.TYPE )
public @interface OmsOrderType {
OmsOrderTypeEnum value();
}
- 创建策略配置类实现启动监听加载策略
@Component
public class OrderTypeServiceMap implements ApplicationListener<ContextRefreshedEvent> {
@Autowired
private Map<String, OrderService> orderServiceMap;
public static final Map<OmsOrderTypeEnum,OrderService> SERVICE_HAND_MAP = new HashMap<>(2);
@Override
public void onApplicationEvent(ContextRefreshedEvent contextRefreshedEvent) {
for (Map.Entry<String, OrderService> entry:orderServiceMap.entrySet()){
Class<? extends OrderService> aClass = entry.getValue().getClass();
if (aClass.isAnnotationPresent(OmsOrderType.class)){
OmsOrderType annotation = aClass.getAnnotation(OmsOrderType.class);
SERVICE_HAND_MAP.put(annotation.value(),entry.getValue());
continue;
}
/*
* 说明:如果service类方法添加事务,那么service会被spring事务代理,这里的aClass会是一个代理类,
* 所以这里需要做一个兼容,如果是代理类则需要找到目标类获取目标类上的注解
*/
Class<?> superclass = aClass.getSuperclass();
if (superclass.isAnnotationPresent(OmsOrderType.class)){
OmsOrderType annotation = superclass.getAnnotation(OmsOrderType.class);
SERVICE_HAND_MAP.put(annotation.value(),entry.getValue());
}
}
}
}
- 创建Service接口
public interface OrderService {
/**
* 新增订单
* @param omsOrderCmd
* @return
*/
String saveOrder(Supplier<OmsOrderCmd> omsOrderCmd);
}
- 在对应的Service实现类上加上自定注解即可
@OmsOrderType(value = OmsOrderTypeEnum.JD_ORDER)
@Slf4j
@Service
public class JDOrderServiceImpl implements OrderService {
@Override
public String saveOrder(Supplier<OmsOrderCmd> omsOrderCmd) {
log.info("新增京东订单:{}", JSONObject.toJSONString(omsOrderCmd.get()));
return "新增京东订单成功";
}
}
@OmsOrderType(value = OmsOrderTypeEnum.WYYX_ORDER)
@Slf4j
@Service
public class WYYXOrderServiceImpl implements OrderService {
@Override
public String saveOrder(Supplier<OmsOrderCmd> omsOrderCmd) {
log.info("新增网易严选订单:{}", JSONObject.toJSONString(omsOrderCmd.get()));
return "新增网易严选订单成功";
}
}
- 通过controller来看下效果
@PostMapping("/save/order")
public CommonResp saveOrder(@RequestParam("code") String code, @RequestBody OmsOrderCmd omsOrderCmd){
OrderService orderService = OrderTypeServiceMap.SERVICE_HAND_MAP.get(OmsOrderTypeEnum.queryOmsOrderTypeEnumByCode(code));
if (orderService != null){
return CommonResp.ok(orderService.saveOrder(()->omsOrderCmd));
}
return CommonResp.error(RespCode.BAD_REQUEST,ErrorBody.build(new Throwable("非法订单类型")));
}
当code等于JD的时候,请求就会进到京东的Service实现类了,当code等于WYYX的时候就会进入网易严选的Service实现类了,这样做的好处是代码结构清晰易懂!