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