看到同事的代码,想到多年以前自己刚开始工作的时候写的代码,即使有再多的需求,再多的业务逻辑,都是满屏的if-else解决的,全然忘记什么叫做“面向对象编程”,但是写的多了都忘记了哪里是头,哪里是尾,但是当时并没有感觉到有多别扭,等慢慢的工作之后,回头再去看之前的代码,我艹,这是哪个**写的代码。

可能刚入门的同行可能会忽略一点,if-else可能是面向过程的实现。

那么如何解决掉满屏的if-else呢?让代码看起来变的更加的优雅、简洁呢?

我一般都是使用工厂模式 + 自定义注解 + 反射 + 枚举 + Map集合来实现一个策略模式,但也不是只要出现if-else就使用策略模式,当if-else => 4 , 我就会考虑使用这一套策略模式。 

那么具体如何使用呢?我们通过一段业务来说明。

    比如现在某一个平台有不同等级的用户,分别分为普通用户、黄金用户、铂金用户、钻石用户,需要通过不同的等级的用户查询相关的数据,比如优惠力度等。

一、首先定义一个自定义注解 :StrategyType,用于标记具体的策略类。

@Retention(RetentionPolicy.RUNTIME)@Target(ElementType.TYPE)public @interface StrategyType {    /**     * 策略类型对应的具体的策略的编码     * @return 策略类型对应的具体的策略的编码     */    String code() ;
    /**     * 策略类型枚举编码     * @return 枚举类型编码     */    StrategyTypeEnum type() ;
}

二、然后定义一个枚举类 :StrategyTypeEnum,用于列举所有支持的策略。

public enum StrategyTypeEnum {
    USER_STRATEGY_TYPE("userStrategyType" , "用户类型策略枚举编码");    /*     * @TODO 如果有其它策略也可以加,比如根据不同的角色加载不同的数据 RoleStrategyType     * ROLE_STRATEGY_TYPE("roleStrategyType" ," 角色策略枚举编码")     */
    /**     * 策略类型编码     */    private final String strategyTypeCode;
    /**     * 策略类型描述     */    private final String strategyTypeDesc;
    /**     *     * @param strategyTypeCode 策略类型编码     * @param strategyTypeDesc 策略类型描述     */    StrategyTypeEnum(String strategyTypeCode, String strategyTypeDesc) {        this.strategyTypeCode = strategyTypeCode;        this.strategyTypeDesc = strategyTypeDesc;    }
    public String getStrategyTypeCode() {        return strategyTypeCode;    }
    public String getStrategyTypeDesc() {        return strategyTypeDesc;    }
    /**     * 根据策略类型编码获取枚举     * @param strategyTypeCode 策略类型编码     * @return StrategyTypeEnum 策略类型枚举     * @throws NullPointerException     */    public static StrategyTypeEnum getUserTypeEnumByCode(String strategyTypeCode) throws NullPointerException {        if(strategyTypeCode.isEmpty()){            throw new NullPointerException("策略类型编码不能为空");        }        for (StrategyTypeEnum strategyTypeEnum : StrategyTypeEnum.values()) {            if(strategyTypeEnum.strategyTypeCode.equals(strategyTypeCode)){                return strategyTypeEnum;            }        }
        return null;    }
}

三、再然后、定义一个策略类型编码常量类 :StrategyTypeConstant,用来放不同策略对应的具体的策略的编码。

public interface StrategyTypeConstant {
    /**     * 普通用户     */    public static final String USER_TYPE_NORMAL = "NORMAL";    /**     * 普通用户     */    public static final String USER_TYPE_GOLD = "GOLD";    /**     * 普通用户     */    public static final String USER_TYPE_PLATINUM = "PLATINUM";    /**     * 普通用户     */    public static final String USER_TYPE_DIAMOND = "DIAMOND";
    /*     * @TODO 可以进行扩展 , 其它策略的编码     */
}

四、接着,我们创建一个策略工厂StrategyFactory,负责根据注解值获取对应的策略实例。

