用 Go map 要注意这 1 个细节,避免依赖他!
2021-07-30 19:12:00 Author: mp.weixin.qq.com(查看原文) 阅读量:34 收藏

大家好,我是煎鱼。

最近又有同学问我这个日经话题,想转他文章时,结果发现我的公众号竟然没有发过,因此今天我再唠叨两句,好让大家避开这个 “坑”。

有的小伙伴没留意过 Go map 输出、遍历顺序,以为它是稳定的有序的,会在业务程序中直接依赖这个结果集顺序,结果栽了个大跟头,吃了线上 BUG。

有的小伙伴知道是无序的,但却不知道为什么,有的却理解错误?

奇怪的输出结果

今天通过本文,我们将揭开 for range map 输出的 “神秘” 面纱,看看它内部实现到底是怎么样的,顺序到底是怎么样?

开始吸鱼之路。

前言

例子如下:

func main() {
 m := make(map[int32]string)
 m[0] = "EDDYCJY1"
 m[1] = "EDDYCJY2"
 m[2] = "EDDYCJY3"
 m[3] = "EDDYCJY4"
 m[4] = "EDDYCJY5"

 for k, v := range m {
  log.Printf("k: %v, v: %v", k, v)
 }
}

假设运行这段代码,输出的结果是怎么样?是有序,还是无序输出呢?

k: 3, v: EDDYCJY4
k: 4, v: EDDYCJY5
k: 0, v: EDDYCJY1
k: 1, v: EDDYCJY2
k: 2, v: EDDYCJY3

从输出结果上来讲,是非固定顺序输出的,也就是每次都不一样。但这是为什么呢?

首先建议你先自己想想原因。其次我在面试时听过一些说法。有人说因为是哈希的所以就是无(乱)序等等说法。当时我是有点 ???

这也是这篇文章出现的原因,希望大家可以一起研讨一下,理清这个问题 :)

看一下汇编

    ...
 0x009b 00155 (main.go:11) LEAQ type.map[int32]string(SB), AX
 0x00a2 00162 (main.go:11) PCDATA $2$0
 0x00a2 00162 (main.go:11) MOVQ AX, (SP)
 0x00a6 00166 (main.go:11) PCDATA $2$2
 0x00a6 00166 (main.go:11) LEAQ ""..autotmp_3+24(SP), AX
 0x00ab 00171 (main.go:11) PCDATA $2$0
 0x00ab 00171 (main.go:11) MOVQ AX, 8(SP)
 0x00b0 00176 (main.go:11) PCDATA $2$2
 0x00b0 00176 (main.go:11) LEAQ ""..autotmp_2+72(SP), AX
 0x00b5 00181 (main.go:11) PCDATA $2$0
 0x00b5 00181 (main.go:11) MOVQ AX, 16(SP)
 0x00ba 00186 (main.go:11) CALL runtime.mapiterinit(SB)
 0x00bf 00191 (main.go:11) JMP 207
 0x00c1 00193 (main.go:11) PCDATA $2$2
 0x00c1 00193 (main.go:11) LEAQ ""..autotmp_2+72(SP), AX
 0x00c6 00198 (main.go:11) PCDATA $2$0
 0x00c6 00198 (main.go:11) MOVQ AX, (SP)
 0x00ca 00202 (main.go:11) CALL runtime.mapiternext(SB)
 0x00cf 00207 (main.go:11) CMPQ ""..autotmp_2+72(SP), $0
 0x00d5 00213 (main.go:11) JNE 193
 ...

我们大致看一下整体过程,重点处理 Go map 循环迭代的是两个 runtime 方法,如下:

  • runtime.mapiterinit
  • runtime.mapiternext

但你可能会想,明明用的是 for range 进行循环迭代,怎么出现了这两个函数,怎么回事?

看一下转换后

var hiter map_iteration_struct
for mapiterinit(typerange, &hiter); hiter.key != nil; mapiternext(&hiter) {
    index_temp = *hiter.key
    value_temp = *hiter.val
    index = index_temp
    value = value_temp
    original body
}

实际上编译器对于 slice 和 map 的循环迭代有不同的实现方式,并不是 for 一扔就完事了,还做了一些附加动作进行处理。而上述代码就是 for range map 在编译器展开后的伪实现

看一下源码

runtime.mapiterinit

