Erstellen Sie eine benutzerdefinierte Scala-Sammlung, in der standardmäßig die benutzerdefinierte Sammlung zurückgegeben wird?

Question

Das Merkmal TraversableLike[+A, +Repr] ermöglicht es, eine Sammlung zu erstellen, wobei einige Funktionen eine Repr zurückgeben, während andere weiterhin den Typ Parameter That auf die Funktion zurückgeben. Gibt es eine Möglichkeit, eine CustomCollection[A] zu definieren, in der Funktionen wie map, ++ und andere standardmäßig That als Repr werden, wenn nicht anders gedeutet?

Hier ist ein Code-Snippet, dass hoffentlich, was beschreibt Ich möchte:

case class CustomCollection[A](list: List[A]) extends TraversableLike[A, CustomCollection[A]] { 
    protected[this] def newBuilder = new CustomCollectionBuilder[A] 
    def foreach[U](f: (A) => U) {list foreach f} 
    def seq = list 
} 

class CustomCollectionBuilder[A] extends mutable.Builder[A, CustomCollection[A]] { 
    private val list = new mutable.ListBuffer[A]() 
    def += (elem: A): this.type = { 
    list += elem 
    this 
    } 
    def clear() {list.clear()} 
    def result(): CustomCollection[A] = CustomCollection(list.result()) 
} 

object CustomCollection extends App { 
    val customCollection = CustomCollection(List(1, 2, 3)) 
    println(customCollection filter {x => x == 1}) // CustomCollection(1) 
    println(customCollection map {x => x + 1}) // non-empty iterator 
} 

ich die letzte Zeile CustomCollection(2, 3, 4) sein möchte.

Answer 1

Sie benötigen einen Begleiter Objekt, das die raffinierte CanBuildFrom Instanz bietet einzurichten:

import collection.TraversableLike 
import collection.generic.{CanBuildFrom, GenericCompanion, GenericTraversableTemplate, 
    TraversableFactory} 
import collection.mutable.{Builder, ListBuffer} 

object CustomCollection extends TraversableFactory[CustomCollection] { 
    def newBuilder[A] = new CustomCollectionBuilder[A] 
    implicit def canBuildFrom[A]: CanBuildFrom[Coll, A, CustomCollection[A]] = 
    new CanBuildFrom[Coll, A, CustomCollection[A]] { 
     def apply(): Builder[A, CustomCollection[A]] = new CustomCollectionBuilder() 
     def apply(from: Coll): Builder[A, CustomCollection[A]] = apply() 
    } 
} 
case class CustomCollection[A](list: List[A]) extends Traversable[A] 
with TraversableLike[A, CustomCollection[A]] 
with GenericTraversableTemplate[A, CustomCollection] { 
    override def companion: GenericCompanion[CustomCollection] = CustomCollection 
    def foreach[U](f: A => U) { list foreach f } 
    override def seq = list 
} 

class CustomCollectionBuilder[A] extends Builder[A, CustomCollection[A]] { 
    private val list = new ListBuffer[A]() 
    def += (elem: A): this.type = { 
    list += elem 
    this 
    } 
    def clear() {list.clear()} 
    def result(): CustomCollection[A] = CustomCollection(list.result()) 
} 

val customCollection = CustomCollection(List(1, 2, 3)) 
val f = customCollection filter {x => x == 1} // CustomCollection[Int] 
val m = customCollection map {x => x + 1}  // CustomCollection[Int]