浅谈用SpringBoot实现策略模式

问题的提出

阅读别人代码的时候最讨厌遇到的就是大段大段的if-else分支语句，一般来说读到下面的时候就忘了上面在判断什么了。很多资料上都会讲到使用策略模式来改进这种代码逻辑。

策略模式的类图如下:

浅谈用SpringBoot实现策略模式

只需要按照这个图写代码就可以了。

策略模式代码的实现

借助Spring框架我们能够轻松的实现策略模式。

举一个简单的例子，我们去咖啡店买咖啡的时候，会根据自己的喜好和胃容量选择大小杯。那么我们就要实现一个CoffeeStategy：

				?

									package com.example.demo.strategy;

									public interface CoffeeStrategy {

									    void offer();

									}

接下来就是各种具体策略的实现了，以中杯咖啡为例：

				?

									package com.example.demo.strategy;

									import lombok.extern.slf4j.Slf4j;

									import org.springframework.stereotype.Component;

									@Component("MID")

									@Slf4j

									public class MidCoffee implements CoffeeStrategy {

									    @Override

									    public void offer() {

									        log.info("你的中杯咖啡");

									    }

									}

用Component注解给这个类起一个名字叫做MID，这个在后面的应用上下文中有起效。现在就开始定义应用上下文类：

				?

									package com.example.demo.strategy;

									import org.springframework.beans.factory.annotation.Autowired;

									import org.springframework.stereotype.Service;

									import java.util.Map;

									@Service

									public class CoffeeContext {

									    @Autowired

									    private Map<String, CoffeeStrategy> coffeeStrategyMap;

									    public void getCoffee(String size) {

									        this.coffeeStrategyMap.get(size).offer();

									    }

									}

因为是使用了Spring框架，所有的Bean都被Spring自行管理，启动之后，Map中会有两个元素：{"MID":MidCoffee}和{"LARGE":LargeCoffee}。在具体的业务逻辑中，只需要引入应用上下文类，每次使用getCoffee方法就可以了。

比如这个Controller方法：

				?

									@GetMapping("/get")

									    public void getCoffee(@Param("size") String size) {

									        this.coffeeContext.getCoffee(size);

									    }

请求这个接口，我们能在后台看到具体的日志内容：

2021-09-30 22:46:32.550 INFO 15628 --- [nio-8099-exec-1] com.example.demo.strategy.LargeCoffee : 您的大杯咖啡
2021-09-30 22:46:39.201 INFO 15628 --- [nio-8099-exec-7] com.example.demo.strategy.LargeCoffee : 您的大杯咖啡

进一步的思考

之前写过Component中起的名字有奇效。如果我们没有用Spring框架去实现策略模式，那么我们的代码要如何编写呢？

首先可以肯定的是策略接口和策略实现类是不需要变的。需要变的地方就是应用上下文了，因为不存在自动注入了。这段代码就会变成大致这样：

				?

									package com.example.demo.strategy;

									public class CoffeeContext {

									    CoffeeStrategy coffeeStrategy;

									    public CoffeeContext(CoffeeStrategy coffeeStrategy) {

									        this.coffeeStrategy = coffeeStrategy;

									    }

									    public void getCoffee() {

									        this.coffeeStrategy.offer();

									    }

									}

这样，在实际使用的时候，我需要先新建一个具体的实现类对象，然后将这个对象传入策略应用上下文去。这种方式怎么看着都没有Spring的实现方式优雅。

				?

									CoffeeStrategy mid = new MidCoffee();

									CoffeeContext context = new CoffeeContext(mid);

									context.getCoffee();

在我实际改造代码的过程中我发现有些策略其实是一样的，只是个别参数不同罢了。我对接的是各个业务供应商，有些供应商的接口逻辑式样的，只是URL和USERNAME不一样罢了。于是好几个策略实现类的代码重复很严重，这个时候我使用了Java8开始提供的接口default方法。这种方法的好处就是能将这种一样的逻辑提取到interface中，只要实现类不重写，那么就会默认使用default方法。

这样改造之后，我的代码又精简了很多。