Question

假设我有一个

enum class ConceptualPosition{
    A1,A2,A3,A4,A5,A6,A7,A8,
    B1,B2,B3,B4,B5,B6,B7,B8,
    C1,C2,C3,C4,C5,C6,C7,C8
}

现在我想构造一个图，其中每个位置都有一个节点。

import [...].domain.model.base.ConceptualPosition
import [...].domain.model.base.ConceptualPosition.*
import [...].domain.model.base.Node
import [...].domain.model.base.Node.MovableDirection.*

class MyGraph:Graph{
    private val internalPositions = mutableMapOf<ConceptualPosition, Node>()
    override val position by lazy {internalPositions.toMap().withDefault {
        throw IllegalArgumentException("position '$it' does not exist")
    }}

    init{
        //create nodes for all positions
        ConceptualPosition.values().forEach { internalPositions[it] = Node(it) }

        //now build graph
        position[A1]!!.connectBothWaysWith(position[A2]!!,RIGHT)
        position[A2]!!.connectBothWaysWith(position[A3]!!,RIGHT)
        [...]

    }
}

因为我有一个withDefault会立即引发异常，并且由于Kotlin正确地将position的类型推断为Map<ConceptualPosition,Node>，所以我认为我可以确定如果查找确实不会失败，我从中得到一个非空值。

科特林显然无法安全地推断出这一点。

我可以（以合理的方式）重构它来摆脱!!吗？

Answer 1

使用Map.getValue代替get：

position.getValue(A1).connectBothWaysWith(position.getValue(A2),RIGHT)

或者，您可以使用更精确的返回类型（Kotlin标准库makes the equivalent private）定义自己的withDefault版本：

class MapWithDefault<K, V>(val map: Map<K, V>, val default: (K) -> V) : Map<K, V> by map {
    override fun get(key: K): V = map[key] ?: default(key)
}

fun <K, V> Map<K, V>.withDefault(default: (K) -> V) = MapWithDefault(this, default)

Kotlin-我可以重写此地图访问权限来摆脱``!!''吗？

1 个答案: