scalaでmutableなTreeMap - xuwei-k's blog

つくった

だれか #scala でmutableなSortedMapとかつくってたりしないのか・・・(´・ω・｀)　#java のやつのラッパーでもいいから・・・でもちゃんとcanBuildFrom実装してある的な・・・less than a minute ago via webkenji yoshida
xuwei_k

で、誰も反応してくれなかったので、とりあえず、ものすごく雑につくった

import collection.mutable
import java.{util => ju}

class TreeMap[A, B] (implicit private val ordering: Ordering[A]
  ) extends mutable.Map[A, B] {

  private val jTreeMap: ju.TreeMap[A, B] = new ju.TreeMap(ordering)

  override def -=(key: A) = {
    jTreeMap.remove(key)
    this
  }

  override def +=(kv: (A, B)) = {
    jTreeMap.put(kv._1,kv._2)
    this
  }

  override def iterator: Iterator[(A, B)] = new Iterator[(A,B)]{
    val ite = jTreeMap.entrySet.iterator

    def next = {
      val entry = ite.next
      (entry.getKey,entry.getValue)
    }

    def hasNext = ite.hasNext
  }

  override def get(key: A): Option[B] = Option(jTreeMap.get(key))

}