Java之使用策略模式替代 if-else

在Java中，通常情况下 if-else 语句用于根据不同条件执行不同的逻辑。而策略模式则是一种设计模式，它允许在运行时选择算法的行为。

策略模式的主要思想是将算法封装成独立的对象，使得它们可以相互替换，使得算法的变化独立于使用算法的客户。这种模式可以帮助我们避免使用大量的
if-else 或者 switch-case 结构，从而增强代码的灵活性和可维护性。

使用策略模式替代 if-else

假设我们有一个根据不同类型的用户给予不同折扣的需求。通常的实现可能会使用 if-else 结构来根据用户类型执行不同的逻辑，例如：

java">java
public class DiscountCalculator {

    public double calculateDiscount(User user, double purchaseAmount) {
        if (user.getType().equals(UserType.REGULAR)) {
            return purchaseAmount * 0.1; // 10% discount for regular users
        } else if (user.getType().equals(UserType.PREMIUM)) {
            return purchaseAmount * 0.2; // 20% discount for premium users
        } else if (user.getType().equals(UserType.GUEST)) {
            return 0; // no discount for guest users
        }
        return 0;
    }
}

这种方式虽然简单，但是随着用户类型的增加或者逻辑的变更，可能会导致 if-else 结构的增加和修改，不利于代码的扩展和维护。

使用策略模式重构
下面是如何使用策略模式重构上述例子：
定义策略接口：

java">public interface DiscountStrategy {
    double applyDiscount(double purchaseAmount);
}

实现具体的策略类：
java

java">public class RegularDiscountStrategy implements DiscountStrategy {
    @Override
    public double applyDiscount(double purchaseAmount) {
        return purchaseAmount * 0.1; // 10% discount for regular users
    }
}

public class PremiumDiscountStrategy implements DiscountStrategy {
    @Override
    public double applyDiscount(double purchaseAmount) {
        return purchaseAmount * 0.2; // 20% discount for premium users
    }
}

public class GuestDiscountStrategy implements DiscountStrategy {
    @Override
    public double applyDiscount(double purchaseAmount) {
        return 0; // no discount for guest users
    }
}

使用策略模式：
java

java">public class DiscountCalculator {

    private DiscountStrategy discountStrategy;

    public DiscountCalculator(DiscountStrategy discountStrategy) {
        this.discountStrategy = discountStrategy;
    }

    public void setDiscountStrategy(DiscountStrategy discountStrategy) {
        this.discountStrategy = discountStrategy;
    }

    public double calculateDiscount(double purchaseAmount) {
        return discountStrategy.applyDiscount(purchaseAmount);
    }
}

客户端使用：
java

java">public class Main {
    public static void main(String[] args) {
        User user = new User(UserType.REGULAR); // or UserType.PREMIUM, UserType.GUEST
        DiscountCalculator calculator = new DiscountCalculator(new RegularDiscountStrategy());
        double discount = calculator.calculateDiscount(100.0);
        System.out.println("Discount applied: " + discount);
    }
}