[Java知识库]Java 策略枚举类

需求:
根据工作时间来计算工人工资.

• 周一到周五,超过八个小时的部分会额外计算加班工资
• 周六周日的所有工作都会有额外工资
• 特别日双倍工资

?策略枚举类

public enum PayrollDay {

    // 周一到周五工资计算
    MONDAY(PayType.WEEKDAY, 1),
    TUESDAY(PayType.WEEKDAY, 2),
    WEDNESDAY(PayType.WEEKDAY, 3),
    THURSDAY(PayType.WEEKDAY, 4),
    FRIDAY(PayType.WEEKDAY, 5),
    // 周六周日工资计算
    SATURDAY(PayType.WEEKEND, 6),
    SUNDAY(PayType.WEEKEND, 7),
    // 特别日工资计算
    OTHER(PayType.SPECIAL, 99);
	
	// 工资支付类型(平日上班支付?周末上班支付?特别日支付?)
    private final PayType payType;
    // 工资支付类型所对应的code
    private final Integer code;

    PayrollDay(PayType payType, Integer code) {
        this.payType = payType;
        this.code = code;
    }
	
	// 根据工作时长和时长对应的工资计算应付酬劳
    public int pay(int minsWorked, int payRate) {
        return payType.pay(minsWorked, payRate);
    }
	
	// 根据code返回对应的枚举类
    public static PayrollDay fromCode(int code) {
        return Arrays.stream(PayrollDay.values()).filter(item -> item.code == code).findAny().orElse(null);
    }

    // 内部枚举类
    private enum PayType {

        // 周一到周五计算加班工资策略
        WEEKDAY {

            @Override
            int overtimePay(int minsWorked, int payRate) {
                return minsWorked <= MINS_PER_SHIFT ? 0 : (minsWorked - MINS_PER_SHIFT) * payRate / 2;
            }
        },
        // 周六到周日计算加班工资策略
        WEEKEND {

            @Override
            int overtimePay(int minsWorked, int payRate) {
                return minsWorked * payRate / 2;
            }
        },
        // 指定的特别日加班工资计算策略
        SPECIAL {

            @Override
            int overtimePay(int minsWorked, int payRate) {
                return minsWorked * payRate;
            }
        };

        // 计算加班工资的抽象方法
        abstract int overtimePay(int minsWorked, int payRate);
        // 每日的正常工作时间 8小时 * 60分钟,超过此部分的工作时间算加班
        private static final int MINS_PER_SHIFT = 8 * 60;

        // 薪酬的计算方法
        int pay(int minsWorked, int payRate) {

            // 基本工资: 工作分钟数 * 每分钟酬劳
            int basePay = minsWorked * payRate;
            return basePay + overtimePay(minsWorked, payRate);
        }
    }
}

?调用

@Service
public class EnumTest implements CommandLineRunner {

    @Override
    public void run(String... args) throws Exception {

        // 从枚举类中获取周五到周日三天的枚举类
        EnumSet enumset = EnumSet.range(PayrollDay.FRIDAY, PayrollDay.SUNDAY);

        // 将EnumSet转换为List
        List<PayrollDay> payrollDays = new ArrayList<>();
        enumset.forEach(item -> payrollDays.add((PayrollDay)item));

        // 模拟数据库查询数据(id为1的员工周五到周日的工作时长)
        List<Staff> staff = List.of(
        	// 工作时间都超过8小时,超出部分需要计算加班工资
        	new Staff("1", 5, 500),
            new Staff("1", 6, 500),
            new Staff("1", 7, 600)
        );

        // 封装枚举Map,key为枚举类对象,value为当前枚举类对象所对应的当日工作时长
        EnumMap<PayrollDay, Integer> payrollDayIntegerEnumMap = new EnumMap<>(PayrollDay.class);
        staff.forEach(item -> payrollDayIntegerEnumMap.put(
        	// 将支付类型code抓换为对应的枚举类
        	PayrollDay.fromCode(item.getDay())
        	// 工作时长
        	, item.getHour()
		));

        for (PayrollDay payrollDay : payrollDays) {
            // 打印每天的工资
            System.out.println(payrollDay.pay(payrollDayIntegerEnumMap.get(payrollDay), 2));
        }

        // 计算指定日期的工资总计
        Integer salaryCount = payrollDays.stream()
        		// 根据工资时间计算工资(item为枚举类对象)
        		.map(item -> item.pay(payrollDayIntegerEnumMap.get(item), 2))
        		// 归约计算工资总和
        		.reduce(Integer::sum)
        		.orElse(0);
        System.out.println("指定日期的工资共计为: " + salaryCount);
    }
}

?效果