Comment pouvez vous retourner un Itérateur en Scala?

Vous pouvez étendre Itérateur, ce qui nécessite de mettre en œuvre les next et hasNext méthodes:

  class MyAnswer extends Iterator[Int] {
    def hasNext = true
    def next = 42
  }

Mais, vous obtiendrez plus de flexibilité si vous étendez Itérable, qui vous oblige à mettre en œuvre elements (ou iterator en 2.8):

  class MyAnswer extends Iterable[Int] {
    def iterator = new Iterator[Int] {
      def hasNext = true
      def next = 42
    }
  }

Un idiome commun semble être à exposer un itérateur pour certains de la collection privée, comme ceci:

  class MyStooges extends Iterable[String] {
    private val stooges = List("Moe", "Larry", "Curly")
    def iterator = stooges.iterator
  }

Ces deux réponses ont bénéficié d'une aide dans les articles ci-dessous et merci @Dima.

• Comment puis-je mettre en œuvre un itérateur pour une seule liste liée?

• pourquoi est-ce itératif de mise en œuvre de produire un stackoverflow?

Supposons que vous avez une classe liste liée. Et que l'exigence soit pour imprimer tous les éléments dans la liste.

trait LinkedList {
  def nodeValue: Int
  def tailList: LinkedList
}

class Node(val nodeValue: Int, val tailList: LinkedList) extends LinkedList

object Nil extends LinkedList {
  def nodeValue = throw new IllegalAccessException("head of Nil")
  def tailList = throw new IllegalAccessException("tail of Nil")
}

val singleLinkedList = new Node(1,Nil)
val chainedLinkedList = new Node(2,singleLinkedList)
print(chainedLinkedList)
A$A44$A$A44$Node@7b7a2c78res0: Unit = ()

Maintenant passons à mettre en œuvre itérateur à cette classe.

trait LinkedList extends Iterator[Int]{
  def nodeValue: Int
  def tailList: LinkedList
}

class Node(val nodeValue: Int, val tailList: LinkedList) extends LinkedList {
  var ptr: LinkedList = this

  //The following two are mandatory for extending Iterator
  override def hasNext: Boolean = ptr match { case Nil => false; case _=> true}

  override def next(): Int = {
    val result = ptr.nodeValue
    ptr = ptr.tailList
    result
  }
}

object Nil extends LinkedList {
  def nodeValue = throw new IllegalAccessException("head of Nil")
  def tailList = throw new IllegalAccessException("tail of Nil")

  //The following two are mandatory for extending Iterator
  override def hasNext: Boolean = false
  override def next(): Int = throw new IllegalAccessException("next of Nil")
}

val singleLinkedList = new Node(1,Nil)
val chainedLinkedList = new Node(2,singleLinkedList)

//Printing this first Time
chainedLinkedList.foreach(println)
//Prints 2 1

//Printing second Time
chainedLinkedList.foreach(println)
//No output

Dans l'itérateur de la mise en œuvre, une fois ptr atteint la fin, il pourrait ne s'avance pas en arrière. Itératif de mise en œuvre résout ce.

trait LinkedList extends Iterable[Int]{
  val nodeValue: Int
  val tailList: LinkedList
  override def toString(): String = this.mkString(" -> ")
}

class Node(val nodeValue: Int, val tailList: LinkedList) extends LinkedList {

  override def iterator: Iterator[Int] = Iterator
    .iterate(this: LinkedList)(_.tailList)
    .takeWhile(_ != Nil)
    .map(_.nodeValue)
}

object Nil extends LinkedList {
  lazy val nodeValue= throw new IllegalAccessException("head of Nil")
  lazy val tailList = throw new IllegalAccessException("tail of Nil")

  override def iterator: Iterator[Int] = Iterator.empty
}

val singleLinkedList = new Node(1,Nil)
val chainedLinkedList = new Node(2,singleLinkedList)

//Printing this first Time
chainedLinkedList.foreach(println)
Output 2 -> 1
chainedLinkedList.foreach(println)
Output 2 -> 1