本文正在参与「金石计划」

本文首发于稀土，转载请联系授权

作者：陈明勇

公众号：Go技能干货

前语

枚举类型是一种常用的数据类型，用于表示一组有限的、预界说的、签字的常量值。在枚举类型中，每个常量都是一个枚举值，它们之间的值相等且唯一。

枚举类型通常用于表示一组相关的常量，比如星期、月份、性别等等。在其他语言里（比如 Java 和 C），都内置了枚举类型，而在 Go 语言里是没有内置枚举类型的，因而咱们需求选用其他办法完成相似的枚举类型功用，本文将介绍如何完成 “枚举类型”。

Go 语言中的 “枚举类型”

枚举类型的值本质上是常量，因而咱们能够运用 Go 语言中的常量来完成相似枚举类型的功用，例如：

const (
   Sunday    = 1
   Tuesday   = 2
   Wednesday = 3
   Thursday  = 4
   Friday    = 5
   Saturday  = 6
   Monday    = 7
)

在这个比如中，咱们运用了 const 关键字界说了一组常量，其间每个常量的称号代表着一个枚举，其值为对应的整数。

虽然这个比如能完成相似的枚举类型，但它不具有枚举类型的所有特征，例如缺少安全性和束缚性，为了处理这两个问题，咱们能够运用自界说类型进行改善：

type WeekDay int
const (
   Sunday    WeekDay = 1
   Tuesday   WeekDay = 2
   Wednesday WeekDay = 3
   Thursday  WeekDay = 4
   Friday    WeekDay = 5
   Saturday  WeekDay = 6
   Monday    WeekDay = 7
)

在改善后的比如中，咱们界说了一个自界说类型 Weekday，用于表示星期几。运用 const 关键字界说一个常量组，其间每个常量都被赋予了一个具体的值，一起运用 Weekday 类型进行类型束缚和类型检查。这样，咱们就能够经过枚举值的称号来表示某个特定的星期几，并且因为运用了自界说类型，编译器能够进行类型检查，然后提高了类型安全性。

运用 iota 优雅完成枚举

经过前面的比如不难发现，当咱们需求界说多个枚举值时，手动指定每个枚举常量的值会变得十分麻烦。为了处理这个问题，咱们能够运用 iota 常量生成器，它能够帮助咱们生成接连的整数值。

例如，运用 iota 界说一个星期几的枚举类型：

type WeekDay int
const (
   Sunday WeekDay = iota
   Tuesday
   Wednesday
   Thursday
   Friday
   Saturday
   Monday
)

在这个比如中，咱们运用 iota 自增常量生成器来界说了一个星期几的枚举类型，每个枚举值都是一个 Weekday 类型的常量。因为 iota 的自增规矩，每个枚举值的值将主动递增，然后生成一系列接连的整数值。

为自界说的枚举增加办法

type WeekDay int
const (
   Sunday WeekDay = iota
   Tuesday
   Wednesday
   Thursday
   Friday
   Saturday
   Monday
)

为了能让咱们完成的 “枚举类型” 更加具有枚举类型的特征，咱们能够为其增加相似 Java 等其他语言中的枚举办法。

Name()

回来枚举值的称号。

// Name 回来枚举值的称号
func (w WeekDay) Name() string {
   if w < Sunday || w > Monday {
      return "Unknown"
   }
   return [...]string{"Monday", "Tuesday", "Wednesday", "Thursday", "Friday", "Saturday", "Sunday"}[w]
}

Original

回来枚举值在枚举类型中的方位。

// Original 因为这个枚举类型的枚举值本质上是整数常量，因而咱们能够直接运用枚举值作为它的序号
func (w WeekDay) Original() int {
   return int(w)
}

String()

完成 String 办法，用于打印输出。

// 将枚举值转成字符串，便于输出
func (w WeekDay) String() string {
   return w.Name()
}

Values()

回来一个包含所有枚举值的切片。

// Values 回来一个包含所有枚举值的切片
func Values() []WeekDay {
   return []WeekDay{Monday, Tuesday, Wednesday, Thursday, Friday, Saturday, Sunday}
}

ValueOf()

根据称号回来对应的枚举值。

// ValueOf 运用 switch 语句来根据称号回来对应的枚举值
func ValueOf(name string) (WeekDay, error) {
   switch name {
   case "Sunday":
      return Sunday, nil
   case "Monday":
      return Monday, nil
   case "Tuesday":
      return Tuesday, nil
   case "Wednesday":
      return Wednesday, nil
   case "Thursday":
      return Thursday, nil
   case "Friday":
      return Friday, nil
   case "Saturday":
      return Saturday, nil
   default:
      return 0, fmt.Errorf("invalid WeekDay name: %s", name)
   }
}

小结

在日常开发中，枚举类型是很常用的，虽然 Go 语言中没有内置枚举类型，但也不阻碍咱们自己完成一个相似的 “枚举类型”。在完成的时分，需求考虑类型束缚和安全性的问题。

完成相似枚举类型功用的办法有很多种，本文仅仅介绍了运用自界说类型的办法，如果理解了核心思想，咱们还能够运用结构体等办法来完成相似枚举类型的功用。