Обновить измененное значение HashMap, которое является изменяемой коллекцией

Question

У меня есть карта, которая выглядит так: Map[ A -> Collection[B]]. Эта карта обновляется в цикле - особенно важно, что обновление в основном означает просто добавление элемента B в коллекцию [B] (для некоторой клавиши A).

Я пытаюсь выяснить, могу ли я получить ускорение, изменив тип моей коллекции из списка [] в ListBuffer [].

До сих пор мой код выглядит следующим образом (упрощенно):

var incoming = new HashMap[A, List[B]() { 
    override def default(a: A) = List() 
} 
.. 
for(b < someCollectionOfBs){ 
    .. 
    incoming(b.getA) = b :: incoming(b.getA) 
    .. 
} 

Это прекрасно работает. Теперь я изменил тип карты, так это выглядит следующим образом:

var incoming = new collection.mutable.HashMap[A, ListBuffer[B]() { 
    override def default(a: A) = collection.mutable.ListBuffer() 
} 
.. 
for(b < someCollectionOfBs){ 
    .. 
    incoming(b.getA) += b 
    .. 
} 

Обратите внимание на изменение в том, как элемент B добавляется в коллекцию в 2-ом примере (не более неизменного списка, следовательно, нам не нужно создавать и назначать новую коллекцию ...).

Но. Это не работает: incoming(X) += .. делает не обновляет значение карты для X, фактически ничего не меняет.

Что мне здесь не хватает? Я думал, что мне нужно будет обновить значения изменчивого HashMap ... Итак, если мои значения являются изменяемыми коллекциями, почему я не могу просто добавлять к ним элементы?

Answer 1

Значение по умолчанию возвращается, когда ключ не найден, но он не обновляет карту со значением по умолчанию. Вы можете использовать для этого getOrElseUpdate.

incoming.getOrElseUpdate(b.getA, ListBuffer()) += b 

Это должно делать то, что вы хотите.

Дополнительное примечание:

Если вы беспокоитесь о производительности, я не думаю, что замена List с ListBuffer будет покупать вам много, потому что вы предваряя к List, и это должно быть очень быстрым. ListBuffer удобен, если вы хотите добавить в список. Вы должны посмотреть на использование java.util.HashMap и посмотреть, поможет ли это.