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