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

XTZM • 2024-11-07 09:50 • 编程

在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中的元素，例如对它们进行过滤、映射、累加等操作。

原创文章，作者：XTZM，如若转载，请注明出处：https://www.506064.com/n/150176.html