掌握 Go 中的 ENUM

通常，在我们开发的系统中，我们会遇到恒定的值。这些值的一个示例可以是注册状态的表示。在这种情况下，请考虑包含除活动和非活动之外的更多变化的状态。

如果这些状态被定义为字符串，它们在系统内的验证可能会成为一个令人头疼的问题。此外，这种方法可能会“膨胀”二进制文件，因为每次验证都会涉及两个字符串（预期值和正在验证的值）。

为了避免这些问题，我们可以使用众所周知的枚举类型。如果您不熟悉这种类型，它本质上是一种固定或有限大小的类型。

为了让它更清楚，让我们深入研究一些代码。按照前面提出的想法，我们将创建一个枚举类型来验证注册状态。

定义新类型

基于 go 的标准类型创建枚举可能会出现问题。让我解释一下。假设我们将状态定义为 uint8 类型。现在，假设我们的系统还有另一个用于流派的 uint8 类型的枚举。

现在想象一下值 1 代表待处理状态和乡村音乐流派。如果执行 if pending == country 验证会发生什么？没错，它会返回 true。

为了防止这种情况，我们将创建一个专门用于处理状态的新类型。该类型将基于 uint8，但由于它是一个不同的类型，前面提到的验证不会返回 true。

type status uint8

创建枚举

定义了新类型后，让我们为注册状态创建常量及其相应的值。

const (
  created status = 0
  pending = 1
  approved = 2
  rejected = 3
)

虽然像我们上面那样分配值本身并没有什么错误，但还有一种更简单的方法。我们可以使用 iota 关键字，而不是为每个常量赋值。该关键字使 go 将 0 分配给第一个常量，然后为每个后续常量按顺序将值加 1。

const (
  created status = iota
  pending
  approved
  rejected
)

打印枚举

由于当前实现了枚举，打印常量 created 将显示值 0。但是，为了更好的可读性，显示单词 created 而不是值 0 更有帮助。

解决方案非常简单。只需实现神奇的 string() 方法即可。

func (s status) string() string {
  switch s {
    case created:
      return "created"
    case pending:
      return "pending"
    case approved:
      return "approved"
    case rejected:
      return "rejected"
   }

   return "unknown"
}

结论

为了测试这一点，让我们简单打印一下 pending 状态。

package main

import "fmt"

type Status uint8

const (
  Created Status = iota
  Pending
  Approved
  Rejected
)

func (s Status) String() string {
  switch s {
    case Created:
      return "created"
    case Pending:
      return "pending"
    case Approved:
      return "approved"
    case Rejected:
      return "rejected"
  }

  return "unknown"
}

func main() {
  fmt.Println(Pending)
}

执行命令 go run 应该在终端上输出挂起。

就是这样！我希望这对您有帮助。

下次再见！

以上就是掌握 Go 中的 ENUM的详细内容，更多请关注硕下网其它相关文章！