Go iota关键字与枚举类型实现原理


作者: 树獭叔叔​​​​​​​ 时间: 1月前 人气: 2 评论: 0

1. iota关键字

iota是go语言的常量计数器,只能在常量的表达式中使用,其值从0开始,在const中每新增一行iota自己增长1,其值一直自增1直到遇到下一个const关键字,其值才被重新置为0。

const (
    a int = iota // 0
    b            // 1
    c            // 2
    d            // 3
)

另外,iota在使用时,可以跳过某几行(需要注意的是,空格并不算行)。

const ( 
    a = iota // 0 
    b        // 1 
    c        // 2 
    _ 
    _ 
    d        // 5 
)

并且,iota可以参与数学运算。

type Allergen int
const (
    IgEggs Allergen = 1 << iota // 1 << 0 which is 00000001 = 1
    IgChocolate                 // 1 << 1 which is 00000010 = 2
    IgNuts                      // 1 << 2 which is 00000100 = 4
    IgStrawberries              // 1 << 3 which is 00001000 = 8
    IgShellfish                 // 1 << 4 which is 00010000 = 16
)

并且,在同一行中,一个iota可以多次使用:

const (
    a, q int = iota, iota + 1 // 0, 1
    b, w                      // 1, 2
    c, e                      // 2, 3
    d, r                      // 3, 4
)

综上所示,我们可以发现,iota就类似于在一个const的()中的一个行计数,当我们调用它时,不论我们使用与否,使用多少次,itap都会会对每一行进行一个计数。

2. 枚举类型

枚举用来表示只包含有限数量的固定值的类型,在开发中一般用于标识错误码或者状态机。拿一个实体对象的状态机来说,它通常与这个对象在数据库里对应记录的标识状态的字段值相对应。

Java 为我们提供了现成的枚举类的实现形式。但是在Go中并没有,因此我们可以借助type定义一个类型与iota结合实现枚举类的效果:

type Season int

const (
	Summer Season = 1
	Autumn        = 2
	Winter        = 3
	Spring        = 4
)

虽然底层的实现仍然是int类型,但是当我们使用int类型直接与Season类型的变量比较时,会出现编译不通过的错我,这样我们就可以对我们需要的枚举类型做类型限制了。

当然,我们也可以结合上面提到的itao,进行打优化:

type Season int

const (
	Summer Season = iota + 1
	Autumn        
	Winter        
	Spring        
)

到此这篇关于Go iota关键字与枚举类型实现原理的文章就介绍到这了,更多相关Go iota关键字 内容请搜索旗云源码以前的文章或继续浏览下面的相关文章希望大家以后多多支持旗云源码!

go iota 关键字 枚举 类型

原文链接:https://juejin.cn/post/7121660177815699487

评论