通常,在我们开发的系统中,我们会遇到恒定的值。这些值的一个示例可以是注册状态的表示。在这种情况下,请考虑包含除活动和非活动之外的更多变化的状态。
如果这些状态被定义为字符串,它们在系统内的验证可能会成为一个令人头疼的问题。此外,这种方法可能会“膨胀”二进制文件,因为每次验证都会涉及两个字符串(预期值和正在验证的值)。
为了避免这些问题,我们可以使用众所周知的枚举类型。如果您不熟悉这种类型,它本质上是一种固定或有限大小的类型。
为了让它更清楚,让我们深入研究一些代码。按照前面提出的想法,我们将创建一个枚举类型来验证注册状态。
定义新类型
基于 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 应该在终端上输出挂起。
就是这样!我希望这对您有帮助。
下次再见!