在Spring Boot项目开发中,常常会遇到根据不同的业务场景执行不同逻辑的需求,策略模式就是一种很好的设计模式来应对这种情况。同时,Spring Boot强大的依赖注入机制可以方便地将不同的策略类进行管理和调用。
1. 定义策略接口
定义一个策略接口,所有具体的策略类都将实现这个接口:
public interface BaseStrategy {void handle(String request);
}
2. 实现具体的策略
实现几个具体的策略类,例如处理订单和支付:
@Component("order")
@Slf4j
public class OrderStrategy implements BaseStrategy {@Overridepublic void handle(String request) {System.out.println("下单成功" + request);}
}@Component("pay")
@Slf4j
public class PayStrategy implements BaseStrategy {@Overridepublic void handle(String request) {System.out.println("支付成功" + request);}
}
也可以通过配置类的方式注入策略模式实现类,先注释其中的@Component注解
@Configuration
public class HandlerConfig {@Bean("order")public BaseStrategy handleOrder() {return new OrderStrategy();}@Bean("pay")public BaseStrategy handlePay() {return new PayStrategy();}
}
3. 创建策略工厂
为了方便获取不同的策略,我们可以创建一个工厂类 BaseFactory,它将通过依赖注入管理策略的实例,并提供获取策略的功能。
@Component
@Slf4j
public class BaseFactory {@Resourceprivate Map<String, BaseStrategy> handlerMap;public BaseStrategy getHandler(String type) {return handlerMap.getOrDefault(type, (request) -> {log.error("Unknown request type: " + type);});}
}
在这个类中,我们使用 @Resource 注解注入了一个 Map<String, BaseStrategy>,这个 Map 将所有注册的策略按名称存储起来。通过策略名称,我们可以从 Map 中获取到对应的策略对象。
4. 使用策略工厂处理请求
@RestController
@RequestMapping("base")
@Slf4j
public class BaseController {@Resourceprivate BaseFactory baseFactory;@GetMapping("/getHandler")public ResponseEntity<String> getHandler(@RequestParam String type, @RequestParam String message) {BaseStrategy handler = baseFactory.getHandler(type);if (handler != null) {handler.handle(message);return ResponseEntity.ok("操作已执行: " + type);} else {return ResponseEntity.badRequest().body("策略类型不存在: " + type);}}
}