基于自定义注解实现策略模式

什么是策略模式?

-在策略模式(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实现类了,这样做的好处是代码结构清晰易懂!

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容