文章详情

短信预约-IT技能 免费直播动态提醒

请输入下面的图形验证码

提交验证

短信预约提醒成功

三分钟带你用 Go 语言实现枚举

2024-12-02 06:54

关注

Go语言是没有 enum 这个枚举关键字的。

那 Go 语言里面又是怎么处理枚举这个问题的呢?

我们这篇文章讲解下!

需要枚举的场景

我们在处理人的姓名的时候,一般都是男或者女,或者最大再加一个未知,不会有其他类型吧。

像上面这种常见,我们一般就会使用枚举处理。

我们会定义一个变量,希望他能只能赋 男、女、未知 这三种值。

如果是在其他语言:

from enum import Enum

class Sex(Enum):
man = 1
woman = 2
unknown = 3


然后在使用的时候就可以直接 Sex.main 直接调用。

Go 的实现

在 Go 语言中,我们如果要实现枚举,则需要使用常量计数器 iota 结合 const 来实现。

iota 是 go 语言的常量计数器,只能在 const 里面使用。

比如:

const(
SexMan = iota
SexWoman
SexUnknown
)

func main() {

fmt.Println(SexMan,SexWoman,SexUnknown)
}

// 执行结果
$ go run main.go
0 1 2


首先我们知道 const 是用来定义常量的,后面接括号就是定义一组常量。

我们给第一个常量赋值为 iota,在这组常量中,之后的常量就会为我们自动赋值递增。

iota有啥特点

?iota 在 const 关键字出现时,在 const 内部的第一行之前就被初始化为 0 了,const 中每增加一行常量声明就将自动加 1 ,哪怕是 _ 这种声明。

const(
SexMan = iota
SexWoman
_
SexUnknown
)

func main() {

fmt.Println(SexMan,SexWoman,SexUnknown)
}

// 执行结果
$ go run main.go
0 1 3


你会发现,最后一个变量的值变了。

其他情况

还会出现这种情况

const(
SexMan = iota
SexWoman = "woman"
SexUnknown
)

func main() {

fmt.Println(SexMan,SexWoman,SexUnknown)
}

// 执行结果
$ go run main.go
0 woman woman


在出现了 iota 的常量定义中,如果有一个值被特殊赋值后,后面的值都会一样。

解决办法就是在需要恢复计数的地方再次赋值 iota 即可:

const(
SexMan = iota
SexWoman = "woman"
SexUnknown = iota
SexUnknown1
)

func main() {

fmt.Println(SexMan,SexWoman,SexUnknown,SexUnknown1)
}

// 执行结果
$ go run main.go
0 woman 2 3


技巧升级

我们在定义枚举时,常常会显示的声明其类型,比如这样:

type Sex int

const(
SexMan Sex = iota
SexWoman
SexUnknown
)


为什么要这样做呢?

这样做有一个好处,我们可以对我们的 Sex 类型进行扩展。

比如这样:

type Sex int

func (this Sex) String() string {
switch this {
case 0:
return "男"
case 1:
return "女"
default:
return "未知"
}
}

const(
SexMan Sex = iota
SexWoman
SexUnknown
)

func main() {

fmt.Println(SexMan,SexWoman,SexUnknown)
}

// 执行结果
$ go run main.go
男 女 未知


fmt 在 println 时会去调用这个变量的 String 方法,于是打印就会出现男、女、未知了。

当然你还可以根据你的需要,进行扩展哇。

你学废了么?

来源:GoLang全栈内容投诉

免责声明:

① 本站未注明“稿件来源”的信息均来自网络整理。其文字、图片和音视频稿件的所属权归原作者所有。本站收集整理出于非商业性的教育和科研之目的,并不意味着本站赞同其观点或证实其内容的真实性。仅作为临时的测试数据,供内部测试之用。本站并未授权任何人以任何方式主动获取本站任何信息。

② 本站未注明“稿件来源”的临时测试数据将在测试完成后最终做删除处理。有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341

软考中级精品资料免费领

  • 历年真题答案解析
  • 备考技巧名师总结
  • 高频考点精准押题
  • 2024年上半年信息系统项目管理师第二批次真题及答案解析(完整版)

    难度     813人已做
    查看
  • 【考后总结】2024年5月26日信息系统项目管理师第2批次考情分析

    难度     354人已做
    查看
  • 【考后总结】2024年5月25日信息系统项目管理师第1批次考情分析

    难度     318人已做
    查看
  • 2024年上半年软考高项第一、二批次真题考点汇总(完整版)

    难度     435人已做
    查看
  • 2024年上半年系统架构设计师考试综合知识真题

    难度     224人已做
    查看

相关文章

发现更多好内容

猜你喜欢

AI推送时光机
位置:首页-资讯-后端开发
咦!没有更多了?去看看其它编程学习网 内容吧
首页课程
资料下载
问答资讯