@Componentpublic class StrategyTypeFactory implements ApplicationContextAware {
    private final Map<String, UserDataHandler> handlerMap = new HashMap<>();
    public UserDataHandler getUserDataHandler(String userTypeCode) {        return handlerMap.get(userTypeCode);    }
    /**     * Set the ApplicationContext that this object runs in.     * Normally this call will be used to initialize the object.     * <p>Invoked after population of normal bean properties but before an init callback such     * as {@link InitializingBean#afterPropertiesSet()}     *     * @param applicationContext the ApplicationContext object to be used by this object     * @throws BeansException              if thrown by application context methods     * @see BeanInitializationException     */    @Override    public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {
        // 添加用户数据处理策略        addUserDataHandlerStrategy(applicationContext);        // @TODO 可以进行扩展 , 添加其它策略....    }
    private void addUserDataHandlerStrategy(ApplicationContext applicationContext){
        Map<String, Object> beansWithAnnotation = applicationContext.getBeansWithAnnotation(StrategyType.class);        beansWithAnnotation.entrySet().iterator().forEachRemaining(entrySet ->{            StrategyType annotation = entrySet.getValue().getClass().getAnnotation(StrategyType.class);            // 获取到用户类型枚举            String userTypeCode = annotation.code();            // 获取到策略类型枚举            String strategyTypeCode = annotation.type().getStrategyTypeCode();            // 策略的类型编码以及用户类型编码拼接为key , 表示某某策略类型的某某编码对应的策略            // key : strategyTypeCode + ":" +  userTypeCode表示 : 用户类型策略的普通用户            // value : 对应的普通用户的处理类            this.handlerMap.put(strategyTypeCode + ":" +  userTypeCode , (UserDataHandler) entrySet.getValue());        });    }}

五、继续、定义一个接口:UserDataHandler,所有具体用户策略类都要实现该接口。

public interface UserDataHandler {
    /**     * 处理用户数据     */    Object handleUserData();}

六、然后,我们创建四个具体的用户策略类NormalUserStrategy、GoldUserStrategy、PlatinumUserStrategy、DiamondUserStrategy,并使用@StrategyType注解标记每个策略类的类型:

@StrategyType(code = StrategyTypeConstant.USER_TYPE_DIAMOND , type = StrategyTypeEnum.USER_STRATEGY_TYPE)@Servicepublic class DiamondUserDataHandler implements UserDataHandler{    /**     * 处理用户数据     *     * @return 处理结果     */    @Override    public Object handleUserData() {      // @TODO 对应的业务逻辑        System.out.println("钻石用户业务逻辑");        return "钻石用户业务逻辑";    }}
@StrategyType(code = StrategyTypeConstant.USER_TYPE_GOLD , type = StrategyTypeEnum.USER_STRATEGY_TYPE)@Servicepublic class GoldUserDataHandler implements UserDataHandler{    /**     * 处理用户数据     *     * @return 处理结果     */    @Override    public Object handleUserData() {    // @TODO 对应的业务逻辑        System.out.println("黄金用户业务逻辑");        return "黄金用户业务逻辑";    }}
@StrategyType(code = StrategyTypeConstant.USER_TYPE_NORMAL , type = StrategyTypeEnum.USER_STRATEGY_TYPE)@Servicepublic class NormalUserDataHandler implements UserDataHandler{    /**     * 处理用户数据     *     * @return 处理结果     */    @Override    public Object handleUserData() {      // @TODO 对应的业务逻辑        System.out.println("普通用户业务逻辑");        return "普通用户业务逻辑";    }}
@StrategyType(code = StrategyTypeConstant.USER_TYPE_PLATINUM , type = StrategyTypeEnum.USER_STRATEGY_TYPE)@Servicepublic class PlatinumUserDataHandler implements UserDataHandler{    /**     * 处理用户数据     *     * @return 处理结果     */    @Override    public Object handleUserData() {      // @TODO 对应的业务逻辑        System.out.println("铂金用户业务逻辑");        return "铂金用户业务逻辑";    }}

七、测试:

@RestController@RequestMapping("sysUser")public class UserDataController {
    @Resource    private StrategyTypeFactory strategyTypeFactory;
    @RequestMapping("userDataHandle")    public Result<Object> userDataHandle(String userTypeCode){        // 根据不同的用户类型执行不同的策略        UserDataHandler userDataHandler = strategyTypeFactory.getUserDataHandler(StrategyTypeEnum.USER_STRATEGY_TYPE.getStrategyTypeCode() + ":" + userTypeCode);        if(userDataHandler == null){            return Result.error(HttpStatusCodeEnum.ERROR);        }        return Result.success(userDataHandler.handleUserData());    }}
12-13 23:48