Mappa con Default di un ListBuffer vuoto in scala

Question

Sto tentando di creare un Map mutabile con un valore predefinito che crea un nuovo ListBuffer quando viene richiesto un elemento che non è già nella mappa. Tuttavia, mentre la nuova mappa viene restituita come predefinita, non rimane nella mappa. Forse è proprio così che funziona, ho pensato, ma quando l'ho testato con uno Int anziché con uno ListBuffer ha fatto esattamente quello che volevo. Ecco un codice per spiegare cosa intendo - cosa sto sbagliando?

In primo luogo, qui si sta lavorando con un Map[Int]:

scala> val a = collection.mutable.Map(1 -> 1).withDefault(i => 0) 
a: scala.collection.mutable.Map[Int,Int] = Map(1 -> 1) 

scala> a(1) += 1 // adding to an existing element works as expected 

scala> a 
res48: scala.collection.mutable.Map[Int,Int] = Map(1 -> 2) 

scala> a(2) += 1 // what about adding to a non-existing element? 

scala> a // the new element has been added to the map 
res50: scala.collection.mutable.Map[Int,Int] = Map(1 -> 2, 2 -> 1) 

Ora, con un Map[ListBuffer[Int]]:

scala> val b = collection.mutable.Map(1 -> collection.mutable.ListBuffer[Int]()).withDefault(i => collection.mutable.ListBuffer.empty[Int]) 
b: scala.collection.mutable.Map[Int,scala.collection.mutable.ListBuffer[Int]] = Map(1 -> ListBuffer()) 

scala> b(1) += 1 // appending to an existing element works as expected 
res51: scala.collection.mutable.ListBuffer[Int] = ListBuffer(1) 

scala> b 
res52: scala.collection.mutable.Map[Int,scala.collection.mutable.ListBuffer[Int]] = Map(1 -> ListBuffer(1)) 

scala> b(2) += 1 // but appending to a non-existing element... 
res53: scala.collection.mutable.ListBuffer[Int] = ListBuffer(1) 

scala> b // leaves the map unchanged 
res54: scala.collection.mutable.Map[Int,scala.collection.mutable.ListBuffer[Int]] = Map(1 -> ListBuffer(1)) 

Answer 1

La differenza è questa:

Nel primo caso a(2) è un Int . Poiché Int non ha un metodo +=, a(2) += 1 equivale a a(2) = a(2) + 1 e quindi a a.update(2, a(2) + 1). Lo update modifica effettivamente la mappa.

Ma ListBuffer[Int]fa hanno un metodo di +=, quindi la chiamata viene a(2).+=(1), e non si imposta a(2) a nulla!

Answer 2

È possibile utilizzare getOrElseUpdate(key: A, op: => B) in cui è possibile creare una nuova istanza di ListBuffer quando la chiave non è presente.

E.g.

val m = collection.mutable.Map[Int, ListBuffer[Int]]() 
m.getOrElseUpdate(1, ListBuffer()) += 1