Kotlin：如何使用扩展函数扩展枚举类

Question

我正在尝试使用以下函数扩展 String 类型的枚举类，但无法像这样在调用站点使用它：

fun <T: Enum<String>> Class<T>.join(skipFirst: Int = 0, skipLast: Int = 0): String {
    return this.enumConstants
        .drop(skipFirst)
        .dropLast(skipLast)
        .map { e -> e.name }
        .joinToString()
}

MyStringEnum.join(1, 1);

我在这里做错了什么？

Answer 1

我建议以下解决方案：

fun <T : Enum<*>> KClass<T>.join(skipFirst: Int = 0, skipLast: Int = 0): String {
    return this.java
            .enumConstants
            .drop(skipFirst)
            .dropLast(skipLast)
            .map { e -> e.name }
            .joinToString()
}

我没有将扩展函数附加到 Class，而是将它附加到 KotlinClass。

现在，您可以简单地使用它了：

enum class Test {ONE, TWO, THREE }

fun main(args: Array<String>) {
    println(Test::class.join())
}
// ONE, TWO, THREE

Answer 2

@IRus 的回答是正确的，但您不必使用反射。对于每个枚举类，编译器会自动生成一个values() 方法。此方法返回一个包含所有条目的数组。我们可以像这样让扩展函数直接对这个数组进行操作：

fun <T : Enum<*>> Array<T>.join(skipFirst: Int = 0, skipLast: Int = 0)
        = drop(skipFirst)
        .dropLast(skipLast)
        .map { e -> e.name }
        .joinToString()

然后这样称呼它：

fun main(args: Array<String>) {
    Test.values().join()
}

Answer 3

使用::class 是一种令人讨厌的解决方法。 我建议您查看 stdlib 中的 enumValues<E> 和 enumValueOf<E> 并执行相同的操作：

inline fun <reified E : Enum<E>> joinValuesOf(skipFirst: Int = 0, skipLast: Int = 0): String =
        enumValues<E>().join(skipFirst, skipLast)

@PublishedApi
internal fun Array<out Enum<*>>.join(skipFirst: Int, skipLast: Int): String =
        asList()
                .subList(skipFirst, size - skipLast)
                .joinToString(transform = Enum<*>::name)

用法：joinValuesOf<Thread.State>()

Answer 4

我会用通配符稍微改写你的加入：

fun <T: Enum<*>> Class<T>.join(skipFirst: Int = 0, skipLast: Int = 0): String {
    return this.enumConstants
            .drop(skipFirst)
            .dropLast(skipLast)
            .map { e -> e.name }
            .joinToString()
}

然后，假设您的 MyStringEnum 是这样定义的：

enum class MyStringEnum { FOO, BAR, BAZ }

你可以这样称呼它：

println(MyStringEnum.values()[0].javaClass.join())

获取输出“FOO, BAR, BAZ”

由于您在 Class 上定义连接，因此您需要一个实际的 Class 对象来调用它。枚举类显然不是那样工作的，但是它定义的枚举可以产生一个带有javaClass 的类。所以这是我能想到的最好的，我认为符合您要求的一般精神。我不知道是否有更优雅的方式来实现你想要为所有这样的枚举类做的事情。

编辑：你可以用这个把它收紧一点：

fun Enum<*>.join(skipFirst: Int = 0, skipLast: Int = 0): String {
    return this.javaClass.join(skipFirst, skipLast)
}

这让你可以这样调用：

println(MyStringEnum.values()[0].join())