使用新属性扩展 scala 集合

Question

如果我有一个带有附加属性的新集合，例如：

class NewColl(var name: String, var values: Vector[Int])
    extends IndexedSeq[Int] with IndexedSeqLike[Int, NewColl] {...} 

如何定义 canBuildFrom 和 newBuilder （参见 此处）这样，如果：

var v1 = NewColl("foo",Vector(1,2,3))
var v2 = v1 filter(_ > 1)
var v3 = v1 map(_ + 1)

则 v2.name=="foo"和v3.name==“foo”？

Answer 1

试试这个：

class NewColl(val name: String, val values: Vector[Int])
    extends IndexedSeq[Int] with IndexedSeqLike[Int, NewColl] with SeqForwarder[Int] {
  override def newBuilder: Builder[Int, NewColl] = NewColl.newBuilder(name)
  protected override def underlying = values
}

object NewColl {
  def apply(name: String, elems: Vector[Int]) = new NewColl(name, elems)
  implicit def canBuildFrom: CanBuildFrom[NewColl, Int, NewColl] = {
    new CanBuildFrom[NewColl, Int, NewColl] {
      def apply(from: NewColl) = from.newBuilder
      def apply() = newBuilder(defaultName)
    }
  }
  private def newBuilder(name: String) = Vector.newBuilder[Int].mapResult(vector => new NewColl(name, vector))
  private def defaultName: String = error("no default name")
}

请注意，我已将您的 var 更改为 val，以便与不可变集合保持一致。已混合使用 SeqForwarder 以避免实现将全部转发到 values 上的同一方法的方法列表。 newBuilder 已在伴生对象上实现，需要一个 String 参数。

请注意，有时，集合可以自由地在 CanBuildFrom 上调用 apply()，而无需指定原始集合。在这种情况下，您要么必须提供默认名称，要么（像这里一样）抛出异常（仅当您不设计库并控制新集合的使用时才允许）。

另请参阅该问题。