kotlin - 表达时自定义Kotlin

Question

我真的是Kotlin的新手，我想要的是:

when_assert_no_else {
    CONDITION0 -> {
        doSomething0()
    }
    CONDITION1 -> {
        doSomething1()
    }
}

就像

when {
    CONDITION0 -> {
        doSomething0()
    }
    CONDITION1 -> {
        doSomething1()
    }
    else -> {
        throw RuntimeException()
    }
}

后面的代码片段在我的项目中占了很多时间，我想通过抛出异常来断言控制流没有到达 else块。

或者，是否可以通过提供 when的实现来自定义 when_assert_no_else关键字？

有什么想法吗？谢谢。

Answer 1

由于kotlin没有->运算符，因此您无法编写以上代码，但是您可以做出一些妥协，例如:改用Pair<()->Boolean,()->T>。

fun test(): Int = when_assert_no_else(
        { true } to { 1 },
        { false } to { 2 }
)



fun <T> when_assert_no_else(vararg cases: Pair<() -> Boolean, () -> T>): T {
    //           v--- short-circuiting terminal operation 
    return cases.find{ it.first() }
                .let { it ?: throw RuntimeException() }
                .second()
}