位置：首页-资讯-后端开发

浅谈用SpringBoot实现策略模式

2024-04-02 19:55

短信预约 -IT技能 免费直播动态提醒

问题的提出

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

策略模式的类图如下:

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

策略模式代码的实现

借助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方法。

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