策略模式结构图

策略模式次要由以上三个身份组成,这里咱们就不过多及时策略模式的基础知识,默认大家曾经对策略模式曾经有了一个根底的意识。

业务需要

现有一个广告点击数据埋点上报的需要,上报的埋点数据依据点击的广告地位不同做辨别进行上报,每个广告地位的数据进行分表存储。(eg:这里大家也不用深究分表存储为什么要这么做,咱们只聊策略模式的理论利用)

代码实现

因为是实战案例,那么咱们是基于SpringBoot框架的,次要要应用的Spring的一些性能,所以大家要留神。

第一步:定义策略类

首先咱们定义一个上报的接口

public interface AdvertisingDataReported {    String advertisingDataReported(Object param);}

第二步:定义具体的策略实现类

@Servicepublic class BottomAdvertisingDataReported implements AdvertisingDataReported {        @Override    public String advertisingDataReported(Object param) {          // 具体的业务逻辑略        return null;    }}

第三步:策略管制类

因为策略模式有好多具体的具体策略实现,那么到底应用哪一个策略须要依据咱们的入参,也就是咱们业务中的广告类型进行判断,那么咱们该如何优雅的进行判断呢?

咱们先看看这种形式

public static void main(String[] args) {          String advertisingType = "1";  if (advertisingType.equals("1")) {    // 执行策略A  } else if (advertisingType.equals("2")) {    // 执行策略2  }}

这么写的大有人在,咱们这里也不探讨这些问题。咱们先看一下这么写存在哪些问题?

存在的问题:

1. 违反开闭准则,每次减少新的策略实现类,都要加一个if判断; 2. 随着策略实现类的减少,代码变的臃肿,越来越难以保护;

基于这种状况,咱们可不可以在我的项目启动的时候,将所有的策略实现类进行初始化,存储在Map当中,广告类型作为key,实现类作为Value,咱们看如下代码:

@Componentpublic class StrategyFactory implements ApplicationContextAware {    private final Map<String, AdvertisingDataReported> STRATEGY_MAP = new ConcurrentHashMap<>();    @Override    public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {          // 返回该接口所有的实现类        Map<String, AdvertisingDataReported> tempMap = applicationContext.getBeansOfType(AdvertisingDataReported.class);        tempMap.values().forEach(strategyService -> STRATEGY_MAP.put(strategyService.getClass().getName(), strategyService));    }    public <T extends AdvertisingDataReported> AdvertisingDataReported getInstance(Class<T> clazz) {        return STRATEGY_MAP.get(clazz.getName());    }}

咱们的策略管制类实现了ApplicationContextAware,这个类你能够这么了解,它能够取得ApplicationContext的上下文,因为咱们是至于SpringBoot讲这个案例的,咱们的策略类实现类都加了@Service注解注入到了Spring容器中,所以咱们能够间接从容器中,取到策略类的所有实现类。

获取到所有的策略实现类之后,咱们把类门路作为key,类的实现作为value存储到了map中,到此我过后感觉就功败垂成了。

大家感觉还存在什么问题?

咱们怎么晓得这个入参须要走哪个具体的策略类呢?还须要定义一个独自的类,来对广告类型和策略类进行映射,那这跟判断不又是同一个逻辑的吗?还得始终保护这个映射关系。

革新

如果不想独自的定义一个类对广告类型和策略类进行一一映射,那么咱们可不可以在策略类中进行解决,每个策略类实现类晓得它要解决哪种类型,这样咱们就能够把map中Key类门路的值替换为广告类型,这样就能够依据上报接口入参的广告类型,间接从Map中进行get。

具体的实现有两种,你能够自定义注解,通过加注解的形式进行辨别,也能够应用办法,那么咱们这里间接应用办法进行解决。

革新后的代码:

策略类:

public interface AdvertisingDataReported {      // 新增办法    AdvertisingTypeEnum advertisingType();    String advertisingDataReported(Object param);}

策略实现类:

@Servicepublic class BottomAdvertisingDataReported implements AdvertisingDataReported {    @Override    public AdvertisingTypeEnum advertisingType() {        return AdvertisingTypeEnum.BOTTOM;    }    @Override    public String advertisingDataReported(Object param) {        return null;    }}

策略管制类:

@Componentpublic class StrategyFactory implements ApplicationContextAware {      // Map的Key改为广告类型枚举类    private final Map<AdvertisingTypeEnum, AdvertisingDataReported> STRATEGY_MAP = new ConcurrentHashMap<>();    @Override    public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {        Map<String, AdvertisingDataReported> tempMap = applicationContext.getBeansOfType(AdvertisingDataReported.class);        tempMap.values().forEach(strategyService -> STRATEGY_MAP.put(strategyService.advertisingType(), strategyService));    }      // 依据广告类型获取相应的策略类    public <T extends AdvertisingDataReported> AdvertisingDataReported getInstance(AdvertisingTypeEnum advertisingTypeEnum) {        return STRATEGY_MAP.get(advertisingTypeEnum);    }}

广告枚举类:

public enum AdvertisingTypeEnum {    BOTTOM, TOP;    private String advertisingType;    AdvertisingTypeEnum() {}        // set get省略}

策略类的具体应用

@RestControllerpublic class AdvertisingDataReportedController {    @Resource    private StrategyFactory strategyFactory;    @RequestMapping(value = "/reported/data", method = RequestMethod.POST)    public String reportedData(AdvertisingTypeEnum advertisingTypeEnum, Object obj) {        AdvertisingDataReported dataReported = strategyFactory.getInstance(advertisingTypeEnum);        String result = dataReported.advertisingDataReported(obj);        return "SUCCESS";    }}

小小总结:

到这里咱们这个策略模式的案例就算完结了,有几个问题不晓得大家有没有纳闷,为什么我要用Object作为办法的入参,咱们这种案例中,如同每个策略类的入参如同都是一样的,然而也有可能呈现同一个策略的实现类,然而入参齐全可能不雷同,那么这个时候,咱们就能够通过传入Object的形式,在办法外部进行转换,当然了,如果这样你嫌策略办法太死板了,那么你也能够在办法上退出泛型,具体转换为什么类型,通过调用者传入泛型来转换。

通过这样一番革新之后,方才咱们遇到的两个问题也都通通不是问题了,咱们想要新增一个策略实现类,只须要实现定义的策略类即可,无需减少额定的任何代码。