Расширение Custom Collection на Set в Scala
Это продолжение ответа на следующий вопрос:как создать пользовательскую коллекцию, которая расширяет наборы в Scala?
У меня есть базовый класс, который распространяется наSet
коллекция и большинство функций, таких какfilter
,collect
и другие работают:
import scala.collection._
class TestSet[A]
extends mutable.Set[A]
with mutable.SetOps[A, mutable.Set, TestSet[A]] {
private var data = mutable.Set.empty[A]
override protected def fromSpecific(coll: IterableOnce[A]): TestSet[A] = TestSet.from(coll)
override protected def newSpecificBuilder: mutable.Builder[A, TestSet[A]] = TestSet.newBuilder
override def empty: TestSet[A] = TestSet.empty
override def clear(): Unit = {
data = mutable.Set.empty
}
override def subtractOne(elem: A): TestSet.this.type = {
data -= elem
this
}
override def addOne(elem: A): TestSet.this.type = {
data += elem
this
}
override def contains(elem: A): Boolean = data.contains(elem)
override def iterator: Iterator[A] = {
data.iterator
}
}
object TestSet {
def empty[A] = new TestSet[A]
def from[A](source: IterableOnce[A]): TestSet[A] =
source match {
case pm: TestSet[A] => pm
case _ => (newBuilder ++= source).result()
}
def apply[A](elem: A*): TestSet[A] = from(elem)
def newBuilder[A]: mutable.Builder[A, TestSet[A]] =
new mutable.GrowableBuilder[A, TestSet[A]](empty)
import scala.language.implicitConversions
implicit def toFactory[A](self: this.type): Factory[A, TestSet[A]] =
new Factory[A, TestSet[A]] {
def fromSpecific(it: IterableOnce[A]): TestSet[A] = self.from(it)
def newBuilder: mutable.Builder[A, TestSet[A]] = self.newBuilder
}
}
Это может быть невозможно, но я изначально хотел SetOps со следующими типамиmutable.SetOps[A, TestSet, TestSet[A]]
вместоmutable.SetOps[A, mutable.Set, TestSet[A]]
, но у меня возникли проблемы с переопределениемiterableFactory
.
Рабочий фрагмент можно найти здесь: https://scastie.scala-lang.org/4Qi8ZueDTpScDwxDfqptew .