顛覆認知！我從 Spring 源碼里學到的一個隱藏高手級技巧
在日常開發中，我們經常會遇到這樣的場景：根據不同的條件分支執行不同的邏輯。最常見的寫法自然是大量的 if-else 或 switch-case 語句。這種寫法雖然直觀，但隨著業務復雜度提升，代碼的可讀性、可維護性會迅速下降，成為技術債的溫床。
最近我在閱讀 Spring Framework 源碼時，無意中發現它在多個核心模塊中大量使用了一種非常優雅的設計思路 —— 策略映射（Strategy Map），通過它，Spring 實現了對條件分支邏輯的解耦和模塊化。受到啟發后，我將其應用到自己的業務代碼中，效果立竿見影。
本文將通過一個實際案例帶你了解策略映射的使用方式，并展示如何結合注解與工廠模式自動注冊策略實現類，徹底擺脫臃腫邏輯。
場景背景：文件處理服務
我們假設有一個文件上傳處理服務，支持三種文件類型：CSV、JSON 和 XML。用戶上傳文件后，系統會根據文件類型調用不同的解析方法。
原始實現（if-else 地獄）
public void processFile(FileType fileType, InputStream inputStream) {
if (fileType == FileType.CSV) {
processCsv(inputStream);
} else if (fileType == FileType.JSON) {
processJson(inputStream);
} else if (fileType == FileType.XML) {
processXml(inputStream);
} else {
throw new IllegalArgumentException("Unsupported file type");
}
}問題顯而易見：
- 職責不清：所有邏輯集中在一個方法中；
- 擴展困難：增加新類型需改動已有代碼；
- 測試繁瑣：不能獨立測試某種處理邏輯。
策略模式改寫
Step 1：定義文件類型枚舉
public enum FileType {
CSV, JSON, XML
}Step 2：定義統一接口
public interface FileProcessor {
void process(InputStream inputStream);
}方法一：基本策略映射（初級）
我們可以用 Spring 自動注入 + 手動注冊策略類：
@Component
public class CsvFileProcessor implements FileProcessor {
@Override
public void process(InputStream inputStream) {
// 解析 CSV
}
}
@Component
public class JsonFileProcessor implements FileProcessor {
@Override
public void process(InputStream inputStream) {
// 解析 JSON
}
}手動維護 Map<FileType, FileProcessor>：
@Service
public class FileProcessingService {
private final Map<FileType, FileProcessor> processorMap = new EnumMap<>(FileType.class);
@Autowired
public FileProcessingService(List<FileProcessor> processors) {
for (FileProcessor processor : processors) {
if (processor instanceof CsvFileProcessor) {
processorMap.put(FileType.CSV, processor);
} else if (processor instanceof JsonFileProcessor) {
processorMap.put(FileType.JSON, processor);
}
// ...
}
}
public void processFile(FileType fileType, InputStream inputStream) {
FileProcessor processor = processorMap.get(fileType);
if (processor == null) {
throw new IllegalArgumentException("Unsupported file type: " + fileType);
}
processor.process(inputStream);
}
}雖然比 if-else 好一些，但仍需手動維護映射，擴展成本仍在。
方法二：注解式自動注冊（高級推薦）
目標：
讓每個處理器類用注解聲明其支持的文件類型，然后由統一的“策略工廠”自動注冊進映射容器，徹底做到零修改、自動發現、自動注冊。
定義策略注解
@Target(ElementType.TYPE)
@Retention(RetentionPolicy.RUNTIME)
@Documented
public @interface FileTypeHandler {
FileType value();
}修改處理器實現類
@Component
@FileTypeHandler(FileType.CSV)
public class CsvFileProcessor implements FileProcessor {
@Override
public void process(InputStream inputStream) {
System.out.println("處理 CSV 文件");
}
}
@Component
@FileTypeHandler(FileType.JSON)
public class JsonFileProcessor implements FileProcessor {
@Override
public void process(InputStream inputStream) {
System.out.println("處理 JSON 文件");
}
}編寫策略工廠
@Component
public class FileProcessorFactory implements ApplicationContextAware {
private final Map<FileType, FileProcessor> processorMap = new EnumMap<>(FileType.class);
@Override
public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {
Map<String, FileProcessor> beans = applicationContext.getBeansOfType(FileProcessor.class);
for (FileProcessor processor : beans.values()) {
FileTypeHandler annotation = processor.getClass().getAnnotation(FileTypeHandler.class);
if (annotation != null) {
processorMap.put(annotation.value(), processor);
}
}
}
public FileProcessor getProcessor(FileType type) {
FileProcessor processor = processorMap.get(type);
if (processor == null) {
throw new IllegalArgumentException("未支持的文件類型: " + type);
}
return processor;
}
}服務類調用更簡潔
@Service
public class FileProcessingService {
private final FileProcessorFactory factory;
@Autowired
public FileProcessingService(FileProcessorFactory factory) {
this.factory = factory;
}
public void processFile(FileType type, InputStream inputStream) {
factory.getProcessor(type).process(inputStream);
}
}總結：策略映射的進階之道
實現方式 | 可擴展性 | 可讀性 | 解耦程度 | 推薦程度 |
| 低 | 差 | 耦合嚴重 | 不推薦 |
手動構建 | 中 | 一般 | 部分解耦 | 可用 |
注解 + 工廠自動注冊 | 高 | 優 | 完全解耦 | 強烈推薦 |
策略映射 + 注解注冊，是應對多分支邏輯的一種現代、模塊化、可維護的架構方式。Spring 中諸如 HandlerAdapter、Converter、MessageConverter 等核心機制大量應用了這一思路。
寫在最后
當你的代碼中開始出現 3 層以上的 if-else，就是引入策略映射的最佳時機。它不只是設計模式里的教材，更是讓代碼結構清晰可擴展的“生產力工具”。
與其維護一個千瘡百孔的 if 邏輯，不如優雅地構建一個策略工廠。
