Enum

Go enum

May 12, 2022
Go
Enum

Go enum #

Go是没有内置枚举类型的，那么，当需要使用枚举时，该怎么办呢？

枚举说白了，就是一连串互斥的值，每个值代表一样事物或一个类型。

比如，现在需要一个颜色枚举，可以这样定义：

const (
    Red = "Red" // 红色
    Blue = "Blue" // 蓝色
    Green = "Green" // 绿色
)

也有这样定义的：

type Color string // 定义一个特定类型

// 枚举常量均声明为该类型
const (
    Red     Color = "Red" // 红色
    Blue    Color = "Blue" // 蓝色
    Green   Color = "Green" // 绿色
)

这样做的好处是可以通过这个类型来更明显的标记出枚举字段来：

type Car struct {
    Name string
    Color Color // 颜色字段声明为Color类型，在阅读代码的时候就能知道这个字段正常的可选值范围
}

但是，上面的做法都需要面临一个问题，就是我需要一个返回全部枚举值的集合时，需要这样做：

func All() []Color {
    return []Color{
        Red,
        Blue,
        Green,
    }
}

func (color Color) Name() string {
    switch color {
    case Red:
        return "红色"
    case Blue:
        return "蓝色"
    case Green:
        return "绿色"
    }
    return ""
}

当在定义处新增值时，All和Name也要同步添加，对于开发人员来说，非常容易遗漏。

...