Kotlin枚举类和扩展

2024-01-14 17:03:39

枚举类

枚举类是一种特殊的类，它包含一组常量。这些常量代表一组相关的、相互排斥的值。枚举类的基本用法如下：

enum class Color {
    RED,
    GREEN,
    BLUE
}

这将在Color类中创建三个常量：RED、GREEN和BLUE。这些常量可以像普通变量一样使用：

val color = Color.RED

枚举类还可以具有方法和属性。例如，我们可以为Color类添加一个方法来返回颜色的名称：

enum class Color {
    RED,
    GREEN,
    BLUE;

    fun getName(): String {
        return when (this) {
            RED -> "Red"
            GREEN -> "Green"
            BLUE -> "Blue"
        }
    }
}

val color = Color.RED
val colorName = color.getName()

扩展

扩展允许我们为现有类添加新的功能，而无需修改其源代码。这可以通过扩展函数和扩展属性来实现。

扩展函数就像普通函数一样，但它们可以作用于任何类型，即使该类型不是函数定义的一部分。例如，我们可以为Color类添加一个扩展函数来将颜色转换为十六进制代码：

fun Color.toHex(): String {
    return when (this) {
        RED -> "#FF0000"
        GREEN -> "#00FF00"
        BLUE -> "#0000FF"
    }
}

val color = Color.RED
val hexCode = color.toHex()

扩展属性就像普通属性一样，但它们可以作用于任何类型，即使该类型不是属性定义的一部分。例如，我们可以为Color类添加一个扩展属性来返回颜色的互补色：

val Color.complementary: Color
    get() {
        return when (this) {
            RED -> GREEN
            GREEN -> RED
            BLUE -> ORANGE
        }
    }

val color = Color.RED
val complementaryColor = color.complementary