Java中如何解决if-else（策略+枚举）

最近接到了一个新需求，按照不同的编码去执行不同的逻辑，但最后返回的数据类型是一致的，都是相同对象的List集合。
完成这个需求的话可以使用if-else来执行不同的方法，虽然if-else可以实现，但if-else是一种面向过程的实现。虽然这个需求中编码是固定的，但是编码也太多了，如果用if-else写出来可读性估计也不高。
这个时候就想到之前无意中看到过一篇文章，文章的大致内容，就是使用策略+枚举的方式去解决if-else。正好今天用到了，那么就展开说说怎么使用策略+枚举的方式怎么来替换掉if-else吧.

举个例子吧，比如说每周有7天，我每天都回家做不同的菜，根据入参1-7代表周一到周日，提醒我今天该做什么菜了，那么我们就使用if-else来实现的话，代码是这样的。

public static void main(String[] args) {//获取今天是周几LocalDate today = LocalDate.now();DayOfWeek dayOfWeek = today.getDayOfWeek();int dayOfWeekValue = dayOfWeek.getValue();if(dayOfWeekValue == 1){System.out.println("今天是周一，今天做农家一碗香。");}else if(dayOfWeekValue == 2){System.out.println("今天是周二，今天做鱼香肉丝。");}else if(dayOfWeekValue == 3){System.out.println("今天是周三，今天做韭菜鸡蛋。");}else if(dayOfWeekValue == 4){System.out.println("今天是周四，今天做回锅肉。");}else if(dayOfWeekValue == 5){System.out.println("今天是周五，今天做蛋炒饭。");}else if(dayOfWeekValue == 6){System.out.println("今天是周六，今天做麻婆豆腐。");}else if(dayOfWeekValue == 7){System.out.println("今天是周日，今天做土豆丝。");}else{throw new ServiceException("今天是周几，今天做什么菜？");}}

如果我们用枚举+策略去实现呢？
首先我们先定义一个这样的枚举

@Getter
public enum DoTheCookingEnum {MONDAY(1),TUESDAY(2),WEDNESDAY(3),THURSDAY(4),FRIDAY(5),SATURDAY(6),SUNDAY(7);/*** @description: 周几的数字表达形式**/public final int dayOfWeekValue;private DoTheCookingEnum(int dayOfWeekValue) {this.dayOfWeekValue = dayOfWeekValue;}}

包含了参数，构造器以及周一到周日对应的枚举实例，并使用了Lombok生成了dayOfWeekValue参数的get方法。参数dayOfWeekValue对应的是1-7的条件，接着我们在枚举中定义一个根据dayOfWeekValue参数来获取枚举的方法。

	/*** @description: 根据dayOfWeekValue来获取不同的枚举* @author: Vinci* @date: 2024/3/15 13:08**/public static DoTheCookingEnum getEnum(int dayOfWeekValue) {for (DoTheCookingEnum doTheCookingEnum : DoTheCookingEnum.values()) {if (doTheCookingEnum.dayOfWeekValue == dayOfWeekValue) {return doTheCookingEnum;}}return null;}

这样我们就可以根据不同的dayOfWeekValue获取到相应的枚举了，接下来的步骤就是重点了，我们在枚举中定义一个抽象方法，作为做菜的行为。

	/*** @description: 做菜* @author: Vinci* @date: 2024/3/15 13:12**/public abstract void cooking();

定义了做菜的抽象方法后，我们就可以在每个枚举实例当中去重写cooking()这个抽象方法。

MONDAY(1) {@Overridepublic void cooking() {System.out.println("今天是周一，今天做农家一碗香。");}
},
TUESDAY(2) {@Overridepublic void cooking() {System.out.println("今天是周二，今天做鱼香肉丝。");}
},
WEDNESDAY(3) {@Overridepublic void cooking() {System.out.println("今天是周三，今天做韭菜鸡蛋。");}
},
THURSDAY(4) {@Overridepublic void cooking() {System.out.println("今天是周四，今天做回锅肉。");}
},
FRIDAY(5) {@Overridepublic void cooking() {System.out.println("今天是周五，今天做蛋炒饭。");}
},
SATURDAY(6) {@Overridepublic void cooking() {System.out.println("今天是周六，今天做麻婆豆腐。");}
},
SUNDAY(7) {@Overridepublic void cooking() {System.out.println("今天是周日，今天做土豆丝。");}
};

枚举完整代码如下：

package com.example.demo.enums;import lombok.Getter;/*** @package: com.example.demo.enums* @FileName: DoTheCookingEnum* @author: Vinci* @description: 周几做什么菜* @date: 2024/3/15 11:42*/
@Getter
public enum DoTheCookingEnum {MONDAY(1) {@Overridepublic void cooking() {System.out.println("今天是周一，今天做农家一碗香。");}},TUESDAY(2) {@Overridepublic void cooking() {System.out.println("今天是周二，今天做鱼香肉丝。");}},WEDNESDAY(3) {@Overridepublic void cooking() {System.out.println("今天是周三，今天做韭菜鸡蛋。");}},THURSDAY(4) {@Overridepublic void cooking() {System.out.println("今天是周四，今天做回锅肉。");}},FRIDAY(5) {@Overridepublic void cooking() {System.out.println("今天是周五，今天做蛋炒饭。");}},SATURDAY(6) {@Overridepublic void cooking() {System.out.println("今天是周六，今天做麻婆豆腐。");}},SUNDAY(7) {@Overridepublic void cooking() {System.out.println("今天是周日，今天做土豆丝。");}};/*** @description: 周几的数字表达形式**/public final int dayOfWeekValue;private DoTheCookingEnum(int dayOfWeekValue) {this.dayOfWeekValue = dayOfWeekValue;}/*** @description: 根据dayOfWeekValue来获取不同的枚举* @author: Vinci* @date: 2024/3/15 13:08**/public static DoTheCookingEnum getEnum(int dayOfWeekValue) {for (DoTheCookingEnum doTheCookingEnum : DoTheCookingEnum.values()) {if (doTheCookingEnum.dayOfWeekValue == dayOfWeekValue) {return doTheCookingEnum;}}return null;}/*** @description: 做菜* @author: Vinci* @date: 2024/3/15 13:13**/public abstract void cooking();}

枚举定义完成后，我们就可以去调用不同枚举元素来实现不同的逻辑了。

public static void main(String[] args) {//获取今天是周几LocalDate today = LocalDate.now();DayOfWeek dayOfWeek = today.getDayOfWeek();int dayOfWeekValue = dayOfWeek.getValue();//调用getEnum根据dayOfWeekValue获取枚举对象DoTheCookingEnum anEnum = DoTheCookingEnum.getEnum(dayOfWeekValue);//执行做菜方法，获取今天做什么菜if(anEnum != null){anEnum.cooking();}}

因为今天是周五，所以今晚做蛋炒饭了：
在这里插入图片描述
以上便是策略+枚举来解决if-else的方法，这样写代码耦合度变低了，也更加简洁了，扩展性也很强，比如以后有了个星期八（只是举个例子），是不是我们只需要，在枚举中再创建一个实例，就不用改代码了。当然还有很多种避免使用if-else的方法，欢迎小伙伴们一起讨论。