Kotlin: если HashMap не содержит определенного значения, создайте ключ

Question

У меня есть следующий HashMap:

val myHashMap: HashMap<key, ValueClass>

где:

data class ValueClass(
    var a: Long = 0,
    var b: Long = 0
)

Я хотел бы использовать myHashMap следующим образом: если myHashMap не содержит ключ x, создайте новый ValueClass и вставьте его в myHashMap, а затем myHashMap [x] .a = newValue

каков самый чистый способ сделать это? В Java у меня было бы:

    if(!myHashMap.containsKey("x"))
    {
        map.put("x",new ValueClass());
    }
    map.get("x").a=newValue;

Answer 1

Использование MutableMap # getOrPut

myMap.getOrPut("key") { "default_value" }

Answer 2

Другая альтернатива:

val foo = map["x"] ?: ValueClass().also { map["x"] = it }

foo.x = newValue

Answer 3

Вы можете использовать run() для создания значения, если оно не существует.

val foo = map["x"] ?: run {
    val bar = ValueClass()
    map["x"] = bar
    bar
}

foo.x = newValue

Это приведет к извлечению map.get("x") при его наличии, в противном случае будет оцениваться блок run. Блок создает ваш новый экземпляр, помещает его в карту и «возвращает» новый экземпляр.

Следовательно, foo всегда ненулевой, и вы можете обновить его.