使用具体类型扩展 Scala Set

Question

真的很难想出用一个代表具体类型 Set 的类来扩展不可变 Set。我这样做是为了尝试创建一个不错的 DSL。

我想要一个 Thing 类，当你将“事物”加在一起时，你会得到一个扩展 Set 的 ThingSet 对象。

class Thing(val name:String){
  def +(other: Thing):ThingSet = new ThingSet() + other
}

我只是不知道如何制作 ThingSet 对象。我知道我需要混合 GenericSetTemplate、SetLike 等特征。但我就是做不到。

请，任何人都可以给我一些指示，因为我找不到任何足够明确的东西来学习。我试过查看 BitSet 和 HashSet 实现，但迷路了。

Answer 1

改编自this Daily Scala 的帖子以及BitSet 的源代码，这是一个很好的例子，因为它不是一个参数化的集合，而且相当简短。

import scala.collection.SetLike
import scala.collection.generic.{GenericSetTemplate, GenericCompanion, CanBuildFrom}
import scala.collection.mutable.{Builder, SetBuilder}

class ThingSet(seq : Thing*) extends Set[Thing] 
                             with SetLike[Thing, ThingSet]
                             with Serializable {
    override def empty: ThingSet = new ThingSet()
    def + (elem: Thing) : ThingSet = if (seq contains elem) this 
        else new ThingSet(elem +: seq: _*)
    def - (elem: Thing) : ThingSet = if (!(seq contains elem)) this
        else new ThingSet(seq filterNot (elem ==): _*)
    def contains (elem: Thing) : Boolean = seq exists (elem ==)
    def iterator : Iterator[Thing] = seq.iterator
}

object ThingSet {
    def empty: ThingSet = new ThingSet()
    def newBuilder: Builder[Thing, ThingSet] = new SetBuilder[Thing, ThingSet](empty)
    def apply(elems: Thing*): ThingSet = (empty /: elems) (_ + _)
    def thingSetCanBuildFrom = new CanBuildFrom[ThingSet, Thing, ThingSet] {
        def apply(from: ThingSet) = newBuilder
        def apply() = newBuilder
    }
}

Answer 2

从 Set 上的 scaladocs 来看，我认为您只需要实现 4 或 5 个方法。在这种情况下，我决定使用支持 Seq[Thing] 来创建我的实际集合实现。

class ThingSet(things: Seq[Thing]) extends Set[Thing] {
  def contains(key: Thing) = { things.contains(key) }
  def iterator: Iterator[Thing] = { things.iterator }
  def +(elem: Thing) = new ThingSet(things :+ elem)
  def -(elem: Thing) = new ThingSet(things.filterNot(_ == elem))
  override def empty = new ThingSet(Nil)
}

class Thing(val name: String) {
  def +(other: Thing) = { new ThingSet(List(this,other)) }
  override def toString = name
}

val thing = new Thing("I'm a new thing")
println(thing + new Thing("I'm some other thing"))

结果：

Set(I'm a new thing, I'm some other thing)

注意：由于 Thing 和 ThingSet 之间的循环依赖关系，我认为您必须在脚本文件而不是 REPL 中运行它。