利用工厂和策略模式解决接口选择与多重if-else问题
在软件开发中,我们经常会遇到需要根据不同的条件选择不同实现的情况。传统的if-else或switch-case方式虽然直观,但随着业务逻辑复杂度的增加,会导致代码难以维护和扩展。工厂模式和策略模式的组合可以优雅地解决这个问题。
问题场景
假设我们有一个支付系统,需要根据不同的支付方式(支付宝、微信、银行卡等)调用不同的支付接口。传统实现可能是:- public void pay(String paymentType, BigDecimal amount) {
- if ("alipay".equals(paymentType)) {
- // 调用支付宝支付逻辑
- } else if ("wechat".equals(paymentType)) {
- // 调用微信支付逻辑
- } else if ("bank".equals(paymentType)) {
- // 调用银行卡支付逻辑
- }
- // 更多支付方式...
- }
复制代码 这种实现方式存在以下问题:
- 违反开闭原则 - 新增支付方式需要修改原有代码
- 代码臃肿 - 随着支付方式增加,方法会越来越长
- 难以维护 - 所有逻辑集中在一个方法中
解决方案:工厂 + 策略模式
1. 定义策略接口
- public interface PaymentStrategy {
- void pay(BigDecimal amount);
- }
复制代码 2. 实现具体策略类
- public class AlipayStrategy implements PaymentStrategy {
- @Override
- public void pay(BigDecimal amount) {
- // 支付宝支付具体实现
- System.out.println("使用支付宝支付:" + amount);
- }
- }
- public class WechatPayStrategy implements PaymentStrategy {
- @Override
- public void pay(BigDecimal amount) {
- // 微信支付具体实现
- System.out.println("使用微信支付:" + amount);
- }
- }
- public class BankCardStrategy implements PaymentStrategy {
- @Override
- public void pay(BigDecimal amount) {
- // 银行卡支付具体实现
- System.out.println("使用银行卡支付:" + amount);
- }
- }
复制代码 3. 创建策略工厂
- public class PaymentStrategyFactory {
- private static final Map<String, PaymentStrategy> strategies = new HashMap<>();
-
- static {
- strategies.put("alipay", new AlipayStrategy());
- strategies.put("wechat", new WechatPayStrategy());
- strategies.put("bank", new BankCardStrategy());
- }
-
- public static PaymentStrategy getStrategy(String paymentType) {
- if (paymentType == null || paymentType.isEmpty()) {
- throw new IllegalArgumentException("支付类型不能为空");
- }
- PaymentStrategy strategy = strategies.get(paymentType);
- if (strategy == null) {
- throw new IllegalArgumentException("不支持的支付类型: " + paymentType);
- }
- return strategy;
- }
- }
复制代码 4. 使用策略模式
- public class PaymentService {
- public void pay(String paymentType, BigDecimal amount) {
- PaymentStrategy strategy = PaymentStrategyFactory.getStrategy(paymentType);
- strategy.pay(amount);
- }
- }
复制代码 优势分析
- 符合开闭原则:新增支付方式只需添加新的策略类并在工厂中注册,无需修改现有代码
- 代码清晰:每个支付方式的逻辑封装在各自的类中
- 易于维护:支付逻辑分散到各个策略类,降低复杂度
- 可扩展性强:可以轻松添加新的支付方式
- 便于测试:每个策略类可以单独测试
进阶优化
使用Spring框架的依赖注入
如果使用Spring框架,可以进一步优化:- @Service
- public class PaymentStrategyFactory {
- @Autowired
- private Map<String, PaymentStrategy> strategies;
-
- public PaymentStrategy getStrategy(String paymentType) {
- PaymentStrategy strategy = strategies.get(paymentType);
- if (strategy == null) {
- throw new IllegalArgumentException("不支持的支付类型: " + paymentType);
- }
- return strategy;
- }
- }
- // 策略实现类添加@Component注解
- @Component("alipay")
- public class AlipayStrategy implements PaymentStrategy {
- // 实现
- }
- @Component("wechat")
- public class WechatPayStrategy implements PaymentStrategy {
- // 实现
- }
复制代码 结合枚举类型
- public enum PaymentType {
- ALIPAY("alipay", "支付宝支付"),
- WECHAT("wechat", "微信支付"),
- BANK("bank", "银行卡支付");
-
- private String code;
- private String desc;
-
- // 构造方法、getter等
- }
复制代码 然后在工厂中使用枚举值作为key,提高类型安全性。
总结
工厂模式和策略模式的组合是解决条件分支过多问题的经典方案。它将选择逻辑与业务逻辑分离,使系统更加灵活、可扩展。在实际开发中,可以根据项目具体情况选择适合的实现方式,结合框架特性进一步优化。
来源:程序园用户自行投稿发布,如果侵权,请联系站长删除
免责声明:如果侵犯了您的权益,请联系站长,我们会及时删除侵权内容,谢谢合作! |