斯卡拉 - 與Iterator.toList

Question

我是新來Scala和我有一個函數如下怪異的行爲：

def selectSame(messages: BufferedIterator[Int]) = { 
    val head = messages.head 
    messages.takeWhile(_ == head) 
} 

這是從一個緩衝的迭代器只選擇相匹配的頭elems的。我隨後使用此代碼：

val messageStream = List(1,1,1,2,2,3,3) 
if (!messageStream.isEmpty) { 
    var lastTimeStamp = messageStream.head.timestamp 
    while (!messageStream.isEmpty) { 
    val messages = selectSame(messageStream).toList 
    println(messages) 
} 

一旦如預期第一執行我得到（1,1,1），但後來我只得到列表（2），就像如果我失去了一個元素向下行...可能我正在用迭代器/列表做錯，但我在這裏有點失落。

Answer 1

的Iterator Scaladoc說，大約takeWhile：

重用：調用此方法後，應該丟棄 拜訪了迭代器，並只使用返回迭代器。使用 舊迭代器未定義，可能會更改，也可能導致 更改爲新迭代器。

所以這就是爲什麼。這基本上意味着你不能直接做你想用Iterators和takeWhile。恕我直言，最簡單的辦法就是快速編寫自己的遞歸函數來做到這一點。

如果您想堅持使用Iterators，您可以使用Iterator上的sameElements方法生成重複的地方，您可以撥打dropWhile。

更妙的是：使用span反覆：

def selectSame(messages: BufferedIterator[Int]) = { 
    val head = messages.head 
    messages.span(_ == head) 
} 

def iter(msgStream: BufferedIterator[Int]): Unit = if (!msgStream.isEmpty) { 
    val (msgs, rest) = selectSame(msgStream) 
    println(msgs.toList) 
    iter(rest) 
} 

val messageStream = List(1,1,1,2,2,3,3) 
if (!messageStream.isEmpty) { 
    var lastTimeStamp = messageStream.head.timestamp 
    iter(messageStream0 
}