java - Kotlin:如果条件为真，则将变量转换为不可空

Question

我的意思是跟随。

考虑这段代码:

// Can be a Long or null
val data1= param1.toLongOrNull()
val data2= param2.toLongOrNull()
val dataN= paramN.toLongOrNull()

// Proceed with the action if ALL of the data are not nulls
if(notNull(data1, data2, dataN)){
    // do something with data1,2,N
}

notNull() 是一个接受可变参数列表的实用函数。 如果所有参数都不为空，则返回 true。

fun <T> notNull(vararg elements: T): Boolean {
    elements.forEach {
        if (it == null) {
            return false
        }
    }
    return true
}

我遇到的问题是 Kotlin 不知道在 if block 内，data1,2,N 不能为 null。结果，编译失败:

if(notNull(data1, data2, dataN)){
     data1 + data2 + dataN
     // Fail: Required Long, Found Long?
     // Operator + not allowed on nullable receiver.
}

另一方面，如果明确检查变量是否为 null，则 Kotlin 知道这很好并且不会提示。

if(data1!=null && data2!=null && dataN!=null){
     data1 + data2 + dataN
}

以这种方式“配置”notNull 方法会很好，这样 Kotlin 就知道一旦它返回 true，任何传递的参数都不能在 block 内为 null。
这可能吗？

Answer 1

显然有多种方法可以给猫剥皮，但您可以这样做:

inline fun <T> doIfNotNull(elements: Array<T?>, block : (Array<T>) -> Unit) {
    val validElements = arrayListOf<T>()
    for (i in elements) {
       if (i == null) return
       validElements.add(i)
    }
    block.invoke(validElements.toArray(elements))
}

与可变参数一起使用(必须是最后一个参数，使接收函数成为第一个参数，感觉不太好)；

inline fun <T> doIfNotNull(block : (Array<T>) -> Unit, vararg elements : T?) {
    val validElements = arrayListOf<T>()
    for (i in elements) {
       if (i == null) return
       validElements.add(i)
    }
    block.invoke(validElements.toArray(elements))
}

示例:

fun test() {
    val elements = arrayOf(1L, 2L, null)
    doIfNotNull(elements, {
        it.sum()
    })
}