mutable.Map deep merge
Czy istnieje zwięzły sposób na głębokie scalenie dwóch zmiennych map w Scali?
case class K1(i: Int)
case class K2(i: Int)
def deepMerge(map: mutable.Map[K1, Map[K2, List[Int]]],
mergee: mutable.Map[K1, Map[K2, List[Int]]]
): Unit = ???
Przykłady:
JA.
val map = mutable.Map(K1(1) -> Map(K2(1) -> List(1)))
val mergee = mutable.Map(K1(1) -> Map(K2(1) -> List(2)))
deepMerge(map, mergee)
map = mutable.Map(K1(1) -> Map(K2(1) -> List(1, 2)))
II.
val map = mutable.Map(K1(1) -> Map(K2(1) -> List(1)))
val mergee = mutable.Map(K1(1) -> Map(K2(2) -> List(1)))
deepMerge(map, mergee)
map = mutable.Map(K1(1) -> Map(K2(1) -> List(1), K2(2) -> List(1)))
III.
val map = mutable.Map(K1(1) -> Map(K2(1) -> List(1)))
val mergee = mutable.Map(K1(2) -> Map(K2(2) -> List(1)))
deepMerge(map, mergee)
map = mutable.Map(K1(1) -> Map(K2(1) -> List(1)), K1(2) -> Map(K2(2) -> List(1)))
To znaczy, jeśli w obu mapach jest przedstawiony ten sam klucz, wówczas wartości, którym odpowiadają klucze ( List[Int]
), są scalane.
Czy istnieje sposób, aby zaimplementować to w zwięzły sposób, unikając wielu sprawdzania, czy dany klucz jest prezentowany, czy nie na innej mapie? Używanie bibliotek FP, takich jak scalaz lub cats, jest również w porządku.
Odpowiedzi
Dodaję kolejną odpowiedź za pomocą kotów.
To, co opisujesz, to w rzeczywistości zachowanie kotów . Możesz więc po prostu użyć |+|
operatora connect ( ) do głębokiego scalania map:
import cats.implicits._
import cats._
case class K1(i: Int)
case class K2(i: Int)
val map = Map(K1(1) -> Map(K2(1) -> List(1)))
val mergee = Map(K1(1) -> Map(K2(1) -> List(2)))
val deepMerged = map |+| mergee
println(deepMerged) // HashMap(K1(1) -> HashMap(K2(1) -> List(1, 2)))
Problem polega na tym, że cats lib nie dostarcza instancji Semigroup dla mutable.Map
, ale możesz ją wyprowadzić z jednej dla niezmienności:
import cats.implicits._
import scala.collection.immutable
import scala.collection.mutable
import cats._
//here I derivive Semigroup instance for mutable.Map from instance for immutable.Map
implicit def mutableMapSemigroup[K, V: Semigroup]: Semigroup[mutable.Map[K, V]] = Semigroup[immutable.Map[K, V]].imap(c => mutable.Map.from(c))(c => immutable.Map.from(c))
case class K1(i: Int)
case class K2(i: Int)
val map = mutable.Map(K1(1) -> mutable.Map(K2(1) -> List(1)))
val mergee = mutable.Map(K1(1) -> mutable.Map(K2(1) -> List(2)))
println(map |+| mergee)
Pamiętaj jednak, że to faktycznie konwertuje mapę mutowalną na mapę niezmienną, a następnie łączy się, a następnie konwertuje z powrotem na mapę mutowalną, więc prawdopodobnie nie jest zbyt wydajna.
To może wystarczyć.
def deepMerge(mergeA: Map[K1, Map[K2, List[Int]]],
mergeB: Map[K1, Map[K2, List[Int]]]
): Map[K1,Map[K2,List[Int]]] =
(mergeA.toList ++ mergeB.toList).groupMap(_._1)(_._2).map{
case (k1,ms) =>
k1 -> ms.flatMap(_.toList).groupMap(_._1)(_._2).map{
case (k2,ls) => k2 -> ls.flatten
}
}
Nie testowałem tego z mutable
Mapami, ale powinno działać mniej więcej tak samo.