策略模式替换条件逻辑
策略模式是一种常见的优化条件逻辑的方法。当代码中有一个包含大量条件逻辑(即 if 语句)的方法时,你应该考虑使用策略模式对其进行优化,这样代码更加清晰,同时也更容易维护。
假设我们有这样一段代码:
public class IfElseDemo {
public double calculateInsurance(double income) {
if (income <= 10000) {
return income*0.365;
} else if (income <= 30000) {
return (income-10000)*0.2+35600;
} else if (income <= 60000) {
return (income-30000)*0.1+76500;
} else {
return (income-60000)*0.02+105600;
}
}
}
下面是使用策略+工厂模式重构后的代码:
首先定义一个接口 InsuranceCalculator
,其中包含一个方法 calculate(double income)
,用于计算保险费用。
public interface InsuranceCalculator {
double calculate(double income);
}
然后,分别创建四个类来实现这个接口,每个类代表一个保险费用计算方式。
public class FirstLevelCalculator implements InsuranceCalculator {
public double calculate(double income) {
return income * 0.365;
}
}
public class SecondLevelCalculator implements InsuranceCalculator {
public double calculate(double income) {
return (income - 10000) * 0.2 + 35600;
}
}
public class ThirdLevelCalculator implements InsuranceCalculator {
public double calculate(double income) {
return (income - 30000) * 0.1 + 76500;
}
}
public class FourthLevelCalculator implements InsuranceCalculator {
public double calculate(double income) {
return (income - 60000) * 0.02 + 105600;
}
}
最后,我们可以为每个策略类添加一个唯一的标识符,例如字符串类型的 name
属性。然后,在工厂类中创建一个 Map
来存储策略对象和它们的标识符之间的映射关系(也可以用 switch 来维护映射关系)。
import java.util.HashMap;
import java.util.Map;
public class InsuranceCalculatorFactory {
private static final Map<String, InsuranceCalculator> CALCULATOR_MAP = new HashMap<>();
static {
CALCULATOR_MAP.put("first", new FirstLevelCalculator());
CALCULATOR_MAP.put("second", new SecondLevelCalculator());
CALCULATOR_MAP.put("third", new ThirdLevelCalculator());
CALCULATOR_MAP.put("fourth", new FourthLevelCalculator());
}
public static InsuranceCalculator getCalculator(String name) {
return CALCULATOR_MAP.get(name);
}
}
这样,就可以通过 InsuranceCalculatorFactory
类手动获取相应的策略对象了。
double income = 40000;
// 获取第二级保险费用计算器
InsuranceCalculator calculator = InsuranceCalculatorFactory.getCalculator("second");
double insurance = calculator.calculate(income);
System.out.println("保险费用为:" + insurance);
这种方式允许我们在运行时根据需要选择不同的策略,而无需在代码中硬编码条件语句。