知道策略模式！但不会在项目里使用？

前言

在学开发的第二年就开始听说要想代码写得好，一定要会设计模式。于是就兴致冲冲的啃了《Head First 设计模式》，看完之后对于策略模式映像很深刻，觉得这个模式好，易上手，应用广，我又能优化一波代码了(装波逼了)，于是兴致冲冲的打开了我的 IDEA，开整！！！

策略模式初体验（错误示范）

在讲诉我的策略模式首秀前，我们先回顾下策略模式的基本概念。

策略模式

意图：定义一系列的算法,把它们一个个封装起来, 并且使它们可相互替换。
主要解决：在有多种算法相似的情况下，使用 if...else 所带来的复杂和难以维护。
何时使用：一个系统有许多许多类，而区分它们的只是他们直接的行为。

简单的来说当做某个事情有多个方式的时候，可以抽象为接口，然后每个实现是一种解决方式，由调用方来选择不同的实现方式。

理解了后我开始对我们的代码进行了重构，当时我第一家公司有这样一段代码，大概是这个意思（时间长了，我凭记忆重写的）。

有这样一个抽奖的方法，我们后台控制中奖率，不同的时候我们会调整不同的中奖策略。

public class NumStrategy {

    enum RandomEnum{
        /**
         * 平均策略
         */
        AVERAGE,
        /**
         * 80%的几率中奖
         */
        RANDOM28;
    }

    /**
     * 抽奖方法，根据不同的策略进行抽奖
     * @param randomEnum
     * @return true：代表中奖  false：代表没中奖
     */
    public boolean luckDraw(RandomEnum randomEnum){
        if(randomEnum.equals(RandomEnum.AVERAGE)){
            Random random = new Random();
            int num = random.nextInt(100);
            return num >= 50;
        }else if(randomEnum.equals(RandomEnum.RANDOM28)){
            Random random = new Random();
            int num = random.nextInt(100);
            return num >= 20;
        }
        return false;
    }
}

我一看，这不就是妥妥的策略模式吗？开搞。

一顿改造之后变成了这样：

public class NumStrategy2 {

    enum RandomEnum{
        /**
         * 平均策略
         */
        AVERAGE,
        /**
         * 80%的几率中奖
         */
        RANDOM28;
    }

    /**
     * 抽奖方法，根据不同的策略进行抽奖
     * @param randomEnum
     * @return true：代表中奖  false：代表没中奖
     */
    public boolean luckDraw(RandomEnum randomEnum){
        if(randomEnum.equals(RandomEnum.AVERAGE)){
           return new AverageStrategy().luckDraw();
        }else if(randomEnum.equals(RandomEnum.RANDOM28)){
            return new Random28Strategy().luckDraw();
        }
        return false;
    }

    interface LuckDrawStrategy{
        boolean luckDraw();
    }

    class AverageStrategy implements LuckDrawStrategy{

        @Override
        public boolean luckDraw() {
            Random random = new Random();
            int num = random.nextInt(100);
            return num >= 50;
        }
    }


    class Random28Strategy implements LuckDrawStrategy{

        @Override
        public boolean luckDraw() {
            Random random = new Random();
            int num = random.nextInt(100);
            return num >= 20;
        }
    }

}

改造完成之后我满意的提交了代码，但是在组长 review 的时候给我又改了回来。说你整这么多类干嘛？我理直气壮的说我这是用策略模式优化代码。他说没必要，先改回去吧。

我愤愤的接受了，但心里想着：哎，你连策略模式都不懂？

经过这么多年，我开始理解我当时的做法其实不对，本来很简单的代码，而且里面的逻辑不会有变动，其实不需要抽象出来。我的改动有过度设计之嫌。把原来的 30 行代码搞成了 80 行

一报还一报，这几年我见过太多次当年的我这样写代码的了。

即为了用设计模式而用设计模式。而忘了设计模式的初衷是为了代码更易理解，更可靠，更易维护。

甚至还见过有人学了策略模式后说要把项目里所有的 if/else 都安排上策略模式。

梅开二度

又过了一年多，在一次面试的时候，也有着关于策略模式的讨论。

【面试官】问：你说你用过策略模式，请问你为什么用它？

【我】：为了抽离各个不同实现逻辑，优化 if/else，使代码更简单易懂

【面试官】：你具体说说，怎么去掉的 if/else

【我】：内心 OS（背的知识点，我也好久没用了啊）。我硬着头皮说，我可以使用工厂模式+策略模式来做。

【面试官】：那你工厂模式的那里不是也要用 if/else 判断吗？

【我】：。。。额。唔。。。那确实还是要用到 if/else

把我问住了，我支支吾吾的回答确实还是要 if/else 来判断一次，只不过把判断移到了工厂模式里面去了。

