kotlin - 将枚举成员转换为相应的文本

Question

在Kotlin中，我有一个枚举，如下所示:

enum class MediaType() {
    AUDIO,
    VIDEO,
    ARTICLE;
}

我想添加一个函数或一些属性，以允许将枚举成员转换为相应的文本。例如:

var mediaType = MediaType.AUDIO
var text = mediaType.toText() // returns the string "MP3"

mediaType = MediaType.VIDEO
text = mediaType.toText() // returns the string "mpeg"

虽然可以将toText函数添加到MediaType类中，但是我不确定该函数如何引用它设置的值。

Answer 1

您可以向枚举添加属性...

enum class MediaType(val text: String) {
    AUDIO("mp3"),
    VIDEO("mpeg"),
    ARTICLE("text");
}

然后像这样使用它:

println(MediaType.AUDIO.text)

如果您想要的是 toText()函数而不是属性，则也可以添加该函数，但可能不是惯用的:

enum class MediaType(private val text: String) {
    AUDIO("mp3"),
    VIDEO("mpeg"),
    ARTICLE("text");

    fun toText(): String = text
}

更新:

另一种方法是添加扩展功能，并将此逻辑完全保留在枚举之外:

fun MediaType.toText(): String = 
    when(this) {
       MediaType.AUDIO -> "mp3"
       MediaType.VIDEO -> "mpeg"
       MediaType.ARTICLE -> "text"
    }