当前位置：首页 > backend >正文

Spring 策略模式实现

backend 2025/7/26 11:58:28

Spring 策略模式实现：工厂方法与自动注入详解

1. 背景介绍

在复杂的业务系统中,我们常常需要根据不同的场景选择不同的处理策略。本文将详细介绍在 Spring 框架中实现策略模式的两种主要方法。

2. 方案一: 手动注册工厂模式

2.1 定义工厂类

@Component
public class CalculateHandlerFactory implements InitializingBean, ApplicationContextAware {private ApplicationContext applicationContext;// 存储不同策略的映射private static final Map<String, CalculateHandler> HANDLER_MAP = new HashMap<>();// 初始化方法,自动注册所有处理器@Overridepublic void afterPropertiesSet() throws Exception {applicationContext.getBeansOfType(CalculateHandler.class).values().forEach(v -> HANDLER_MAP.putIfAbsent(v.getType(), v));}// 设置 ApplicationContext@Overridepublic void setApplicationContext(ApplicationContext applicationContext) throws BeansException {this.applicationContext = applicationContext;}// 根据类型获取对应的处理器public CalculateHandler getCalculateHandler(String calculateType) {return HANDLER_MAP.getOrDefault(calculateType, null);}
}

2.2 实现步骤

实现 InitializingBean 接口,在 Bean 初始化时自动注册处理器
实现 ApplicationContextAware 接口,获取 Spring 应用上下文
使用 Map 存储不同类型的处理器
提供根据类型获取处理器的方法

2.3 优点

对注册过程有精细化控制
可以添加自定义的注册逻辑
灵活性高

3. 方案二: 自动注入策略模式

3.1 定义策略接口

public interface CalculateHandler {// 获取处理类型String getType();// 具体计算方法double calculate(double a, double b);
}

3.2 实现具体策略

@Component
public class AddCalculateHandler implements CalculateHandler {@Overridepublic String getType() {return "add";}@Overridepublic double calculate(double a, double b) {return a + b;}
}@Component
public class SubtractCalculateHandler implements CalculateHandler {@Overridepublic String getType() {return "subtract";}@Overridepublic double calculate(double a, double b) {return a - b;}
}

3.3 策略管理服务

@Service
public class CalculateService {@Autowiredprivate Map<String, CalculateHandler> calculateHandlerMap;public double calculate(String type, double a, double b) {CalculateHandler handler = calculateHandlerMap.get(type);if (handler == null) {throw new IllegalArgumentException("未找到对应的计算处理器");}return handler.calculate(a, b);}
}

3.4 工作原理详解

3.4.1 自动注册机制

当使用 @Autowired Map<String, Interface> 时,Spring 会:

扫描所有实现指定接口的 Bean
使用 Bean 名称作为 Map 的 Key
使用 Bean 实例作为 Map 的 Value

3.4.2 Bean 名称规则

默认使用类名首字母小写作为 Bean 名称
可以通过 @Component("customName") 自定义 Bean 名称

4. 高级用法

4.1 按优先级排序

@Service
public class CalculateService {@Autowiredprivate Map<String, CalculateHandler> calculateHandlerMap;// 按照 @Order 注解排序@Autowiredprivate List<CalculateHandler> calculateHandlerList;
}

4.2 自定义 Key 获取

@Service
public class CalculateService {@Autowiredprivate Map<String, CalculateHandler> calculateHandlerMap;// 使用自定义方法获取 Keypublic double calculate(String type, double a, double b) {CalculateHandler handler = calculateHandlerMap.values().stream().filter(h -> h.getType().equals(type)).findFirst().orElseThrow(() -> new IllegalArgumentException("未找到处理器"));return handler.calculate(a, b);}
}