Scala流处理

Question

我使用streams在scala中编写了以下代码

def foo(x: Int) : Stream[Int] = {println("came inside"); (2 * x)} #:: foo(x + 1)
foo(1).takeWhile(_ < 6).toList.foreach(x => println(s"+++ $x"))

这将工作并产生以下输出

came inside
came inside
came inside
+++ 2
+++ 4

但我希望处理过程如下所示

came inside
+++ 2
came inside
+++ 4
came inside

基本上是一个接一个的处理，直到满足<6的终止条件。我猜这是我的"tolist“方法，它首先创建一个巨大的列表，然后才处理它。

Answer 1

首先，以一种更具可读性的方式格式化代码。然后，删除toList：它所做的就是将整个流拉入一个变量中。这样做会强制计算所有的值。因为每个值都是在这一点上计算出来的，所以我们知道“内部”println将在“外部”println之前执行。您希望将函数的定义(包括它的起始值)存储为延迟计算的值。这应该是可行的：

def river(x: Int) : Stream[Int] = {
  println("Inside function.")
  (2 * x) #:: river(x + 1)
}

lazy val creek = river(1)
creek.takeWhile(_ < 6).foreach(x => println(s"+++ $x"))