不可空的可变地图

时间:2016-10-12 18:56:23

标签: kotlin kotlin-extension

  • Java:1.8.0_102
  • Kotlin:1.0.4

我正在尝试创建一个地图,您可以在其中执行类似于javascript的map["key"] += 5

Kotlin已经有withDefault解决了其中的一部分,但是地图的get函数仍然返回一个可以为空的值,所以我继续使用withDefault <启发我自己的实现/ p>

interface NonNullableMutableMap<K,V> : MutableMap<K,V> {
  override fun put(key: K, value: V): V
  override fun get(key: K): V
}

fun <K,V> MutableMap<K,V>.withoutNullValues(default: () -> V): NonNullableMutableMap<K, V> {
  return NonNullableMapWrapper(this, default)
}

class NonNullableMapWrapper<K,V>(val map: MutableMap<K,V>, val default: () -> V) : NonNullableMutableMap<K,V> {
  override fun put(key: K, value: V): V = map.put(key, value) ?: default()

  override fun get(key: K): V {
      val value = map.getOrPut(key, default)
      return value
  }
  override val size: Int get() = map.size

  override fun containsKey(key: K): Boolean = map.containsKey(key)

  override fun containsValue(value: V): Boolean = map.containsValue(value)

  override fun isEmpty(): Boolean = map.isEmpty()

  override val entries: MutableSet<MutableMap.MutableEntry<K, V>> get() = map.entries
  override val keys: MutableSet<K> get() = map.keys
  override val values: MutableCollection<V> get() = map.values

  override fun clear() {
      map.clear()
  }

  override fun putAll(from: Map<out K, V>) {
    map.putAll(from)
  }

  override fun remove(key: K): V {
    return map.remove(key) ?: default()
  }
}

我创建了以下单元测试来测试它

class NonNullableMapTest {

  @Test
  fun notNullableTest() {
      val map = HashMap<String, Long>().withoutNullValues { 0 }
      map["first"] += 10L
      map["second"] -= 10L
      assertThat(map["first"]).isEqualTo(10L)
      assertThat(map["second"]).isEqualTo(-10L)
      assertThat(map["third"]).isEqualTo(0L)
  }
}

但是我在运行测试时遇到以下错误:

tried to access method kotlin.collections.MapsKt__MapsJVMKt.set(Ljava/util/Map;Ljava/lang/Object;Ljava/lang/Object;)V from class foo.bar.NonNullableMapTest
java.lang.IllegalAccessError: tried to access method kotlin.collections.MapsKt__MapsJVMKt.set(Ljava/util/Map;Ljava/lang/Object;Ljava/lang/Object;)V from class foo.bar.NonNullableMapTest

知道如何解决此问题吗?

2 个答案:

答案 0 :(得分:2)

对我来说这看起来像个错误。我建议在Kotlin (KT) | YouTrack报告。

解决此问题的一种方法是在set界面上明确定义NonNullableMutableMap。 e.g:

interface NonNullableMutableMap<K, V> : MutableMap<K, V> {
    override fun put(key: K, value: V): V
    override fun get(key: K): V
    operator fun set(key: K, value: V) {
        put(key, value)
    }
}

答案 1 :(得分:2)

关于您获得的运行时错误,目前存在+=运算符如何针对内联MutableMap.set扩展函数进行编译的错误:https://youtrack.jetbrains.com/issue/KT-14227

解决方法是不使用+=

map["first"] = map["first"] + 10L