小技巧分享：在Go如何实现枚举？

[[404637]]

为新华等地区用户提供了全套网页设计制作服务，及新华网站建设行业解决方案。主营业务为成都做网站、成都网站设计、新华网站设计，以传统方式定制建设网站，并提供域名空间备案等一条龙服务，秉承以专业、用心的态度为用户提供真诚的服务。我们深信只要达到每一位用户的要求，就会得到认可，从而选择与我们长期合作。这样，我们也可以走得更远！

本文转载自微信公众号「脑子进煎鱼了」，作者陈煎鱼。转载本文请联系脑子进煎鱼了公众号。

大家好，我是煎鱼。

在日常的业务工程开发中，我们常常会有使用枚举值的诉求，枚举控的好，测试值边界一遍过...

有的小伙伴会说，在 Go 语言不是有 iota 类型做枚举吗，那煎鱼你这篇文章还讲什么?

讲道理，Go 语言并没有 enum 关键字，有用过 Protobuf 等的小伙伴知道，Go 语言只是 ”有限的枚举“ 支持，我们也会用常量来定义，枚举值也需要有字面意思的映射。

示例

在一些业务场景下，是没法达到我们的诉求的。示例如下：

 
 
 
 

  
  
  
  
type FishType int 
  
  
  
  
 
  
  
  
  
const ( 
  
  
  
  
 A FishType = iota 
  
  
  
  
 B 
  
  
  
  
 C 
  
  
  
  
 D 
  
  
  
  
) 
  
  
  
  
 
  
  
  
  
func main() { 
  
  
  
  
 fmt.Println(A, B, C, D) 
  
  
  
  
} 
 
 
 
 

输出结果为：“0 1 2 3”。这时候就一脸懵逼了...枚举值，应该除了键以外，还得有个对应的值。也就是这个 “0 1 2 3” 分别对应着什么含义，是不是应该输出 ”A B C D“

但 Go 语言这块就没有直接的支撑了，因此这不是一个完整的枚举类型的实现。

同时假设我们传入超过 FishType 类型声明范围的枚举值，在 Go 语言中默认也不会有任何控制，是正常输出的。

上述这种 Go 枚举实现，在某种情况下是不完全的，严格意义上不能成为 enum(枚举)。

使用 String 做枚举

如果要支持枚举值的对应输出的话，我们可以通过如下方式：

 
 
 
 

  
  
  
  
type FishType int 
  
  
  
  
 
  
  
  
  
const ( 
  
  
  
  
 A FishType = iota 
  
  
  
  
 B 
  
  
  
  
 C 
  
  
  
  
 D 
  
  
  
  
) 
  
  
  
  
 
  
  
  
  
func (f FishType) String() string { 
  
  
  
  
 return [...]string{"A", "B", "C", "D"}[f] 
  
  
  
  
} 
 
 
 
 

运行程序：

 
 
 
 

  
  
  
  
func main() { 
  
  
  
  
 var f FishType = A 
  
  
  
  
 fmt.Println(f) 
  
  
  
  
 switch f { 
  
  
  
  
 case A: 
  
  
  
  
  fmt.Println("脑子进煎鱼了") 
  
  
  
  
 case B: 
  
  
  
  
  fmt.Println("记得点赞") 
  
  
  
  
 default: 
  
  
  
  
  fmt.Println("别别别...") 
  
  
  
  
 } 
  
  
  
  
} 
 
 
 
 

输出结果：

 
 
 
 

  
  
  
  
A 
  
  
  
  
脑子进煎鱼了 
 
 
 
 

我们可以借助 Go 中 String 方法的默认约定，针对于定义了 String 方法的类型，默认输出的时候会调用该方法。

这样就可以达到获得枚举值的同时，也能拿到其映射的字面意思。

自动生成 String

但每次手动编写还是比较麻烦的。在这一块，我们可以利用官方提供的 cmd/string 来快速实现。

我们安装如下命令：

 
 
 
 

  
  
  
  
go install golang.org/x/tools/cmd/stringer 
 
 
 
 

在所需枚举值上设置 go:generate 指令：

 
 
 
 

  
  
  
  
//go:generate stringer -type=FishType 
  
  
  
  
type FishType int 
 
 
 
 

在项目根目录执行：

 
 
 
 

  
  
  
  
go generate ./... 
 
 
 
 

会在根目录生成 fishtype_string.go 文件：

 
 
 
 

  
  
  
  
. 
  
  
  
  
├── fishtype_string.go 
  
  
  
  
├── go.mod 
  
  
  
  
├── go.sum 
  
  
  
  
└── main.go 
 
 
 
 

fishtype_string 文件内容：

 
 
 
 

  
  
  
  
package main 
  
  
  
  
 
  
  
  
  
import "strconv" 
  
  
  
  
 
  
  
  
  
const _FishType_name = "ABCD" 
  
  
  
  
 
  
  
  
  
var _FishType_index = [...]uint8{0, 1, 2, 3, 4} 
  
  
  
  
 
  
  
  
  
func (i FishType) String() string { 
  
  
  
  
 if i < 0 || i >= FishType(len(_FishType_index)-1) { 
  
  
  
  
  return "FishType(" + strconv.FormatInt(int64(i), 10) + ")" 
  
  
  
  
 } 
  
  
  
  
 return _FishType_name[_FishType_index[i]:_FishType_index[i+1]] 
  
  
  
  
} 
 
 
 
 

所生成出来的文件，主要是根据枚举值和映射值做了个映射，且针对超出枚举值的场景进行了判断：

 
 
 
 

  
  
  
  
func main() { 
  
  
  
  
 var f1 FishType = A 
  
  
  
  
 fmt.Println(f1) 
  
  
  
  
 var f2 FishType = E 
  
  
  
  
 fmt.Println(f2) 
  
  
  
  
} 
 
 
 
 

执行 go run . 查看程序运行结果：

 
 
 
 

  
  
  
  
$ go run . 
  
  
  
  
A 
  
  
  
  
FishType(4) 
 
 
 
 

总结

在今天这篇文章中，我们介绍了如何在 Go 语言实现标准的枚举值，虽然有些繁琐，但整体不会太难。

也有小伙伴已经在社区中提出了 ”proposal: spec: add typed enum support“ 的提案，相信未来有机会能看到 Go 自身支持 enum(枚举)的那一天。

网页名称：小技巧分享：在Go如何实现枚举？
网站地址：http://www.shufengxianlan.com/qtweb/news43/399943.html

网站建设、网络推广公司-创新互联，是专注品牌与效果的网站制作，网络营销seo公司；服务项目有等

广告

声明：本网站发布的内容（图片、视频和文字）以用户投稿、用户转载内容为主，如果涉及侵权请尽快告知，我们将会在第一时间删除。文章观点不代表本网站立场，如需处理请联系客服。电话：028-86922220；邮箱：631063699@qq.com。内容未经允许不得转载，或转载时需注明来源： 创新互联