Как создать конечный итератор с содержимым, являющимся результатом выражения?

Question

Я хотел бы создать Iterator, который получает следующий элемент путем (многократного) вычисления выражения, и я хочу, чтобы выражение могло возвращать определенное значение для его завершения.

Единственное, что я нашел, это Iterator.continually () , что кажется бесконечным. Важно, чтобы выражение не оценивалось до тех пор, пока next() не будет вызвано для Iterator.

Есть ли способ получить такое поведение?

например:

def getNext = {
  // some complicated code
  val next = ... // either a STOP value or a real value to be returned by the iterator
} 

val myIter = Iterator.continually(getNext) // want this to stop at some point

Answer 1

Iterator.continually обычно сочетается с takeWhile:

var count = 0
def complexCompute(): Int = { count +=1; println("eval " + count); count }

val iter = Iterator.continually { complexCompute() }
iter.takeWhile(_ < 3).foreach(println)

Какие отпечатки:

eval 1
1
eval 2
2
eval 3

Таким образом, если условие, которое определяет, следует ли продолжить вычисление, может быть оценено за пределами вычисления, то это работает довольно хорошо.

В основном, я думаю, я говорю, что Iterator.continually(getNext()).takeWhile(_ != certainValue) достигнет того, что вы пытаетесь сделать. Это лениво оценивается.

Answer 2

Вы смотрели на scala.collection.immutable.Stream ?Его цель - создать последовательность, подобную объекту, где лениво оценивается следующий элемент.Может быть конечным или бесконечным.

Например:

Welcome to Scala version 2.9.0.final (Java HotSpot(TM) Client VM, Java 1.6.0_24).
Type in expressions to have them evaluated.
Type :help for more information.

scala> import collection.immutable._
import collection.immutable._

scala> def next(i: Int): Stream[Int] = Stream.cons(i*i, next(i*i))
next: (i: Int)scala.collection.immutable.Stream[Int]

scala> val stream = next(2)
stream: scala.collection.immutable.Stream[Int] = Stream(4, ?)

scala> stream.find(_ > 1000)
res0: Option[Int] = Some(65536)