从 Kotlin 中的值列表创建 EnumMap

Question

我想定义一个enum class的扩展，它接受与枚举常量相同的顺序的值列表并输出一个EnumMap。

相反，我能够做的是创建和扩展List<V>对象，并使用Enum<K>.values()中的键数组作为输入。

似乎困难在于enum class本身不是一个具有.values()方法的对象。所以也许我的要求是不明智的。

下面是使用列表扩展方法的工作示例。

import java.util.*
enum class Shoes(
val title: String,
val weight: Double) {
WORKBOOT("tough", 11.0),
SNEAKER("fast", 6.0),
SLIPPER("soft", 3.0);
}
fun main(args: Array<String>) {
val prices= listOf(11.5,8.2,3.5)
val map = prices.enumMapOf(Shoes.values())
map.print()
}
inline fun <reified K : Enum<K>, V> List<V>.enumMapOf(keys:Array<K>): EnumMap<K, V> {
val map = EnumMap<K, V>(K::class.java)
keys.forEachIndexed{i,k-> map[k]=this[i]}
return map
}
fun <K:Enum<K>,V> EnumMap<K,V>.print() {
this.forEach{k,v-> println("%d: %s --> %.2f".format(k.ordinal,k.name,v)) }
}

Answer 1

我认为您缺少的主要内容是可以为您提供通用枚举值的enumValues方法。

这将允许您执行此操作(我也重新排列了创建EnumMap的代码，当然，如果这看起来太复杂，您可以坚持自己的解决方案(：

inline fun <reified K : Enum<K>, V> Enum<K>.mappedTo(values: List<V>): EnumMap<K, V> {
return enumValues<K>().zip(values).toMap(EnumMap(K::class.java))
}

当然，现在的问题是调用语法 - 在枚举的实例上，必须像这样调用此函数：

Shoes.WORKBOOT.mappedTo(listOf(11.5, 8.2, 3.5))

这没有多大意义。最好打这个电话：

Shoes.mappedTo(listOf(11.5, 8.2, 3.5))

但据我所知，没有办法实现这种语法，因为你不能向通用枚举的伴随对象添加扩展，然后期望在特定枚举的伴随对象上调用它 - 它们不会像那样相互继承。

因此，也许您可以使用常规函数而不是创建扩展，并从泛型参数而不是接收器中获取枚举的类型：

inline fun <reified K : Enum<K>, V> mapEnumTo(values: List<V>): EnumMap<K, V> {
return enumValues<K>().zip(values).toMap(EnumMap(K::class.java))
}
mapEnumTo<Shoes, Double>(listOf(11.5, 8.2, 3.5))

您还必须在此处指定键的类型，因为您不能先有一个显式的泛型类型参数，然后是另一个隐式的泛型类型参数。但我认为这是非常好的语法，因为您会看到正在创建的地图的键和值类型。

Answer 2

作为消除@zsmb13答案中显式Double的解决方法，您可以借用标准的 Scala 技巧，将具有单独类型参数列表拆分为两个方法调用，其中一个是显式的，另一个是推断的：

interface EnumMapper<K : Enum<K>> {
operator fun <V> invoke(values: List<V>): EnumMap<K, V>
}
inline fun <reified K : Enum<K>> mapEnumTo() = object : EnumMapper<K> {
override operator fun <V> invoke(values: List<V>): EnumMap<K, V> {
return enumValues<K>().zip(values).toMap(EnumMap(K::class.java))
}
}
mapEnumTo<Shoes>()(listOf(11.5, 8.2, 3.5))

Answer 3

在枚举本身中： private val shoeMap by lazy { values((.associateBy { it.title } }