raggruppare gli elementi in un iterable cercando un valore sentinella (in scala)

Question

Ho un iteratore di righe da un file molto grande che deve essere inserito in gruppi mentre mi muovo. So dove finisce ogni gruppo perché c'è un valore sentinella sull'ultima riga di ogni gruppo. Quindi in pratica voglio scrivere una funzione che prende un iteratore e un valore sentinella, e restituisce un iteratore di gruppi ciascuno terminato dal valore sentinella. Qualcosa di simile:

scala> groups("abc.defg.hi.jklmn.".iterator, '.') 
res1: Iterator[Seq[Char]] = non-empty iterator 

scala> groups("abc.defg.hi.jklmn.".iterator, '.').toList 
res19: List[Seq[Char]] = List(List(a, b, c, .), List(d, e, f, g, .), List(h, i, .), List(j, k, l, m, n, .)) 

Nota che desidero includere gli elementi sentinella alla fine di ogni gruppo. Ecco la mia soluzione attuale:

def groups[T](iter: Iterator[T], sentinel: T) = new Iterator[Seq[T]] {     
    def hasNext = iter.hasNext 
    def next = iter.takeWhile(_ != sentinel).toList ++ List(sentinel) 
} 

penso che questo funzionerà, e credo che sia bene, ma di dover aggiungere nuovamente la sentinella ogni volta che mi dà un odore di codice. C'è un modo migliore per farlo?

Answer 1

Brutto, ma dovrebbe essere più performante rispetto la vostra soluzione:

def groups[T](iter: Iterator[T], sentinel: T) = new Iterator[Seq[T]] {     
    def hasNext = iter.hasNext 
    def next = iter.takeWhile{ 
     var last = null.asInstanceOf[T] 
     c => { val temp = last; last = c; temp != sentinel} 
    }.toList 
    } 

Answer 2

meno leggibile della tua, ma più "corretta" quando il gruppo finale non ha un valore di terminazione sentinella:

def groups[T](iter: Iterator[T], sentinel: T) = new Iterator[Seq[T]] { 
def hasNext = iter.hasNext 
def next: Seq[T] = { 
    val builder = scala.collection.mutable.ListBuffer[T]() 
    while (iter.hasNext) { 
     val x = iter.next 
     builder.append(x) 
     if (x == sentinel) return builder 
    } 
    builder 
} 
} 

Oppure, in modo ricorsivo:

def groups[T](iter: Iterator[T], sentinel: T) = new Iterator[Seq[T]] { 
    def hasNext = iter.hasNext 
    def next: Seq[T] = { 
     @scala.annotation.tailrec 
     def build(accumulator: ListBuffer[T]): Seq[T] = { 
     val v = iter.next 
     accumulator.append(v) 
     if (v == sentinel || !iter.hasNext) => accumulator 
     else build(accumulator) 
     } 
     build(new ListBuffer[T]()) 
    } 
    }