使用带有枚举而不是多个if的kotlin子句

时间:2019-01-28 20:29:03

标签: if-statement kotlin enums

我正在使用下一个代码:

  override fun presentNativeItem(dcsItem: DCSItem?): Any {
        if (dcsItem?.type == "NavMenu") {
            return buildNavMenu(dcsItem)
        } else if (dcsItem?.type == "NavLink") {
            return buildNavLink(dcsItem)
        } else if (dcsItem?.type == "Image") {
            return buildImage(dcsItem)
        }
        else throw IllegalStateException("Unknown Type ${dcsItem?.type} of NavItem")
    }

但我不想使用多个if,而是要使用下一个枚举:

enum class DSCType {
    NAVMENU,
    NAVLINK,
    IMAGE;

    override fun toString(): String {
        return this.name.toLowerCase()
    }

    companion object {
        fun fromString(value: String?): DSCType? {
            return when (value?.toLowerCase()) {
                "NavMenu" -> NAVMENU
                "NavLink" -> NAVLINK
                "Image" -> IMAGE
                else -> null
            }
        }
    }
}

关于如何以Kotlin方式实现这一目标的任何想法?

谢谢

2 个答案:

答案 0 :(得分:1)

使输入参数不可为空,并将函数更改为:

override fun presentNativeItem(dcsItem: DCSItem) = when(dcsItem) {
    NAVMENU -> buildNavMenu(dcsItem)
    NAVLINK -> buildNavLink(dcsItem)
    IMAGE -> buildImage(dcsItem)
}

答案 1 :(得分:0)

尝试一下:

fun presentNativeItem(dcsItem: DCSItem?): Any {
    return enumValues<DSCType>().firstOrNull { dcsItem?.type == it.typeName }
            ?.build(dcsItem)
            ?: throw IllegalStateException("Unknown Type ${dcsItem?.type} of NavItem")
}

enum class DSCType(val typeName: String) {

    NAV_MENU("NavMenu") {
        override fun build(dcsItem: DCSItem?): Any {
            TODO("not implemented")
        }
    },

    NAV_LINK("NavLink") {
        override fun build(dcsItem: DCSItem?): Any {
            TODO("not implemented")
        }
    },

    IMAGE("Image") {
        override fun build(dcsItem: DCSItem?): Any {
            TODO("not implemented")
        }
    };

    abstract fun build(dcsItem: DCSItem?): Any
}