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

以上就是本文的全部內容，希望對大家的學習有所幫助，也希望大家多多支援我們。