主页 / coding / 问题 / 79290494
Accepted
William Hu
Asked: 2024-12-18 16:50:50 +0800 CST

使用泛型类型通过原始值获取枚举中的关联类型

  • 772

这是一个枚举类：

enum class Foo(rawValue: Int) {
    FIRST(1),
    SECOND(2)
}

我可以用来companion object通过原始值检索类型：

enum class Foo(rawValue: Int) {
    FIRST(1),
    SECOND(2)


   companion object {
    
      fun getCaseBy(value: Int): Foo? {
        return entries.firstOrNull { it.rawValue == value }
      }
   }
}

然后像这样使用它： Foo.getCaseBy(1)

有没有办法使用泛型将其getCaseBy放入interface？那么就不需要为每个枚举添加这种方法了？

类似于：

interface EnumInterface<T> {
    
    companion object {
        
        fun getCaseBy(value: E): T? {
            return T.entries.firstOrNull { it.rawValue == value }
        }
    }
}

enum class Foo(rawValue: Int): EnumInterface(Foo::class) {
    FIRST(1),
    SECOND(2)
}
  1. T仅限于Enum类，E是检索T的 rawValue 类型。
  2. 只有Enum输入才能确认此接口EnumInterface

这可能吗？

kotlin

1 个回答

  1. Best Answer

    这是一个受Louis Wasserman的评论启发的更清晰的实现。

    import kotlin.reflect.KClass

abstract class EnumGetter<E : Enum<E>, V>(private val enumClass: KClass<E>) {

  abstract fun predicate(e: E, value: V): Boolean;

  fun getEnum(value: V): E? {
    return enumClass.java.enumConstants.firstOrNull { predicate(it, value) }
  }
}

enum class Foo(val rawValue: Int) {
  FIRST(1),
  SECOND(2);

  companion object : EnumGetter<Foo, Int>(Foo::class) {

    override fun predicate(e: Foo, value: Int): Boolean {
      return e.rawValue == value
    }
  }
}

fun main() {
  println(Foo.getEnum(1))
  println(Foo.getEnum(2))
  println(Foo.getEnum(3))
}

    如果你觉得重写predicate函数太麻烦，可以按照下面与旧版本相同的方法，为关键字段创建注解。

    这是旧版本。

    您可以通过反射来实现这一点。

    import kotlin.reflect.full.hasAnnotation
import kotlin.reflect.full.memberProperties

@Target(AnnotationTarget.PROPERTY)
@Retention(AnnotationRetention.RUNTIME)
annotation class Key

inline fun <reified E : Enum<E>> getValue(enumObject: E): Any? {
  val field = E::class.memberProperties.first { it.hasAnnotation<Key>() }
  return field.get(enumObject)
}

inline fun <reified E : Enum<E>> getValueBy(value: Any?): E {
  return enumValues<E>().first { getValue<E>(it) == value }
}

enum class Foo(@Key val rawValue: Int) {
  FIRST(1),
  SECOND(2);
}

fun main() {
  println(getValueBy<Foo>(2));
}

    由于枚举类可以具有各种属性名称，因此使用注解Key来标记用于比较的属性。

    请注意，如果没有注释任何属性，getValue将会引发NoSuchElement错误。

    • 0

相关问题