Kotlin 密封类 - 如何按密封类排序类似于按枚举排序

Question

在 Java 中，我们可以通过枚举轻松地对集合进行排序，如下所示：

Collections.sort(toSortEnumList, new Comparator<theEnum>() {
    @Override
    public int compare(theEnum o1, theEnum o2) {
        return o1.ordinal().compareTo(o2.ordinal());
    }
});

并且toSortEnumList 将按升序排列。如何使用 Kotlin 的密封类来做到这一点？我尝试按类名排序，但这不是按枚举位置。必须有某种方法可以按枚举位置排序：

sealed class GoodCountries {
    
    class Brazil : GoodCountries() {}

    class USA : GoodCountries() {}
    
    class Germany : GoodCountries() {}
    
    class China : GoodCountries() {}
}

 // later on
    
 var toSortList = listOf<GoodCountries>(China(), Brazil(), USA(), Germany())

 Collections.sort(
     toSortList,
     { x: GoodCountries, y: GoodCountries -> y::class.java.name.compareTo(x::class.java.name) }
 )

 Log.v("myTag", toSortList.toString())

打印出来：

美国、德国、中国、巴西

降序。不是我想要的。我想按密封的类顺序（如 Java 枚举中的序号）排序，如下所示：

巴西、美国、德国、中国

我认为密封类应该比枚举更好，但如果我不能做到这一点，枚举可能会有优势。

更新：多亏了 Roland 的帮助，我才得以找到密封类的列表。但现在我想对其进行排序：这是我目前所拥有的：

Collections.sort(toSortList, object : Comparator<GoodCountries> {
    override fun compare(left: GoodCountries, right: GoodCountries): Int {
        return Integer.compare(
            GoodCountries::class.sealedSubclasses.indexOf(left), 
            GoodCountries::class.sealedSubclasses.indexOf(right)
        )
    }
})

但我在indexOf 收到以下错误：

类型推断失败。类型参数 T 的值应在输入类型（参数类型、接收者类型或预期类型）中提及。尝试明确指定它。

Answer 1

您可以像这样为您的GoodCountries 提供order 属性：

sealed class GoodCountries(val order: Int) {
    class Brazil : GoodCountries(0)
    class USA : GoodCountries(1)
    class Germany : GoodCountries(2)
    class China : GoodCountries(3)
}

这不是一个完美的解决方案，因为您必须手动枚举，但这样可以保证所需的顺序。

这样做会大大简化您的比较代码：

val sorted = toSortList.sortedBy(GoodCountries::order)
println(sorted.map { it::class.simpleName })

输出：

[巴西、美国、德国、中国]

编辑： 更新了 Chrispher 的好主意，这使代码更加简洁（在 GoodCountries 的构造函数中具有 order 属性，而不是使其成为被子类）。

Answer 2

也许不完全是您正在寻找的东西，但也许是......

虽然密封类似乎没有您已经注意到的类似序数的东西，但 class 本身有 sealedSubclasses（即 GoodCountries::class.sealedSubclasses）。此外，似乎sealedSubclasses 的顺序是定义的类之一，即此列表中的Brazil 总是排在第一位，USA 排在第二位，等等。如果它们不是全部嵌套，则顺序不同（即，如果有一些在外面，则首先列出）。

但是：文档并未说明此顺序是故意选择的。在'Sealed classes' reference documentation 和sealedSubclasses (k)documentation 中都没有。

关于您关于按密封类顺序对实体进行排序的问题，您可能需要使用以下内容：

val entityList = listOf(Germany(), China(), USA(), Brazil(), Germany())
entityList.sortedBy { // or use sortedWith(compareBy {
  GoodCountries::class.sealedSubclasses.indexOf(it::class)
}.forEach(::println) // or toList...

或类似：

GoodCountries::class.sealedSubclasses
    .asSequence()
    .flatMap { klazzInOrder ->
      entityList.asSequence().filter { it::class == klazzInOrder }
    }
    .forEach(::println)

就性能而言，两者可能都不是最佳选择，但我想你明白了。

我之前添加的排序示例（当我没有意识到您实际上想要对实体而不是类型进行排序时）：

println("Listing the sealed classes in the order of their declaration*")
GoodCountries::class.sealedSubclasses.forEach(::println)

println("Listing the sealed classes ordered by their simple name")
GoodCountries::class.sealedSubclasses.sortedWith(compareBy(String.CASE_INSENSITIVE_ORDER) { it.simpleName!! })
  .forEach(::println)
// same result, but written differently
GoodCountries::class.sealedSubclasses.sortedBy { it.simpleName?.toLowerCase() }
  .forEach(::println)

您甚至可能想要组合 nullsLast 和 CASE_INSENSITIVE_ORDER（很可能是如果您不处理密封类），在这种情况下您可以编写如下内容：

GoodCountries::class.sealedSubclasses.sortedWith(compareBy(nullsLast(String.CASE_INSENSITIVE_ORDER)) { it.simpleName })
  .forEach(::println)

Answer 3

快速回答您的问题是在比较器中切换x 和y，即x::class.java.name.compareTo(y::class.java.name)

更长的答案是，对于您的用例，枚举可能会更好。当某些子类看起来与其他子类不同时，密封类会发光，并且拥有它们的多个实例是有意义的。例如，sealed class Result 具有子类 Success 和 Error，其中 Success 保存数据，Error 保存异常。假设您想对所有国家/地区一视同仁，您的用例似乎更适合传统枚举。