乐趣区

SpringBoot结合策略模式实战套路

jiezi

1. SpringBoot 结合策略模式实战套路

1.1. 前言

我们都知道设计模式好，可以让我们的代码更具可读性，扩展性，易于维护，但大部分程序猿一开始都学过至少一遍设计模式吧，实战中不知用到了几成。接下来让我介绍一个结合 SpringBoot 的策略模式套路，让你的代码少些 if-else

1.2. 开撸

废话不多说，直接告诉你今天的核心是 @autowired，看到这个是不是很熟悉，你每天都在用，不就是自动注入 Spring 管理的 Bean 吗？但我们对它的用法很多时候就局限在 全局变量 的注入了，忘记了，它其实还可以 构造器注入，类型注入或命名注入，那么结合策略模式会绽放怎样的火花呢？跟着我的代码来看

1.2.1. 计算策略接口

/**
 * @author laoliangliang
 * @date 2019/10/28 10:10
 */
public interface CalculateStrategy {int doOperation(int num1,int num2);
}

1.2.2. 实现类

分别实现加减乘三个运算，可以看到我用了 spring 的注解@Component，也就是实例由 spring 管理了

@Component
public class AddOperation implements CalculateStrategy {
    @Override
    public int doOperation(int num1, int num2) {return num1 + num2;}

}




@Component
public class SubstractOperation implements CalculateStrategy {
    @Override
    public int doOperation(int num1, int num2) {return num1 - num2;}

}
@Component
public class MultiplyOperation implements CalculateStrategy {
    @Override
    public int doOperation(int num1, int num2) {return num1 * num2;}

}

1.2.3. 上下文

之后创建上下文管理，用于提取策略。这个上下文才是本文的重点，注意到 @autowired 注解放的位置和对应的参数列表了吗？实际上它还可以注入到 Map 和 List，Map 的 key 就是它注入时的名，List 则是存放全部实例对象

import com.google.common.base.Preconditions;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
import org.springframework.util.StringUtils;

import java.util.List;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;

/**
 * @author laoliangliang
 * @date 2019/10/28 10:14
 */
@Component
public class CalculatelOperationContext {

//    @Autowired
//    private Map<String, CalculateStrategy> strategyMap;

    private final Map<String, CalculateStrategy> strategyMap = new ConcurrentHashMap<>();

    @Autowired
    public void stragegyInteface(Map<String, CalculateStrategy> strategyMap) {this.strategyMap.clear();
        strategyMap.forEach(this.strategyMap::put);
        System.out.println(this.strategyMap);
    }


    @Autowired
    public void stragegyInteface2(List<CalculateStrategy> strategyMap) {strategyMap.forEach(System.out::println);
    }

    public CalculateStrategy strategySelect(String mode) {Preconditions.checkArgument(!StringUtils.isEmpty(mode), "不允许输入空字符串");
        return this.strategyMap.get(mode);
    }
}

打印结果：

{multiplyOperation=com.laoliang.springboot.pattern.strategy.MultiplyOperation@372ea2bc, addOperation=com.laoliang.springboot.pattern.strategy.AddOperation@4cc76301, substractOperation=com.laoliang.springboot.pattern.strategy.SubstractOperation@2f08c4b}
com.laoliang.springboot.pattern.strategy.AddOperation@4cc76301
com.laoliang.springboot.pattern.strategy.MultiplyOperation@372ea2bc
com.laoliang.springboot.pattern.strategy.SubstractOperation@2f08c4b

可以看到 Map 中 key，value 的关系，key 的默认值为类的第一个字母小写

1.2.4. 控制层

/**
 * @author laoliangliang
 * @date 2019/10/28 10:52
 */
@RestController
public class StrategyController {

    @Autowired
    private CalculatelOperationContext calculatelOperationContext;

    @RequestMapping(value = "/operation")
    public String strategySelect(@RequestParam("mode") String mode) {return String.valueOf(calculatelOperationContext.strategySelect(mode).doOperation(20, 5));
    }
}

启动 SpringBoot，浏览器调用 http://localhost:8080/operation?mode=multiplyOperation，结果 100。模式可以选择另外两个addOperation 和substractOperation

我这里就做个演示，输入参数就写固定了，可以看到我们通过上下文 calculatelOperationContext 调用其方法 strategySelect，通过 不同的调用参数 获得不同的策略，所以业务中只要可以抽象的方法都可以改写成这样的模式。

这种模式套路的 好处 就是当你要新增一种策略，比如除法，你不需要修改原来的代码，只要抽象不变，你新增一个 DivideOperation 类实现 CalculateStrategy 策略接口加上 Spring 注解即可，调用时模式修改为 divideOperation 就可以实现调用了，耦合性大大降低，不需要再改原来那一坨自己都看不下去的代码了

1.3. 总结

可以看到全文中代码量还是相对比较少的，将不同的策略用不同的类实现，且可以不用改动别的代码，这篇文章你 get 到新套路了吗

接下来我争取每篇文章下都分享一套学习视频，希望你们喜欢！！

龙果学院 秒杀系统企业级实战应用（真实工业界案例）视频

退出移动版