Java8跳出循环新方式

Python实战社群

Java实战社群

长按识别下方二维码,按需求添加

扫码关注添加客服

进Python社群▲

扫码关注添加客服

进Java社群

作者丨flydean

来源丨JAVA小咖秀

https://juejin.im/post/5ea4cb166fb9a03c73799e7f 

简介

我们通常需要在java stream中遍历处理里面的数据,其中foreach是最最常用的方法。

但是有时候我们并不想处理完所有的数据,或者有时候Stream可能非常的长,或者根本就是无限的。

一种方法是先filter出我们需要处理的数据,然后再foreach遍历。

那么我们如何直接break这个stream呢?今天本文重点讲解一下这个问题。

使用Spliterator

上篇文章我们在讲Spliterator的时候提到了,在tryAdvance方法中,如果返回false,则Spliterator将会停止处理后续的元素。

通过这个思路,我们可以创建自定义Spliterator。

假如我们有这样一个stream:

Stream ints = Stream.of(1, 2, 3, 4, 5, 6, 7, 8, 9, 10);
我们想定义一个操作,当x > 5的时候就停止。

我们定义一个通用的Spliterator:

public class CustomSpliterator extends Spliterators.AbstractSpliterator  {
private Spliterator splitr;private Predicate predicate;private volatile boolean isMatched = true;
public CustomSpliterator(Spliterator splitr, Predicate predicate) {super(splitr.estimateSize(), 0);this.splitr = splitr;this.predicate = predicate;    }
@Overridepublic synchronized boolean tryAdvance(Consumer consumer) {boolean hadNext = splitr.tryAdvance(elem -> {if (predicate.test(elem) && isMatched) {                consumer.accept(elem);            } else {                isMatched = false;            }        });return hadNext && isMatched;    }}

在上面的类中,predicate是我们将要传入的判断条件,我们重写了tryAdvance,通过将predicate.test(elem)加入判断条件,从而当条件不满足的时候返回false.

看下怎么使用:

@Slf4jpublic class CustomSpliteratorUsage {
public static  Stream takeWhile(Stream stream, Predicate predicate) {        CustomSpliterator customSpliterator = new CustomSpliterator<>(stream.spliterator(), predicate);return StreamSupport.stream(customSpliterator, false);    }
public static void main(String[] args) {        Stream ints = Stream.of(1, 2, 3, 4, 5, 6, 7, 8, 9, 10);        List result =          takeWhile(ints, x -> x < 5 )                        .collect(Collectors.toList());        log.info(result.toString());    }}
我们定义了一个takeWhile方法,接收Stream和predicate条件。

只有当predicate条件满足的时候才会继续,我们看下输出的结果:

[main] INFO com.flydean.CustomSpliteratorUsage - [1, 2, 3, 4]
自定义forEach方法

除了使用Spliterator,我们还可以自定义forEach方法来使用自己的遍历逻辑:

public class CustomForEach {public static class Breaker {private volatile boolean shouldBreak = false;
public void stop() {            shouldBreak = true;        }
boolean get() {return shouldBreak;        }    }
public static  void forEach(Stream stream, BiConsumer consumer) {        Spliterator spliterator = stream.spliterator();boolean hadNext = true;        Breaker breaker = new Breaker();
while (hadNext && !breaker.get()) {            hadNext = spliterator.tryAdvance(elem -> {                consumer.accept(elem, breaker);            });        }    }}
上面的例子中,我们在forEach中引入了一个外部变量,通过判断这个外部变量来决定是否进入spliterator.tryAdvance方法。

看下怎么使用:

@Slf4jpublic class CustomForEachUsage {     public static void main(String[] args) {         Stream ints = Stream.of(1, 2, 3, 4, 5, 6, 7, 8, 9, 10);          List result = new ArrayList<>();            CustomForEach.forEach(ints, (elem, breaker) -> {                  if (elem >= 5 ) {                           breaker.stop();                 } else {                     result.add(elem);                           }             });           log.info(result.toString());    }}
上面我们用新的forEach方法,并通过判断条件来重置判断flag,从而达到break stream的目的。

总结

本文通过两个具体的例子讲解了如何break一个stream,希望大家能够喜欢。

本文的例子github.com/ddean2009/l…

程序员专栏 扫码关注填加客服 长按识别下方二维码进群

近期精彩内容推荐:  

 程序员违规操作损失800万,被判5年半!

 “大多数人,都死在了 30 岁”

 实用,5个案例让 Python 输出漂亮的表格

 如何在分布式场景下生成全局唯一 ID ?

在看点这里好文分享给更多人↓↓

你可能感兴趣的:(Java8跳出循环新方式)