go底层系列-iota底层实现

iota

前言

• 我们知道iota常用于const表达式中
• 我们还知道其值是从零开始
• const声明块中每增加一行iota值自增1
• 使用iota可以简化常量定义，但其规则必须要牢牢掌握

热身

// 下面常量定义源于GO源码，下面每个常量的值是多少？
package main

type Priority int
const (
	LOG_EMERG Priority = iota
	LOG_ALERT
	LOG_CRIT
	LOG_ERR
	LOG_WARNING
	LOG_NOTICE
	LOG_INFO
	LOG_DEBUG
)
// 题目解释：
// 		上面代码源于日志模块，定义了一组代表日志级别的常量
//		常量类型为Priority，实际为int类型
// 参考答案：
//		iota初始值为0，也即LOG_EMERG值为0，下面每个常量递增1



// 下面代码取自Go源码，请问每个常量值是多少？
package main

const (
	mutexLocked = 1 << iota // mutex is locked
	mutexWoken
	mutexStarving
	mutexWaiterShift = iota
	starvationThresholdNs = 1e6
)
// 题目解释：
//		以上代码取自Go互斥锁Mutex的实现，用于指示各种状态位的地址偏移
// 参考答案：
//		mutexLocked == 1；
//		mutexWoken == 2；
//		mutexStarving == 4；
//		mutexWaiterShift == 3；
//		starvationThresholdNs == 1000000



// 请问每个常量值是多少？
package main

const (
	bit0, mask0 = 1 << iota, 1<<iota - 1
	bit1, mask1
	_, _
	bit3, mask3
)
// 题目解释：
//		以上代码取自Go官方文档
// 参考答案：
// bit0 == 1， mask0 == 0， bit1 == 2， mask1 == 1， bit3 == 8， mask3 == 7

规则

• 很多书上或博客描述的规则是这样的：
  • iota在const关键字出现时被重置为0
  • const声明块中每新增一行iota值自增1
• 我曾经也这么理解，看过编译器代码后发现，其实规则只有一条：
  • iota代表了const声明块的行索引（下标从0开始）
  • 这样理解更贴近编译器实现逻辑，也更准确。
• 除此之外，const声明还有个特点
  • 即第一个常量必须指定一个表达 式
  • 后续的常量如果没有表达式，则继承上面的表达式。

编译原理

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

• ValueSpec.Names：这个切片中保存了一行中定义的常量
  • 如果一行定义N个常量，那么 ValueSpec.Names切片长度即为N
• const块实际上是spec类型的切片，用于表示const中的多行

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

• 可以更清晰的看出iota实际上是遍历const块的索引
• 每行中即便多次使用iota，其值也不会递增