解析字节流时的Kotlin字节与枚举

@s1m0nw1的答案的另一个选择是只搜索一次:

interface EnumCodesMap<E : Enum> {
    fun values(): Array<E>
    val codesMap = values().associate { it.code to it }
}

enum class OpCode(val code: Byte) {
    Foo(1),
    Bar(2),
    Yak(3);

    companion object : EnumCodesMap<OpCode>
}

...
val opCode = OpCode.codesMap[stream.read().toByte()]

注意,这样的东西不能内置,因为您可以有多个相同的值 code 以下内容:

enum class OpCode(val code: Byte) {
    Foo(1),
    Bar(1)
}

编辑:可以提取它以支持多个枚举,但需要一些技巧:

inline fun <reified E : Enum<E>, K> EnumCodesMap(crossinline getKey: (E) -> K) = object : EnumCodesMap<E, K> {
    override val codesMap = enumValues<E>().associate { getKey(it) to it }
}

interface EnumCodesMap<E : Enum<E>, K> {
    val codesMap: Map<K, E>
}

enum class OpCode(val code: Byte) {
    Foo(1),
    Bar(2),
    Yak(3);

    companion object : EnumCodesMap<OpCode, Byte> by EnumCodesMap({ it.code })
}

您可以向枚举添加一个同伴,以添加查找相关枚举值的功能:

enum class OpCode(val code: Byte) {
    Foo(1),
    Bar(2),
    Yak(3);

    companion object {
        fun of(code: Byte) = values().find { it.code == code }
                ?: throw IllegalAccessException("")
    }
}

像这样使用:

while (stream.available() > 0) {
     val opCode = OpCode.of(stream.read().toByte())
     when (opCode) {
         OpCode.Foo -> {}
         OpCode.Bar -> {}
         OpCode.Yak -> {}
     }
 }