在Scala中连接TraversableOnce对象?

时间:2012-10-25 16:06:42

标签: scala iterator concatenation

我希望将可遍历的一次连接到可遍历的一次而不解决任何一次。 这是我提出的一个隐含的解决方案,但我不知道我是否缺少原生解决方案......

object ImplicitTraversableOnce {
  implicit def extendTraversableOnce[T](t : TraversableOnce[T]) = new TraversableOnceExtension(t)
}

class TraversableOnceExtension[T <: Any](t : TraversableOnce[T]) {

  def ++ (t2:TraversableOnce[T]):TraversableOnce[T] = new concat(t.toIterator, t2.toIterator)

  private class concat(i1:Iterator[T], i2:Iterator[T]) extends Iterator[T] {
    private var isOnSecond = false

    def hasNext:Boolean =
      if (isOnSecond) i2.hasNext
      else if (!i1.hasNext) {
        isOnSecond = true
        hasNext
      }
    else true

    def next():T = if (isOnSecond) i2.next() else i1.next()
  }
}

2 个答案:

答案 0 :(得分:6)

您可以使用++加入迭代器。所以如果你打算使用迭代器,只需

def ++(t2: TraversableOnce[T]): TraversableOnce[T] = t.toIterator ++ t2.toIterator

不这样做的原因是为不foreach / Traversable的{​​{1}}提供有效的Iterable,但是您需要填写所有Iterator抽象方法。

答案 1 :(得分:3)

怎么样:

  def ++(t2: TraversableOnce[T]) = new Traversable[T] {
    def foreach[U](f: (T) => U) {
      t.foreach(f)
      t2.foreach(f)
    }
  }