Расширение 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 .

0 ответов

Другие вопросы по тегам