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

 

 */
public class AlreadyEvaluatedStatus implements StatusStrategy{

    @Override
    public void status() {
        System.out.println("已评价");
    }

}07状态枚举类

/**
 * @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.扩展性好，增加分支只需要增加对应的策略实现类和枚举。