首先,我们定义一个策略接口PaymentStrategy:
public interface PaymentStrategy {
void pay(double amount);
}
然后,我们定义几个具体的实现类,实现不同的支付逻辑:
@Component
public class AlipayStrategy implements PaymentStrategy {
@Override
public void pay(double amount) {
// 实现支付宝支付逻辑
}
}
@Component
public class WechatpayStrategy implements PaymentStrategy {
@Override
public void pay(double amount) {
// 实现微信支付逻辑
}
}
@Component
public class UnionpayStrategy implements PaymentStrategy {
@Override
public void pay(double amount) {
// 实现银联支付逻辑
}
}
接下来,我们定义一个支付服务类PaymentService,它依赖于PaymentStrategy,并使用策略模式来实现不同的支付策略:
@Service
public class PaymentService {
private Map<String, PaymentStrategy> paymentStrategyMap;
public PaymentService(List<PaymentStrategy> paymentStrategies) {
this.paymentStrategyMap = paymentStrategies.stream()
.collect(Collectors.toMap(PaymentStrategy::getName, Function.identity()));
}
public void pay(String paymentMethod, double amount) {
PaymentStrategy paymentStrategy = paymentStrategyMap.get(paymentMethod);
if (paymentStrategy == null) {
throw new UnsupportedOperationException(String.format("Unsupported payment method: %s", paymentMethod));
}
paymentStrategy.pay(amount);
}
}
在这个示例中,我们使用了Spring Boot的依赖注入功能,将所有实现了PaymentStrategy接口的类注入到了PaymentService的构造函数中。然后,我们在pay()方法中通过传入的支付方式来选择对应的支付策略实现。
最后,我们可以在一个控制器中使用PaymentService来处理不同支付方式的请求:
@RestController
public class PaymentController {
@Autowired
private PaymentService paymentService;
@PostMapping("/payment")
public void pay(@RequestParam("paymentMethod") String paymentMethod,
@RequestParam("amount") double amount) {
paymentService.pay(paymentMethod, amount);
}
}
在上面这个示例中,我使用@Autowired将PaymentService注入到了控制器中,并在pay()方法中根据请求参数选择不同的支付方式。
这就是使用策略模式在Spring Boot中处理不同业务需求的完整示例;