利用工厂和策略模式解决接口选择与多重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,提高类型安全性。
总结
工厂模式和策略模式的组合是解决条件分支过多问题的经典方案。它将选择逻辑与业务逻辑分离,使系统更加灵活、可扩展。在实际开发中,可以根据项目具体情况选择适合的实现方式,结合框架特性进一步优化。