背景
在一次需求开发时,发现使用Lambda的forEach()跳不出循环。如下示例代码,想在遍历满足条件时跳出循环。
public static void main(String[] args) {
List<Integer> list = Arrays.asList(1, 4, 5, 7, 9, 11);
list.forEach(e -> {
if (e % 2 == 0) {
System.out.println("foreach -- " + e);
return;
}
System.out.println(e);
});
System.out.println(list);
}
运行结果:
可以看出在forEach()中使用return并不会退出整个循环,和普通for循环return意义不同,仍会继续遍历。
原因
在普通for循环中,跳出循环使用break,结束本次循环使用continue,结束for循环所在的整个执行方法使用return。
for (Integer e : list) {
if (e % 2 == 0) {
break; // return直接整个函数终止执行返回,break for循环方法终止执行
}
System.out.println(e);
}
Lambda表达式中,函数式接口Consumer 的抽象方法accept引用实现循环体中的逻辑。
所以forEach()处理一个个的执行方法accept(t),非循环体。在执行方法中使用return将处理方法返回,但不能结束整个forEach()。
default void forEach(Consumer<? super T> action) {
Objects.requireNonNull(action);
for (T t : this) {
action.accept(t);
}
}
@FunctionalInterface
public interface Consumer<T> {
void accept(T t);
default Consumer<T> andThen(Consumer<? super T> after) {
Objects.requireNonNull(after);
return (T t) -> { accept(t); after.accept(t); };
}
}
优化方案
跳出Lambda表达式forEach()循环解决思路有以下几种:
- 抛异常
在遍历时,若需要跳出循环,通过抛异常结束forEach()循环,在循环外catch异常不处理。首先此方案不够优雅,其次若循环逻辑块中有其他抛异常的地方,会受影响不易发现。
public static void main(String[] args) {
List<Integer> list = Arrays.asList(1, 4, 5, 7, 9, 11);
try {
list.forEach(e -> {
if (e % 2 == 0) {
System.out.println("foreach -- " + e);
throw new RuntimeException("跳出循环");
}
System.out.println(e);
});
} catch (Exception e) {}
System.out.println(list);
}
执行结果:
-
使用普通for循环
使用普通for循环替代Lambda表达式forEach()循环,在代码块中使用break即可跳出循环。 -
使用Lambda表达式filter()实现
换个实现思路,诉求是遍历list遇到第一个满足条件的item跳出循环,那么转成过滤整个list,返回第一个满足条件的item。
Optional<Integer> first = list.stream().filter(e -> e % 2 == 0).findFirst();
System.out.println(first.orElse(null));
- 使用anyMatch()
原理类似filter(),遇到满足条件的item跳出遍历返回。
list.stream().anyMatch(e -> {
if (e % 2 == 0) {
System.out.println("跳出循环 -- " + e);
return true;
}
System.out.println(e);
return false;
});
执行结果:
总结
Lambda表达式forEach()函数不支持return跳出循环,不建议使用抛异常方式结束循环,可以考虑使用普通for或利用Lambda表达式的函数实现。