func mapiterinit(t *maptype, h *hmap, it *hiter) {
 ...
 it.t = t
 it.h = h
 it.B = h.B
 it.buckets = h.buckets
 if t.bucket.kind&kindNoPointers != 0 {
  h.createOverflow()
  it.overflow = h.extra.overflow
  it.oldoverflow = h.extra.oldoverflow
 }

 r := uintptr(fastrand())
 if h.B > 31-bucketCntBits {
  r += uintptr(fastrand()) << 31
 }
 it.startBucket = r & bucketMask(h.B)
 it.offset = uint8(r >> h.B & (bucketCnt - 1))
 it.bucket = it.startBucket
    ...

 mapiternext(it)
}

通过对 mapiterinit 方法阅读,可得知其主要用途是在 map 进行遍历迭代时进行初始化动作。共有三个形参,用于读取当前哈希表的类型信息、当前哈希表的存储信息和当前遍历迭代的数据

为什么

咱们关注到源码中 fastrand 的部分,这个方法名,是不是迷之眼熟。没错,它是一个生成随机数的方法。再看看上下文:

...
// decide where to start
r := uintptr(fastrand())
if h.B > 31-bucketCntBits {
 r += uintptr(fastrand()) << 31
}
it.startBucket = r & bucketMask(h.B)
it.offset = uint8(r >> h.B & (bucketCnt - 1))

// iterator state
it.bucket = it.startBucket

在这段代码中,它生成了随机数。用于决定从哪里开始循环迭代。更具体的话就是根据随机数,选择一个桶位置作为起始点进行遍历迭代

因此每次重新 for range map,你见到的结果都是不一样的。那是因为它的起始位置根本就不固定!

runtime.mapiternext

func mapiternext(it *hiter) {
    ...
    for ; i < bucketCnt; i++ {
  ...
  k := add(unsafe.Pointer(b), dataOffset+uintptr(offi)*uintptr(t.keysize))
  v := add(unsafe.Pointer(b), dataOffset+bucketCnt*uintptr(t.keysize)+uintptr(offi)*uintptr(t.valuesize))
  ...
  if (b.tophash[offi] != evacuatedX && b.tophash[offi] != evacuatedY) ||
   !(t.reflexivekey || alg.equal(k, k)) {
   ...
   it.key = k
   it.value = v
  } else {
   rk, rv := mapaccessK(t, h, k)
   if rk == nil {
    continue // key has been deleted
   }
   it.key = rk
   it.value = rv
  }
  it.bucket = bucket
  if it.bptr != b {
   it.bptr = b
  }
  it.i = i + 1
  it.checkBucket = checkBucket
  return
 }
 b = b.overflow(t)
 i = 0
 goto next
}

在上小节中,咱们已经选定了起始桶的位置。接下来就是通过 mapiternext 进行具体的循环遍历动作。该方法主要涉及如下:

  • 从已选定的桶中开始进行遍历,寻找桶中的下一个元素进行处理
  • 如果桶已经遍历完,则对溢出桶 overflow buckets 进行遍历处理

通过对本方法的阅读,可得知其对 buckets 的遍历规则以及对于扩容的一些处理(这不是本文重点。因此没有具体展开)

总结

在本文开始,咱们先提出核心讨论点:“为什么 Go map 遍历输出是不固定顺序?”。

经过这一番分析,原因也很简单明了。就是 for range map 在开始处理循环逻辑的时候,就做了随机播种...

你想问为什么要这么做?

当然是官方有意为之,因为 Go 在早期(1.0)的时候,虽是稳定迭代的,但从结果来讲,其实是无法保证每个 Go 版本迭代遍历规则都是一样的。而这将会导致可移植性问题。

因此,改之。也请不要依赖...

参考

  • Go maps in action

关注煎鱼,吸取他的知识 👆

你好,我是煎鱼。高一折腾过前端,参加过国赛拿了奖,大学搞过 PHP。现在整 Go,在公司负责微服务架构等相关工作推进和研发。

从大学开始靠自己赚生活费和学费,到出版 Go 畅销书《Go 语言编程之旅》,再到获得 GOP(Go 领域最有观点专家)荣誉,点击蓝字查看我的出书之路

日常分享高质量文章,输出 Go 面试、工作经验、架构设计,加微信拉读者交流群,记得点赞!


文章来源: http://mp.weixin.qq.com/s?__biz=MzAxNzY0NDE3NA==&mid=2247487730&idx=2&sn=281a55ad0d2f8eca4078e59a567761c9&chksm=9be33f13ac94b605eb2092bed7d5c8147541951b184fa3e0d7e1c668122de73dff9f0b9bf3a6#rd
如有侵权请联系:admin#unsafe.sh