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) { 	@Override 	public void cooking() { 		System.out.println("今天是周一，今天做农家一碗香。"); 	} }, TUESDAY(2) { 	@Override 	public void cooking() { 		System.out.println("今天是周二，今天做鱼香肉丝。"); 	} }, WEDNESDAY(3) { 	@Override 	public void cooking() { 		System.out.println("今天是周三，今天做韭菜鸡蛋。"); 	} }, THURSDAY(4) { 	@Override 	public void cooking() { 		System.out.println("今天是周四，今天做回锅肉。"); 	} }, FRIDAY(5) { 	@Override 	public void cooking() { 		System.out.println("今天是周五，今天做蛋炒饭。"); 	} }, SATURDAY(6) { 	@Override 	public void cooking() { 		System.out.println("今天是周六，今天做麻婆豆腐。"); 	} }, SUNDAY(7) { 	@Override 	public 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) { 		@Override 		public void cooking() { 			System.out.println("今天是周一，今天做农家一碗香。"); 		} 	}, 	TUESDAY(2) { 		@Override 		public void cooking() { 			System.out.println("今天是周二，今天做鱼香肉丝。"); 		} 	}, 	WEDNESDAY(3) { 		@Override 		public void cooking() { 			System.out.println("今天是周三，今天做韭菜鸡蛋。"); 		} 	}, 	THURSDAY(4) { 		@Override 		public void cooking() { 			System.out.println("今天是周四，今天做回锅肉。"); 		} 	}, 	FRIDAY(5) { 		@Override 		public void cooking() { 			System.out.println("今天是周五，今天做蛋炒饭。"); 		} 	}, 	SATURDAY(6) { 		@Override 		public void cooking() { 			System.out.println("今天是周六，今天做麻婆豆腐。"); 		} 	}, 	SUNDAY(7) { 		@Override 		public 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的方法，欢迎小伙伴们一起讨论。