Python中枚举取代IF
策略模式+工厂模式（反射）+枚举代替 if..else if..
邹祥慧 2019-01-24 11:04:06 
509 
收藏 1
实际项目中我们经常碰到需要使用if..else..if的分支判断这种情况。
这种写法带来一些弊端。
01一旦分支多太多，逻辑复杂，会导致代码十分冗长，增加阅读难度。
02如果需要增加或减少分支，需要改动if…elseif，增大因代码改动而出错的风险。这种粗鄙，丑陋的代码是我们不能接受的。
如：
注：实际项目中的订单状态远远不止这些，小编只是模拟一个if...else..if 这样的一个场景
解决的方法是，可以通过策略模式+工厂模式，巧妙地利用反射机制和枚举类的valueof方法，规避掉if…else..if写法带来的弊端。下面我们通过一个订单状态这个实际例子来理解这种做法。
案例
01状态策略接口
/**
*
* @Description:状态策略接口
* @time:2019年1月24日 上午10:03:27
*/
public interface StatusStrategy {
void status();
}02未支付状态
/**
* @Description:未支付状态
* @time:2019年1月24日 上午9:54:08
*/
public class UnpaidStatus implements StatusStrategy{
@Override
public void status() {
System.out.println("未支付状态");
}
}03未发货状态
/**
* @Description:未发货状态
* @time:2019年1月24日 上午9:55:14
*/
public class UnshippedStatus implements StatusStrategy{
@Override
public void status() {
System.out.println("未发货状态");
}
}04已发货状态
/**
* @Description:已发货状态
* @time:2019年1月24日 上午9:56:32
*/
public class ShippedStatus implements StatusStrategy{
@Override
public void status() {
System.out.println("已发货状态");
}
}05确认收货状态
/**
* @Description:确认收货状态
* @time:2019年1月24日 上午9:57:09
*/
public class ConfirmReceiptStatus implements StatusStrategy{
@Override
public void status() {
System.out.println("确认收货状态");
}
}06已评价状态
/**
* @Description:已评价状态 * @time:2019年1月24日 上午10:02:10
*/07状态枚举类
public class AlreadyEvaluatedStatus implements StatusStrategy{
@Override
public void status() {
System.out.println("已评价");
}
}
/**
* @Description:状态枚举类【如果需要增删状态类型，只需要在StatusEnum类添加或删除对应的枚举，而不是增删if…else if…】
* @time:2019年1月24日 上午10:08:28
*/
public enum StatusEnum {
/**
* 未支付状态
*/
UNPAIDSTATUS("demo.UnpaidStatus"),
/**
* 未发货状态
*/
UNSHIPPEDSTATUS("demo.UnshippedStatus"),
/**
* 已发货状态
*/
SHIPPEDSTATUS("demo.ShippedStatus"),
/**
* 确认收货状态
*/
CONFIRMRECEIPTSTATUS("demo.ConfirmReceiptStatus"),
/**
* 已评价状态
*/
ALREADYEVALUATEDSTATUS("demo.AlreadyEvaluatedStatus");
StatusEnum(String className) {
this.setClassName(className);
}
private String className;
public String getClassName() {
return className;
}
public void setClassName(String className) {
this.className = className;
}
}08工厂类
/**
* @Description:工厂类
* @time:2019年1月24日 上午10:13:45
*/
public class StatusFactory {
/**
* 该工厂类提供静态方法，巧妙地利用反射机制和枚举类的valueof方法，返回具体策略实例。
* @param strategyType 策略类型
* @return
* @throws Exception
*/
public static StatusStrategy getStrategy(String strategyType) throws Exception {
String className = StatusEnum.valueOf(strategyType).getClassName();
return (StatusStrategy) Class.forName(className).newInstance();
}
}
DEMO
public class Demo {
public static void main(String[] args) throws Exception {
String statusStrategyParam = "UNPAIDSTATUS";//未支付状态
StatusStrategy strategy = StatusFactory.getStrategy(statusStrategyParam);
strategy.status();
statusStrategyParam = "UNSHIPPEDSTATUS";//未发货状态
strategy = StatusFactory.getStrategy(statusStrategyParam);
strategy.status();
statusStrategyParam = "SHIPPEDSTATUS";//已发货状态
strategy = StatusFactory.getStrategy(statusStrategyParam);
strategy.status();
statusStrategyParam = "CONFIRMRECEIPTSTATUS";//确认收货状态
strategy = StatusFactory.getStrategy(statusStrategyParam);
strategy.status();
statusStrategyParam = "ALREADYEVALUATEDSTATUS";//已评价状态
strategy = StatusFactory.getStrategy(statusStrategyParam);
strategy.status();
}
}测试结果
未支付状态
未发货状态
已发货状态
确认收货状态
已评价
“
总结
if…else if…的缺点
1.多分支和复杂度高的逻辑会使代码冗长，难以理解和维护。
2.增删分支需要改动if…else if结构，增大代码出错风险。
策略模式+工厂模式的优点
1.对应的分支处理分成不同策略类来实现，使代码易于读懂和维护
2.扩展性好，增加分支只需要增加对应的策略实现类和枚举。