Получение правильного итератора

Question

Целью первого фрагмента кода ниже является определение нового признака для iterables, который предоставляет дополнительный метод для добавления нового элемента перед итерируемым.

Однако при запуске кода во втором фрагменте мы видим, что метод +: возвращает итерабельность, дающую бесконечность 0 s.

Что я делаю неправильно и как я могу получить предполагаемое поведение?

Примечание: я добавил outer val, чтобы убедиться, что получаю правильный итератор при определении методов объекта, возвращаемого +:; Я не знаю, как получить доступ к этому итератору иначе (Iterable2.this.iterator не скомпилировал).

trait Iterable2[A] extends Iterable[A] { 
    val outer :Iterable[A] = this 
    def +:(elem :A) = new Iterable2[A] { 
    override def iterator: Iterator[A] = new Iterator[A] { 
     private[this] var virgin = true 
     override def hasNext: Boolean = virgin || outer.iterator.hasNext 
     override def next(): A = { 
     if (virgin) {virgin = false; elem} 
     else outer.iterator.next() 
     } 
    } 
    } 
} 

val i = new Iterable2[Int] { 
    override def iterator: Iterator[Int] = Iterator(1,2,3) 
} 

for (j <- 0 +: i) { 
    println(j) 
} 

Answer 1

Первая ошибка в моем исходном коде является то, что outer.iterator возвращает новый итератор каждый раз, когда он оцененную , как указано @DiegoMartinoia и @Imm.

Однако, также необходимо заменить val outer = this на outer =>, как предложено @misberner.

На данный момент это лишь частичный ответ на мой вопрос, поскольку он не объясняет, почему требуется второе изменение.

Исправленный код черты таким образом:

trait Iterable2[A] extends Iterable[A] { 
    outer :Iterable[A] => 
    def +:(elem :A) = new Iterable2[A] { 
    override def iterator: Iterator[A] = new Iterator[A] { 
     private[this] var virgin = true 
     private[this] val underlyingIterator = outer.iterator 
     override def hasNext: Boolean = virgin || underlyingIterator.hasNext 
     override def next(): A = { 
     if (virgin) {virgin = false; elem} 
     else underlyingIterator.next() 
     } 
    } 
    } 
} 

Answer 2

outer.iterator всегда даст вам новый итератор. Вам нужно создать и копить его где-нибудь, а затем использовать этот единственный спрятали один, а не создавать новую каждый раз, когда:

new Iterator[A] { 
    val outerIterator = outer.iterator 
    override def hasNext = ... 
}