利用策略枚举重构代码_JAVA_编程开发_程序员俱乐部

中国优秀的程序员网站程序员频道CXYCLUB技术地图
热搜:
更多>>
 
您所在的位置: 程序员俱乐部 > 编程开发 > JAVA > 利用策略枚举重构代码

利用策略枚举重构代码

 2013/9/30 22:09:15  liuluo129  程序员俱乐部  我要评论(0)
  • 摘要:问题描述:在创建合同时,会选择一个何时付款的策略,比如,目前策略有:合同执行开始时付款100%,合同执行结束后付款100%合同每月结算当月应付金额...并且付款的策略还会改变。最初的代码使用一大串if...else...,如下:publicList<PaymentPlan>resolve(ExecutionUnitunit,StringpaymentTermsNumber){if(unit!=null){if(PaymentTermEnum.OTHER.getValue()
  • 标签:代码 利用 枚举

问题描述:在创建合同时,会选择一个何时付款的策略,比如,目前策略有:

? ? 合同执行开始时付款100%,

? ? 合同执行结束后付款100%

? ? 合同每月结算当月应付金额

? ? ...

?

并且付款的策略还会改变。

最初的代码使用一大串if...else...,如下:

class="java" name="code"> public List<PaymentPlan> resolve(ExecutionUnit unit, String paymentTermsNumber) {
        if (unit != null) {
            if (PaymentTermEnum.OTHER.getValue().equals(paymentTermsNumber)) {
                return resolve0(unit);
            } else if (PaymentTermEnum.FIVE_DAY_NEXT_MONTH.getValue().equals(paymentTermsNumber)) {
                return resolve1(unit);
            } else if (PaymentTermEnum.BEFORE_END_MONTH_PAY_PREVIEW_MONTH.getValue().equals(paymentTermsNumber)) {
                return resolve2(unit);
            } else if (PaymentTermEnum.BEFORE_END_MONTH_PAY_NEXT_MONTH.getValue().equals(paymentTermsNumber)) {
                return resolve3(unit);
            } 
            ...
        }
        return new ArrayList<PaymentPlan>();
    }
private List<PaymentPlan> resolve0(ExecutionUnit unit) {
        List<PaymentPlan> paymentPlanList = new ArrayList<PaymentPlan>();
        if (unit == null) {
            return paymentPlanList;
        }
        Contract contract = contractService.getByNumber(unit.getContractnum());
        if (contract == null) {
            return paymentPlanList;
        }
        PaymentPlan filter = new PaymentPlan();
        filter.setContractId(contract.getId());
        return paymentPlanService.list(filter);
    }
....

?和旧社会妇女的裹脚布一样,又臭又长。而且当策略改变的时候,还需要把裹脚布再加长一点,这时很可能疏忽了而没有添加(程序猿们经常会拿疏忽了,忘记了当借口,你懂的)。

当受不了恶心的时候就是进步的前兆,因此决定重构这部分代码,首先代码中判断的时候用到了枚举类PaymentTermEnum,但是里面只有下面这些:

public enum PaymentTermEnum {

    // 其他
    OTHER("0"),
    // 当月广告款次月5日结清
    FIVE_DAY_NEXT_MONTH("1"),
    // 每月30日前向乙方支付上月广告款.
    BEFORE_END_MONTH_PAY_PREVIEW_MONTH("2"),

    // 每月30日前向乙方支付下月广告款
    BEFORE_END_MONTH_PAY_NEXT_MONTH("3"),

    // 每个季度执行完广告后,30日内支付相应的款项
    AFTER_QUARTER_PAY_WITHIN_THIRTY("5"),

    // 每个季度执行完广告后,45日内支付相应的款项.
    AFTER_QUARTER_PAY_WITHIN_FORTH_FIVE("6"),

    // 每个季度执行完广告后,在当季内支付相应的款项.
    AFTER_QUARTER_PAY_WITHIN_THIS_QUARTER("7"),

    // 每个月执行完当月的广告后,30日内支付相应的款项.
    AFTER_END_MONTH_PAY_WITHIN_THIRTY("8"),

    // 每个月执行完当月的广告后,在当月月末支付100%的款项.
    AFTER_END_MONTH_PAY_END_THIS_MONTH("9"),

    // 每张执行单执行前预付100%的款项.
    PAY_BEFORE_EXECUTE("10"),

    // 每张执行单执行前预付30%的款项,执行完付剩余70%的款项.
    PAY_THIRTY_BEFORE_EXECUTE_SEVENTY_AFTER_EXECUTED("11"),

    // 每张执行单执行前预付50%的款项,执行完付剩余50%的款项.
    PAY_HALF_BEFORE_EXECUTE_HALF_AFTER_EXECUTED("12"),

    // 每张执行单执行完广告后,30日内支付相应的款项.
    AFTER_EXECUTED_PAY_WITHIN_THIRTY_DAY("13"),
    // 每张执行单执行完广告后,5个工作日内支付相应的款项.
    AFTER_EXECUTED_PAY_WITHIN_FIVE_DAY("14");

    private String value;

    private PaymentTermEnum(String value) {
        this.value = value;
    }

    public String getValue() {
        return this.value;
    }
}

?这只能避免在解析的方法中不使用类似于"0","1"这种tricky的字符串。虽然有点小用,但是没有完全发挥枚举的其他作用。

首先,每个策略最终都是要解析成 List<PaymentPlan>,即付款计划的列表,可以在枚举类中使用一个抽象方法,这样枚举中的每个常量必须要实现,此时枚举类变成:

public enum PaymentTermEnum {
    // 其他
    OTHER("0") {
        @Override
        public List<PaymentPlan> resolve(ExecutionUnit unit) {
            ...
        }
    },
    // 当月广告款次月5日结清
    FIVE_DAY_NEXT_MONTH("1") {
        @Override
        public List<PaymentPlan> resolve(ExecutionUnit unit) {
            ...
        }
    },
    // 每月30日前向乙方支付上月广告款.
    BEFORE_END_MONTH_PAY_PREVIEW_MONTH("2") {
        @Override
        public List<PaymentPlan> resolve(ExecutionUnit executionUnit) {
            ...
        }
    },
	...
    ;

    // 解析付款计划
    public abstract List<PaymentPlan> resolve(ExecutionUnit executionUnit);

    private String value;

    private static ExecutionUnitService executionUnitService = ApplicationContextUtil.getContext().getBean(
            ExecutionUnitService.class);
    private static ContractService contractService = ApplicationContextUtil.getContext().getBean(ContractService.class);
    private static PaymentPlanService paymentPlanService = ApplicationContextUtil.getContext().getBean(
            PaymentPlanService.class);

    private static Map<String, PaymentTermEnum> valueMap = new HashMap<String, PaymentTermEnum>();

    static {
        for (PaymentTermEnum paymentTermEnum : PaymentTermEnum.values()) {
            valueMap.put(paymentTermEnum.value, paymentTermEnum);
        }
    }

    public static PaymentTermEnum fromValue(String value) {
        return valueMap.get(value);
    }

    private PaymentTermEnum(String value) {
        this.value = value;
    }

    public String getValue() {
        return this.value;
    }
}

?解析的Service类的方法就可以简化成:

    public List<PaymentPlan> resolve(ExecutionUnit unit, String paymentTermsNumber) {
        PaymentTermEnum paymentTermEnum = PaymentTermEnum.fromValue(paymentTermsNumber);
        if (paymentTermEnum != null) {
            return paymentTermEnum.resolve(unit);
        }
        return new ArrayList<PaymentPlan>();
    }

?当要添加解析策略的时候,只需要在枚举类中添加一个新的常量即可,而此时编译器会强制我们实现resolve方法,还想假装疏忽没注意,嘿嘿妈妈都不同意。

?

发表评论
用户名: 匿名