Hello! 欢迎来到小浪资源网!

掌握 Go 中的 ENUM


掌握 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 应该在终端上输出挂起。

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

下次再见!

相关阅读