SpringBoot結合策略模式實戰套路

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到新套路了嗎

接下來我爭取每篇文章下都分享一套學習視頻，希望你們喜歡！！

龍果學院 秒殺系統企業級實戰應用（真實工業界案例）視頻

歡迎關注公眾號，一起學習進步