不要再使用 if-else 块了！结合使用策略和工厂模式-java教程-PHP中文网

Don’t use if-else blocks anymore! Use Strategy and Factory Pattern Together

当我们在项目中前进时，迷失在 if-else 块中，在复杂的条件和重复的代码中挣扎，我们寻找解决方案。但为什么我们要陷入 if-else 块中呢？在这篇文章中，让我们结合策略模式和工厂模式来探索摆脱 if-else 混乱的方法。

问题：If-Else 混淆

假设您正在开发一个电子商务应用程序，并且需要支持不同的支付方式，例如信用卡、借记卡和加密货币。您从 if-else 块开始处理付款：

public class PaymentService {

    public void processPayment(String paymentType) {
        if (paymentType.equals("CREDIT_CARD")) {
            System.out.println("Processing credit card payment...");
        } else if (paymentType.equals("DEBIT_CARD")) {
            System.out.println("Processing debit card payment...");
        } else if (paymentType.equals("CRYPTO")) {
            System.out.println("Processing crypto payment...");
        } else {
            throw new IllegalArgumentException("Invalid payment type");
        }
    }
}

登录后复制

虽然一开始看起来很简单，但随着支付方式的增加，if-else 的复杂性也会增加。新的支付方式意味着增加新的条件。结果就是一堆难以管理的代码。而这种方法是违背开闭原则的。

但是，我们可以使用策略模式和工厂模式来解决这个问题。

首先，让我们创建一个枚举：

public enum PaymentType {
    CREDIT_CARD,
    DEBIT_CARD,
    CRYPTO
}

登录后复制

解决方案：使用策略模式进行清理

public interface PaymentStrategy {
    void pay(PaymentRequest request);
}

public class CreditCardPayment implements PaymentStrategy {
    @Override
    public void pay(PaymentRequest request) {
        System.out.println("Processing $type payment".replace("$type", String.valueOf(request.getPaymentType())));
    }
}

public class DebitCardPayment implements PaymentStrategy {
    @Override
    public void pay(PaymentRequest request) {
        System.out.println("Processing $type payment".replace("$type", String.valueOf(request.getPaymentType())));
    }
}

public class CryptoPayment implements PaymentStrategy {
    @Override
    public void pay(PaymentRequest request) {
        System.out.println("Processing $type payment".replace("$type", String.valueOf(request.getPaymentType())));
    }
}

登录后复制

在此阶段，通过通用接口实现每种支付方式的单独策略。现在，通过工厂模式，我们将决定选择哪种策略。

第二步：选择工厂模式的策略

在这一步中，我们可以使用 EnumMap 使工厂模式更加简洁和优化。

public class PaymentFactory {
    private static final Map<PaymentType, PaymentStrategy> strategies = new EnumMap<>(PaymentType.class);

    static {
        strategies.put(PaymentType.CREDIT_CARD, new CreditCardPayment());
        strategies.put(PaymentType.DEBIT_CARD, new DebitCardPayment());
        strategies.put(PaymentType.CRYPTO, new CryptoPayment());
    }

    public static PaymentStrategy getPaymentStrategy(PaymentType paymentType) {
        PaymentStrategy strategy = strategies.get(paymentType);

        if (Objects.isNull(strategy))
            throw new IllegalArgumentException("Strategy not found");

        return strategy;
    }
}

登录后复制

最后一步：服务类别重组

现在，让我们使用我们所做的。

public class PaymentService {

    public void processPayment(PaymentRequest request) {
        // Don't forget to check objects if null!
        if (Objects.isNull(request) || Objects.isNull(request.getPaymentType())
            throw new IllegalArgumentException("Request can not be null!");
        PaymentStrategy strategy = PaymentFactory.getPaymentStrategy(request.getPaymentType());

        strategy.pay(request);
    }
}

登录后复制

事实上，我们不需要任何 if-else 块来进行支付处理。感谢策略和工厂模式，我们的代码更加干净、模块化和可扩展。