怎么构建无限的stream

程序那些事

发布于 2020-07-08 10:23:43

3220

发布于 2020-07-08 10:23:43

文章被收录于专栏：程序那些事程序那些事

简介

在java中，我们可以将特定的集合转换成为stream，那么在有些情况下，比如测试环境中，我们需要构造一定数量元素的stream，需要怎么处理呢？

这里我们可以构建一个无限的stream，然后调用limit方法来限定返回的数目。

基本使用

先看一个使用Stream.iterate来创建无限Stream的例子：

 @Test
    public void infiniteStream(){
        Stream<Integer> infiniteStream = Stream.iterate(0, i -> i + 1);
        List<Integer> collect = infiniteStream                .limit(10)
                .collect(Collectors.toList());
        log.info("{}",collect);
    }

上面的例子中，我们通过调用Stream.iterate方法，创建了一个0，1，2，3，4….的无限stream。

然后调用limit(10)来获取其中的前10个。最后调用collect方法将其转换成为一个集合。

看下输出结果：

INFO com.flydean.InfiniteStreamUsage - [0, 1, 2, 3, 4, 5, 6, 7, 8, 9]

自定义类型

如果我们想输出自定义类型的集合，该怎么处理呢？

首先，我们定义一个自定义类型：

@Data
@AllArgsConstructor
public class IntegerWrapper {
    private Integer integer;}

然后利用Stream.generate的生成器来创建这个自定义类型：

 public static IntegerWrapper generateCustType(){
        return new IntegerWrapper(new Random().nextInt(100));
    }

 @Test
    public void infiniteCustType(){
        Supplier<IntegerWrapper> randomCustTypeSupplier = InfiniteStreamUsage::generateCustType;
        Stream<IntegerWrapper> infiniteStreamOfCustType = Stream.generate(randomCustTypeSupplier);

        List<IntegerWrapper> collect = infiniteStreamOfCustType                .skip(10)
                .limit(10)
                .collect(Collectors.toList());
        log.info("{}",collect);
    }

看下输出结果：

INFO com.flydean.InfiniteStreamUsage - [IntegerWrapper(integer=46), IntegerWrapper(integer=42), IntegerWrapper(integer=67), IntegerWrapper(integer=11), IntegerWrapper(integer=14), IntegerWrapper(integer=80), IntegerWrapper(integer=15), IntegerWrapper(integer=19), IntegerWrapper(integer=72), IntegerWrapper(integer=41)]

总结

本文介绍了两个生成无限stream的例子。

欢迎关注我的公众号:程序那些事，更多精彩等着您！

本文参与腾讯云自媒体分享计划，分享自微信公众号。

原始发表：2020-04-26，如有侵权请联系 cloudcommunity@tencent.com 删除

java