Golang iota

yi个俗人 2022-09-29 我要评论

前言

我们知道iota是go语言的常量计数器，只能在常量的const表达式中使用，在const关键字出现的时将被重置为0，const中每新增一行常量声明iota值自增1（iota可以理解为const语句块中的行索引），使用iota可以简化常量的定义，但其规则必须要牢牢掌握，否则在我们开发中可能会造成误解，本文尝试全面总结其使用用法以及其实现原理，需要的朋友可以参考以下内容，希望对大家有帮助。

iota的使用

iota在const关键字出现时将被重置为0

iota只能在常量的表达式中使用，iota在const关键字出现时将被重置为0。不同const定义块互不干扰。

//const关键字出现时将被重置为0
const (
    a = iota  //0
    b         //1
)
//不同const定义块互不干扰
const (
    c = iota //0
)

按行计数

const每新增一行常量声明，iota计数一次，可以当做const语句中的索引，常用于定义枚举数据。

const (
    n1 = iota //0
    n2 		  //1
    n3 		  //2
    n4 		  //3
)

所有注释行和空行全部忽略

所有注释行和空行在编译时期首先会被清除，所以空行不计数。

const (
    a = iota  //0
    
    b		  //1
    
    //此行是注释
    c         //2
)

跳值占位

如果某个值不需要，可以使用占位 “_”，它不是空行，会进行计数，起到跳值作用。

const (
    a = iota //0
    _
    b        //2
)

多个iota

同一const块出现多个iota，只会按照行数计数，不会重新计数。

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

一行多个iota

一行多个iota，分别计数。

const (
    a, b = iota, iota // a=0,b=0
    c, d              // c=1,d=1
)

首行插队

开头插队会进行计数。

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

中间插队

中间插队会进行计数。

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

没有表达式的常量定义复用上一行的表达式

const (
   a = iota       //  iota = 0 
   b = 1 + iota  // iota = 1
   c           // iota = 2
)

实现原理

iota定义

iota 源码在 Go 语言代码库中的定义位于内建文件 go/src/builtin/builtin.go 中：

const iota = 0 // Untyped int.iota

在这里声明了一个常量标识符，它的值是0；iota只是一个简单的整数0，为什么能作为常量计数器进行自增的，我们再看一下const的实现。

const

const 块中每一行在 Go 中使用 spec 数据结构描述， spec 声明如下：

ValueSpec struct {
 Doc *CommentGroup // associated documentation; or nil
 Names []*Ident // value names (len(Names) > 0)
 Type Expr // value type; or nil
 Values []Expr // initial values; or nil
 Comment *CommentGroup // line comments; or nil
}

在这个结构体中有一个切片 ValueSpec.Names，此切片中保存了一行中定义的常量，如果一行定义N个常量，那么 ValueSpec.Names 切片长度即为N。

const块实际上是spec类型的切片，用于表示const中的多行。

编译期间构造常量时的伪算法如下：

for iota, spec := range ValueSpecs {
 for i, name := range spec.Names {
 obj := NewConst(name, iota...) //此处将iota传入，用于构造常量
 ...
 }
}

iota实际上是遍历const块的索引，每行中即便多次使用iota，其值也不会递增。