例如,如果用户选择选项1 - 3,则执行他们自己的特定任务。如果用户输入4,则执行所有选项1 - 3。是否有一种方法来完成这个使用switch语句,而不必复制和粘贴所有的代码从每个情况?
switch (option) {
case 1: {
System.out.println("1");
break;
}
case 2: {
System.out.println("2");
break;
}
case 3: {
System.out.println("3");
break;
}
case 4: {
System.out.println("1");
System.out.println("2");
System.out.println("3");
break;
}
default: {
System.out.print("Invalid selection.");
break;
}
}
switch
语句不允许重复的case
,所以所要求的功能只能通过多个if
语句使用OR操作来实现:
if (option == 1 || option == 4) {
System.out.println("1");
}
if (option == 2 || option == 4) {
System.out.println("2");
}
if (option == 3 || option == 4) {
System.out.println("3");
}
另一种方法可能包括准备一个特定操作或操作列表的选项映射,以实现所需的逻辑。
Map<Integer, Runnable>
+Predicate.or
// define a map of simple actions
static final Map<Integer, Runnable> actionMap = new LinkedHashMap<>();
static {
actionMap.put(1, ()-> System.out.println("1"));
actionMap.put(2, ()-> System.out.println("2"));
actionMap.put(3, ()-> System.out.println("3"));
};
public static void runOption(Integer option) {
if (option < 1 || option > 4) {
System.out.println("No action found for option = " + option);
} else {
Predicate<Integer> is4 = (x) -> 4 == option;
actionMap.keySet()
.stream()
.filter(is4.or(option::equals))
.map(actionMap::get)
.forEach(Runnable::run);
}
}
测试:
IntStream.range(0, 6).boxed().forEach(MyClass::runOption);
输出:
No action found for option = 0
1
2
3
1
2
3
No action found for option = 5
Map<Integer, List<Runnable>>
withgetOrDefault
这个方法便于操作的任何组合,而不仅仅是用一个操作来运行所有可用的操作。
static Runnable
action1 = ()-> System.out.println("1"),
action2 = ()-> System.out.println("2"),
action3 = ()-> System.out.println("3");
static final Map<Integer, List<Runnable>> actionListMap = Map.of(
1, Arrays.asList(action1),
2, Arrays.asList(action2),
3, Arrays.asList(action3),
4, Arrays.asList(action1, action2, action3)
);
public static void runOptionList(Integer option) {
actionListMap.getOrDefault(
option,
Arrays.asList(() -> System.out.println("No action found for option = " + option))
)
.forEach(Runnable::run);
}