Как реализовать Map с операцией по умолчанию в Scala - PullRequest
Новая
       14

Как реализовать Map с операцией по умолчанию в Scala

40 голосов
/ 29 апреля 2011 
class DefaultListMap[A, B <: List[B]] extends HashMap[A, B] {
    override def default(key: A) = List[B]() 
  }

Я не хочу создавать карту A -> List[B]. В моем случае это Long -> List[String], но когда я получаю ключ от карты, у которой нет значения, я бы хотел создать пустую List вместо брошенной Exception. Я пробовал разные комбинации, но я не знаю, как заставить код выше передать компилятору.

Заранее спасибо.

Ответы [ 5 ]

96 голосов
/ 29 апреля 2011

Почему бы не использовать withDefaultValue (значение)? 

scala> val m = Map[Int, List[String]]().withDefaultValue(List())
m: scala.collection.immutable.Map[Int,List[String]] = Map()

scala> m(123)
res1: List[String] = List()
22 голосов
/ 29 апреля 2011

Вместо того, чтобы использовать apply для доступа к карте, вы всегда можете использовать get, который возвращает Option[V], а затем getOrElse: 

map.get(k) getOrElse Nil

Одной замечательной особенностью scalaz библиотеки функционального программирования является унарный оператор ~, который означает «или ноль», если тип значения имеет «ноль». "определено (что List делает, ноль, конечно, Nil). Таким образом, код становится: 

~map.get(k)

Это вдвойне полезно, потому что тот же синтаксис работает там, где (например) ваши значения Int, Double и т. Д. (Все, для чего существует класс типов Zero).

<Ч />

В списке рассылки scala было много споров об использовании Map.withDefault из-за того, как он ведет себя, в частности, в отношении метода isDefinedAt. По этой причине я стараюсь держаться подальше от этого.

11 голосов
/ 29 апреля 2011

Есть метод withDefaultValue на Map: 

scala> val myMap = Map(1 -> List(10), 2 -> List(20, 200)).withDefaultValue(Nil)
myMap: scala.collection.immutable.Map[Int,List[Int]] = Map((1,List(10)), (2,List(20, 200)))

scala> myMap(2)
res0: List[Int] = List(20, 200)

scala> myMap(3)
res1: List[Int] = List()
3 голосов
/ 29 апреля 2011

Почему вы хотите манипулировать картой, когда у нее уже есть метод для этого? 

val m = Map(1L->List("a","b"), 3L->List("x","y","z"))  
println(m.getOrElse(1L, List("c"))) //--> List(a, b)
println(m.getOrElse(2L, List("y"))) //--> List(y)
2 голосов
/ 05 апреля 2017

withDefault также можно использовать. 

/** The same map with a given default function.
 *  Note: `get`, `contains`, `iterator`, `keys`, etc are not affected
 *  by `withDefault`.
 *
 *  Invoking transformer methods (e.g. `map`) will not preserve the default value.
 *
 *  @param d     the function mapping keys to values, used for non-present keys
 *  @return      a wrapper of the map with a default value
 */
 def withDefault[B1 >: B](d: A => B1): immutable.Map[A, B1]

Пример: 

scala> def intToString(i: Int) = s"Integer $i"
intToString: (i: Int)String

scala> val x = Map[Int, String]().withDefault(intToString)
x: scala.collection.immutable.Map[Int,String] = Map()

scala> x(1)
res5: String = Integer 1

scala> x(2)
res6: String = Integer 2

Надеюсь, это поможет.

...