Kotlin smart使用过滤器投射了一对的第二个值

时间:2016-04-12 08:37:44

标签: kotlin

我试图编写一个映射String和Int的函数?成对,然后在继续映射之前过滤对中非空的第二个值。

我的代码如下所示:

val ids: List<String> = listOf("a", "b", "c")
val ints: Map<String, Int?> = mapOf("a" to 1, "b" to 2, "c" to null)

ids.map { id: String ->
    Pair(id, ints[id])
}.filter { pair -> pair.second != null}.map { pair: Pair<String, Int> ->
    func(id, pair.second)  
}

问题是第二张地图有错误:

Type inference failed: Cannot infer type parameter T in 
                       inline fun <T, R> kotlin.collections.Iterable<T>.map ( transform (T) -> R ): kotlin.collections.List<R>

这看起来是因为编译器不知道在我的Iterable<Pair<String, Int?>>之后聪明地将Iterable<Pair<String, Int>>投射到filter。我该怎么做才能解决这个问题?

1 个答案:

答案 0 :(得分:7)

Kotlin的智能演员通常不适用于方法边界之外。但是,无论如何,有几种方法可以实现您的目标。

首先,您可以通过使用!!运算符告诉编译器该对的第二个值永远不会为空:

ids.map { id: String -> Pair(id, ints[id]) }
        .filter { pair -> pair.second != null }
        .map { pair: Pair<String, Int?> -> func(pair.second!!) }

其次,您可以撤消filtermap的顺序并提前应用!!运算符:

ids.filter { id: String -> ints[id] != null }
        .map { id: String -> id to ints[id]!! } //equivalent to Pair(id, ints[id]!!)
        .map { pair: Pair<String, Int> -> func(pair.second) }

最后,通过使用!!扩展方法在一个步骤中组合过滤和映射,您可以在没有mapNotNull运算符的情况下使其工作:

ids.mapNotNull { id: String -> ints[id]?.let { id to it } }
        .map { pair: Pair<String, Int> -> func(pair.second) }