首页 > 解决方案 > Kotlin:多种类型的智能转换

问题描述

我正在为 antlr 构建一个从 CST 到 AST 的映射器,所以我有大量的*Context类需要映射到它们对应的 AST 节点。我有 ANTLR 生成的类和我的映射器方法:

// Demo data:
open class Super
class Sub0: Super
class Sub1: Super
// Mappers:
fun map(a: Super) = println("Super")
fun map(a: Sub0) = println("Sub0")
fun map(a: Sub1) = println("Sub1")

然后,我想按如下方式使用它:

listOf(Super(), Sub0(), Sub1()).forEach {
    when (it) {
        is B, is C -> { print('*'); map(it) }
        else -> map(it)
    }
}

我希望it被智能转换为 Sub0 或 Sub1 并调用正确map的 ,但这会给出:

Super
*Super
*Super

这表明它选择了正确的路径但没有进行自动投射。这种方法有效,但随着你拥有越来越多SubX的 's 会变得很长:

when (it) {
    is Sub0 -> {
        print("*");
        map(it)
    }
    is Sub1 -> {
         print("*");
         map(it)
    }
    else -> map(it)
}

我知道我可以以反射的形式使用一些黑魔法并遍历所有map(X)s 然后使用一些“聪明的技巧”来选择正确的,但我宁愿不这样做。;)

标签: kotlin

解决方案


智能转换不起作用,因为 when 分支内的代码is B, is C -> map(it)只进行了一次类型检查。它不会被编译成两组不同的指令,一种类型和另一种类型。编译器需要推断出一个it在这两种情况下都可以工作的类型。

Sub0和的类型都Sub1不能选择 和it,因为选择其中一个不会覆盖另一个。所以编译器选择 and 的最不常见的超类型Sub0Sub1Super. map然后使用静态已知类型解析Super对的调用it

因此,确实,拆分分支以便在分支条件中仅提及一种类型是您可以解决此问题的方法。


推荐阅读