我下来后又去实践了下，想着放在 map 里行不行呢？

public class NumStrategy3 {

    enum RandomEnum{
        /**
         * 平均策略
         */
        AVERAGE,
        /**
         * 80%的几率中奖
         */
        RANDOM28;
    }

    static Map<RandomEnum,LuckDrawStrategy> map = new HashMap<>();

    static{
        map.put(RandomEnum.RANDOM28,new Random28Strategy());
        map.put(RandomEnum.AVERAGE,new AverageStrategy());
    }

    /**
     * 抽奖方法，根据不同的策略进行抽奖
     * @param randomEnum
     * @return true：代表中奖  false：代表没中奖
     */
    public boolean luckDraw(RandomEnum randomEnum){
        LuckDrawStrategy luckDrawStrategy = map.get(randomEnum);
        return luckDrawStrategy.luckDraw();
    }

    interface LuckDrawStrategy{
        boolean luckDraw();
    }

    static class AverageStrategy implements LuckDrawStrategy{

        @Override
        public boolean luckDraw() {
            Random random = new Random();
            int num = random.nextInt(100);
            return num >= 50;
        }
    }


    static class Random28Strategy implements LuckDrawStrategy{

        @Override
        public boolean luckDraw() {
            Random random = new Random();
            int num = random.nextInt(100);
            return num >= 20;
        }
    }

}

终于是解决了 if/else 的情况，不过这样很短的 if/else，里面逻辑不怎么变动时，我个人是不建议用策略模式，这里只是示例。

推荐用法

又过了几年，当初的菜鸟也成长为了一个老鸟。

当时项目里有这样一个代码：

下面的代码我进行了一些简化，我们有一个功能，对页面上的指标进行计算，不同的指标对应不同的计算方法。页面上指标一期做 4 个，后续会做到十几个。


public interface TransferService {

    String transfer();
}


@Service
public class SearchTransformService {

    @Autowired
    private UserTransferService userTransferService;

    @Autowired
    private AgeTransferService ageTransferService;

    @Autowired
    private InterestTransferService interestTransferService;

    /**
     * 根据不同的编码进行转换
     * @param code
     * @return
     */
    public String transform(String code){
        if(code.equals("user")){
            return userTransferService.transfer();
        }else if(code.equals("age")){
            return ageTransferService.transfer();
        }else if(code.equals("interest")){
            return interestTransferService.transfer();
        }
        return "";
    }
}

可以看到这样的业务场景下，这样的写法 if/else 就会很长，后续十几个的情况下就很难维护。另外 code 使用的是魔数，也是不好的一种写法。我对此进行了优化如下：

先将 code 用枚举定义

   enum CodeEnum {
    USER("user"),
    AGE("age"),
    INTEREST("interest"),
    ;

    private String code;

    public String getCode() {
        return code;
    }

    CodeEnum(String code) {
        this.code = code;
    }

    private static final Map<String, CodeEnum> map = Arrays.stream(CodeEnum.values()).collect(Collectors.toMap(CodeEnum::getCode, Function.identity()));


    public CodeEnum of(String code) {
        return map.get(code);
    }
}

原有的接口上增加一个 transCode 方法，每个实现需要声明是对应哪个编码的实现

public interface TransferService {

    String transfer();

    CodeEnum transCode();
}

@Service
public class AgeTransferService implements TransferService {
    @Override
    public String transfer() {
        return null;
    }

    @Override
    public CodeEnum transCode() {
        return CodeEnum.AGE;
    }
}

使用 map 存储编码对应的实现类的关联关系,以此来获取对应的转换器实现类

@Service
public class SearchTransformService implements InitializingBean {

    @Autowired
    private List<TransferService> transferServiceList;

    private Map<CodeEnum, TransferService> transferServiceMap;

    @Override
    // 项目启动时将实现类放入到map中去
    public void afterPropertiesSet() throws Exception {
        transferServiceMap = transferServiceList.stream().collect(Collectors.toMap(TransferService::transCode, Function.identity()));
    }
    /**
     * 根据不同的编码进行转换
     * @param code
     * @return
     */
    public String transform(String code){
        TransferService transferService = transferServiceMap.get(CodeEnum.of(code));
        Assert.notNull(transferService,"找不到对应的转换器");
        return transferService.transfer();
    }

}

重构后是不是就很简洁了呢？如果后续新增新的编码转换器，只需要先在枚举里定义，然后新增实现类实现方法就行了，不需要对关心是怎么调用的，只关心具体的实现逻辑，降低了维护成本。

这才是策略模式的真正应用吧。不要再乱用了，哈哈哈。

本文正在参加「金石计划 . 瓜分6万现金大奖」