谈谈Kotlin:Kotlin每一行代码都有返回值

2,138 阅读2分钟

?:+let 实现的 if-else

这周在网上冲浪的时候,看到了这么一T个讨论:“Elvis运算符与return组合的语句,在return前增加逻辑,如何写得优雅?”,里面提到一个「使用let语法糖结合?:运算符实现if-else」的示例:

account?.let {
     it.hello()
     it.name = "Hello"
} ?: run {
     logger.error("account is null")
}

这里藏着一个坑

乍一看,这种写法很新颖很有创意,但实际上let语法糖后接?:这种做法是有问题的。

看下let语法糖的函数声明:public inline fun <T, R> T.let(block: (T) -> R): R

let 语法糖

结合实现,可以看到,let会在block执行完后,返回block的返回值。

而Kotlin和Java不同,在Kotlin里每一行代码都是表达式,也就是说每一行代码执行完毕后都有一个返回值。

接下来考虑如下例子:

// 例1:可空变量为空
val nullVal: Any? = null
nullVal?.let {
    println("[nullVal] not null code block")
    null
} ?: run {
    println("[nullVal] null code block")
}

// 例2:可空变量为非空
val notnull: Any? = Any()
notnull?.let {
    println("[notnull] not null code block")
    null
} ?: run {
    println("[notnull] null code block")
}

会得到如下输出:

[nullVal] null code block

[notnull] not null code block
[notnull] null code block

例2的输出显然是不符合预期的。

在文章一开始的那个例子里,由于it.name = "Hello"的返回值是Unit,是一个非空的值,因此能够如预期,呈现出和if-else等价的效果,但这里实际上会留下一个隐藏的坑。

写代码的时候,肯定不会写出我上面举的例子那么傻的代码,考虑如下变种:

fun test_let() {
    val nullable: Any? = null
    nullable?.let {
        println("[nullable] not null code block")
        maybeReturnNull(0)
    } ?: run {
        println("[nullable] null code block")
    }

    val notnull: Any? = Any()
    notnull?.let {
        println("[notnull] not null code block")
        maybeReturnNull(0)
    } ?: run {
        println("[notnull] null code block")
    }
}

private fun maybeReturnNull(count: Int): Any? = if (count % 2 == 0) null else Any()

一旦命中这样的坑,查起来挺费劲的 QAQ

掉过这样的坑后,就会发现朴素的if (xxx != null)的写法其实是最可爱的。

Tips

介绍一个小技巧:

IDEA编辑器提供了快速判空的模板,在变量后输入.nn回车。

就能收获如下代码。

PS:nnnotnull的缩写,输入.notnull也有同样的功效。