策略枚举(enum)

最近看了看effective java这本书,抄一段书.

考虑用一个enum表示薪资包中的每个工作日的薪资.

这个枚举有一些方法,根据给定某工人的基本工资(按小时)以及当天的工作时间,来计算他当天的报酬.

在五个工作日中,超过正常8小时的工作时间都会产生加班工资,在双休日中,所有工作都产生加班工资.

利用switch语句,很容易通过将多个case标签分别应用到俩个代码片段中,来完成这一计算.

enum PayrollDay {
    MONDAY, TUESDAY, WEDNESDAY, THURSDAY, FRIDAY,
    SATURDAY, SUNDAY;

    private static final int HOURS_PER_SHIFT = 8;

    double pay(double hoursWorked, double payRate){
        double basePay = hoursWorked * payRate;
        double overtimePay;
        switch(this){
            case SATURDAY: case SUNDAY:
                overtimePay = hoursWorked * payRate  / 2; break;
            default:
                overtimePay = hoursWorked <= HOURS_PER_SHIFT ? 0 : (hoursWorked - HOURS_PER_SHIFT) * payRate / 2;
                break;
        }
        return basePay + overtimePay;
    }
}

不可否认,这段代码十分简洁,但是从维护的角度来看,非常危险.

假设将一个元素加到该enum中,或许是一个表示假期天数的特殊值,

但是忘记给switch语句添加相应的case,程序依然可以编译,但pay方法会悄悄地将假期的工资计算与正常工作日的相同..

为了利用特定于常量的方法实现安全的执行工资计算,你可能必须重复计算每个常量的加班工资,或者将计算移到俩个辅助方法中(一个用来计算工作日,一个用来计算双休日),

并且从每个常量中调用相应的辅助方法,这任何一种方法都会产生相当数量的样板代码,结果降低了可读性,并增加了出错的几率.

通过计算工作日加班工资的具体方法代替PayrollDay中抽象的overtimePay方法,可以减少样板代码.

这样,就只有双休日必须覆盖方法了.但是这样也有着与switch语句一样的不足,如果又增加了一天,而没有覆盖overtimePay方法,就会悄悄地延续工作日的计算.

你真正想要的就是每当添加一个枚举常量时,就强制选择一种加班报酬策略.

幸运的是,有一种很好的方法实现这一点.这种想法就是将加班工资计算移动到一个私有的嵌套enum中,将这个策略enum的实例传到PayrollDay 枚举的构造器中.

之后PayrollDay enum 将基板工资计算给委托给策略enum,PayrollDay 中就不需要switch语句或者特定于常量的方法实现了.

虽然这种模式没有switch语句那么简洁,但更加安全,也更加灵活.

策略enum

enum PayrollDay {
	MONDAY(PayType.WEEKDAY), TUESDAY(PayType.WEEKDAY), WEDNESDAY(PayType.WEEKDAY), THURSDAY(PayType.WEEKDAY), FRIDAY(PayType.WEEKDAY),
	SATURDAY(PayType.WEEKEND), SUNDAY(PayType.WEEKEND);
	
	private final PayType payType;
	
	PayrollDay(PayType payType){
		this.payType = payType;
	}
	
	double pay(double hoursWorked, double payRate){
		return payType.pay(hoursWorked, payRate);
	}
	
	//the strategy enum type
	private enum PayType{
		WEEKDAY {
			double overtimePay(double hours, double payRate){
				return hours <= HOURS_PER_SHIFT ? 0 : (hours - HOURS_PER_SHIFT) * payRate / 2;
			}	
		},
		WEEKEND{
			double overtimePay(double hours, double payRate){
				return hours * payRate / 2;
			}
		};
		
		private static final int HOURS_PER_SHIFT = 8;
		
		abstract double overtimePay(double hrs, double payRate);
		
		double pay(double hoursWorked, double payRate){
			double basePay = hoursWorked * payRate;
			return basePay + overtimePay(hoursWorked, payRate);
		}
	}
}