您的位置:

深入了解Java 8中的stream.generate()

在Java 8中,stream.generate()是一个非常强大的函数,它可以让我们创建一个无限或有限的stream,这个stream会得益于一个提供了无限元素的Supplier。在这篇文章中,我们将会从多个方面对stream.generate()进行详细的阐述。

一、创建无限stream

Stream.generate(() -> "random").forEach(System.out::println);

如上述代码所示,使用stream.generate()函数来创建一个无限的stream,传入一个lambda表达式,这个lambda表达式的功能是返回一个字符串"random",然后我们使用forEach()函数来打印出stream中的每一个元素。因为Supplier是无限提供元素的,所以这个stream会一直运行下去,直到程序强制终止为止。

当我们创建无限stream时,谨慎使用limit()函数是非常重要的。如果我们没有使用limit()函数来限制stream元素的数量,那么stream会无休止地运行下去,这会占用很多内存空间并导致程序崩溃。

二、创建有限stream

Stream.generate(Math::random).limit(5).forEach(System.out::println);

如上述代码所示,我们可以使用limit()函数来限制stream元素的数量,这里我们创建了一个有限stream,限制stream有5个元素。这个stream的元素是使用Math.random()函数生成的随机数字。运行结果如下:

0.13019742173165957
0.19813332522974886
0.392186707372399
0.585466685795064
0.18830230523116438

三、创建自定义Supplier的stream

Stream.generate(new MySupplier()).limit(5).forEach(System.out::println);
class MySupplier implements Supplier<String> {
    private int i = 0;
    @Override
    public String get() {
        i++;
        return "String number " + i;
    }
}

如上述代码所示,我们可以自定义一个Supplier,并传入stream.generate()函数中。在这个例子中,我们的MySupplier类实现了Supplier接口,并且使用了一个简单的计数器,每次调用get()函数时返回一个新的字符串。运行结果如下:

String number 1
String number 2
String number 3
String number 4
String number 5

四、对stream元素进行处理

int sum = Stream.generate(() -> 1).limit(10).mapToInt(Integer::intValue).sum();
System.out.println(sum);

如上述代码所示,我们可以对stream中的元素进行一系列处理,比如使用mapToInt()函数将stream中的元素转换为整数类型,并使用sum()函数将所有元素求和。在这个例子中,我们创建了10个值为1的元素的stream,并对这些元素进行了求和操作。运行结果是10。

五、使用流来创建集合

List<Integer> list = Stream.generate(() -> 1).limit(5).collect(Collectors.toList());
System.out.println(list);

如上述代码所示,我们可以使用collect()函数来将stream中的元素收集到一个集合中。在这个例子中,我们创建了5个值为1的元素的stream,并使用Collectors.toList()函数将这些元素收集到了一个List集合中。运行结果是[1, 1, 1, 1, 1]。

总之,在Java 8中,stream.generate()是一个功能强大且灵活的工具,它可以用于创建无限stream,也可以用于创建有限stream,甚至可以用于创建自定义的stream。我们可以通过一系列函数来处理这些stream中的元素,例如对它们进行过滤、映射、累加等操作。