伙计,Go项目怎么使用枚举?

网站建设2年前发布
7 00

哈喽,大家好,我是asong​。枚举是一种很重要的数据类型,在java、C语言等主流编程语言中都支持了枚举类型,但是在Go语言中却没有枚举类型,那有什么替代方案吗?本文我们来聊一聊这个事情;,我们以java​语言为例子,在JDK1.5​之前没有枚举类型,我们通常会使用int常量来表示枚举,一般使用如下:,使用int类型会存在以下隐患:,于是乎我们就想到用常量字符来表示,代码就变成了这样:,这样也同样存在问题,因为我们使用的常量字符,那么有些程序猿不按套路出牌就可以使用字符串的值进行比较,这样的代码会被不断模仿变得越来越多的,然后屎山就出现了;,所以我们迫切需要枚举类型的出现来起到约束的作用,假设使用一个枚举类型做入参,枚举类型就可以限定沙雕用户不按套路传参,这样就可以怼他了,哈哈~;,使用枚举的代码就可以变成这样,传了枚举之外的类型都不可以了;,Go语言就没有枚举类型,我们该使用什么方法来替代呢?,枚举通常是一组相关的常量集合,Go​语言中有提供常量类型,所以我们可以使用常量来声明枚举,但也同样会遇到上述的问题,起不到约束的作用,所以为了起到约束我们可以使用Go​语言另外一个知识点 — 类型定义,Go​语言中可以使用type​关键字定义不同的类型,我们可以为整型、浮点型、字符型等定义新的类型,新的类型与原类型转换需要显式转换,这样在一定程度上也起到了约束的作用,我们就可以用Go语言实现如下枚举:,上面的代码就会报错:,定义新的类型可以起到约束作用,比如我们要检查状态机,入参限定了必须是OrderStatus类型,如果是int类型就会报错。,上面我们的枚举实现方式只能获取枚举值,获取不到其映射的字面意思,所以我们可以优化一下,实现String方法,使用官方提供的cmd/string来快速实现,代码如下:,执行命令go generate ./…生成orderstatus_string.go文件:,Go语言使用protobuf会生成对应的枚举代码,我们发现其中也是使用定义新的类型的方式来实现的,然后在封装一些方法,我们来赏析一下protobuf生成的枚举代码:,总结,虽然Go语言没有提供枚举类型,但是我们也可以根据Go语言的两个特性:常量和定义新类型来实现枚举,方法总比困难多吗,开源库是优秀的,我们往往可以从高手那里里学习很多,记住,请永远保持一个学徒之心。

© 版权声明

相关文章