• Go 语言切片扩容规则是扩容2倍?1.25倍?到底几倍


    本次主要来聊聊关于切片的扩容是如何扩的,还请大佬们不吝赐教

    切片,相信大家用了 Go 语言那么久这这种数据类型并不陌生,但是平日里聊到关于切片是如何扩容的,很多人可能会张口就来,切片扩容的时候,如果老切片的容量小于 1024 那么就再扩容 1倍,也就是新的切片容量是老切片容量的两倍,同理,如果老切片容量大于 1024,那么就扩容1.25 倍

    一个人这么说,多个人这么说,你可能就信了😂😂,可是大家都这么认为,我们就应该盲从吗?还是要自己去确认真实的扩容逻辑和实现方式,那就开始吧😁

    结论先行,切片对于扩容并不一定是 2 倍,1.25倍,这个要看实际情况

    本文分别从如下几点来聊聊切片的扩容

    • 扩容是针对切片的,数组无法扩容
    • 切片扩容到底是扩容到原来的几倍?
    • 我们一般使用切片的时候可以如何避免频繁的扩容?

    扩容是针对切片的,数组无法扩容

    首先需要明确,数组是不能扩容的,数组定义的时候就已经是定长的了,无法扩容

    切片是可以扩容的,我们可以通过 append 追加的方式来向已有的切片尾部进行追加,若原有切片已满,那么就会发生扩容

    另外,我们知道数组是一段连续的内存地址,同一种数据类型的数据集合,例如这样

    func main() {
       log.SetFlags(log.Lshortfile)
       var demoArray = [5]int{1, 2, 3, 4, 5}
       log.Print("unsafe.sizeof(int) == ",unsafe.Sizeof(demoArray[0]))
       for i, _ := range demoArray {
          log.Printf("&demoAraay[%d] == %p", i, &demoArray[i])
       }
     }
    
    • 1
    • 2
    • 3
    • 4
    • 5
    • 6
    • 7
    • 8

    可以看到在这个案例的环境中,一个 int 类型的变量占用 8 个字节,自然对于 demoArray 数组中,地址是连续的,每一个元素占用的空间也是我们所期望的

    那么切片的数据地址也是连续的吗??

    如果有人问这个问题,实际上是想问切片的底层数组的地址是不是也是连续的

    我们知道,切片 slice 在 Go 中是一个结构体,其中 array 字段是一个指针,指向了一块连续的内存地址,也就是底层数组

    type slice struct {
       array unsafe.Pointer
       len   int
       cap   int
    }
    
    • 1
    • 2
    • 3
    • 4
    • 5

    其中 len 字段记录了当前底层数组的实际有的元素个数,cap 表示底层数组的容量,自然也是切片slice 的容量

    func main(){
        var demoSli = []int{1,2,3,4,5}
        log.Printf("len == %d,cap == %d",len(demoSli),cap(demoSli))
        for i, _ := range demoSli {
           log.Printf("&demoSli[%d] == %p", i, &demoSli[i])
        }
    }
    
    • 1
    • 2
    • 3
    • 4
    • 5
    • 6
    • 7

    自然,demoSli 中的元素打印出来,地址也是连续的,没有毛病

    此处 xdm 模拟的时候,切勿去打印拷贝值的地址,例如下面这种方式是相当不明智的

    现在简单的去给 切片追加一个元素

    可以看到切片的容量变成了原来的两倍(容量从 5 扩容成 10),且切片中底层数组的元素地址自然也是连续的,不需要着急下结论,继续往下看,好戏在后头

    切片扩容到底是扩容到原来的几倍?

    案例1 向一个cap 为 0 的切片中追加 2000 个元素,查看被扩容了几次

    总共是扩容了 14 次

    可以看到切片容量小于 1024 时,触发扩容都是扩容到原来的 2 倍,但是 大于 1024 之后,有的是 1.25 倍,有的是 1.35 倍,有的大于 1.35 倍,那么这是为什么呢?后面统一看源码

    案例2 再次验证切片容量小于 1024,触发到扩容就一定是扩容 2 倍吗

    • 先初始化一个切片,里面有 5 个元素,len 为 5,cap 为 5
    • 再向切片中追加 6 个元素,分别是 6,7,8,9,10,11
    • 最终查看切片的容量是多少
    func main(){
        var demoSli = []int{1, 2, 3, 4, 5}
        log.Printf("len == %d,cap == %d", len(demoSli), cap(demoSli))
        for i, _ := range demoSli {
           log.Printf("&demoSli[%d] == %p", i, &demoSli[i])
        }
        
        demoSli = append(demoSli,6,7,8,9,10,11)
        log.Printf("len == %d,cap == %d",len(demoSli),cap(demoSli))
        for i, _ := range demoSli {
           log.Printf("&demoSli[%d] == %p", i, &demoSli[i])
        }
    }
    
    • 1
    • 2
    • 3
    • 4
    • 5
    • 6
    • 7
    • 8
    • 9
    • 10
    • 11
    • 12
    • 13

    通过这一段代码,我们可以看到,讲一个 len 为 5,cap 为 5 的切片,追加数字 6 的时候,切片应该要扩容到 10,然后追加到数字 11 的时候,切片应该扩容到 20,可实际真的是这样吗?

    xdm 可以将上述 demo 贴到自己环境试试,得到的结果仍然会是切片的容量 cap 最终是 12,并不是 20

    那么这一切都是为什么呢?我们来查看源码一探究竟

    源码赏析

    查看公共库中 runtime/slice.gogrowslice 函数就可以解开我们的疑惑

    可以看出在我们使用 append 对切片追加元素的时候,实际上会调用到 growslice 函数, growslice 中的核心逻辑我们就可以理解为计算基本的 newcap 和进行字节对齐

    1. 进行基本的新切片容量计算
    // 省略部分
    newcap := old.cap
    doublecap := newcap + newcap
    if cap > doublecap {
       newcap = cap
    } else {
       if old.cap < 1024 {
          newcap = doublecap
       } else {
          // Check 0 < newcap to detect overflow
          // and prevent an infinite loop.
          for 0 < newcap && newcap < cap {
             newcap += newcap / 4
          }
          // Set newcap to the requested cap when
          // the newcap calculation overflowed.
          if newcap <= 0 {
             newcap = cap
          }
       }
    }
    // 省略部分
    
    • 1
    • 2
    • 3
    • 4
    • 5
    • 6
    • 7
    • 8
    • 9
    • 10
    • 11
    • 12
    • 13
    • 14
    • 15
    • 16
    • 17
    • 18
    • 19
    • 20
    • 21
    • 22

    此处逻辑可以知道

    • 如果当前传入的 cap 是比原有切片 cap 的 2 倍还要大,那么就会按照当前传入的 cap 来作为新切片的容量
    • 否则去校验原有切片的容量是否小于 1024

      • 若小于 1024 ,则按照原有的切片容量的 2 倍进行扩容
      • 若大于等于 1024 ,那么就按照原有切片的 1.25 倍继续扩容

    然后是否看到这里就就结束了呢?就下定论来呢?并不,我们切莫断章取义,需要看全整个流程

    1. 进行基本的字节对齐

    growslice 函数 计算出基本的 newcap 之后,还需要按照类型进行基本的字节对齐,此处字节对齐之后主要是 roundupsize 的函数实现,顺便将其涉及到的常量放到一起给大家展示一波

    const (
       _MaxSmallSize = 32768
       smallSizeDiv = 8
       smallSizeMax = 1024
       largeSizeDiv = 128
       _NumSizeClasses = 68
       _PageShift = 13
    )
    func roundupsize(size uintptr) uintptr {
       if size < _MaxSmallSize {
          if size <= smallSizeMax-8 {
             return uintptr(class_to_size[size_to_class8[divRoundUp(size, smallSizeDiv)]])
          } else {
             return uintptr(class_to_size[size_to_class128[divRoundUp(size-smallSizeMax, largeSizeDiv)]])
          }
       }
       if size+ _PageSize < size {
          return size
       }
       return alignUp(size, _PageSize)
    }
    
    func divRoundUp(n, a uintptr) uintptr {
       // a is generally a power of two. This will get inlined and
       // the compiler will optimize the division.
       return (n + a - 1) / a
    }
    var size_to_class8 = [smallSizeMax/smallSizeDiv + 1]uint8{0, 1, 2, 3, 4, 5, 5, 6, 6, 7, 7, 8, 8, 9, 9, ...}
    
    • 1
    • 2
    • 3
    • 4
    • 5
    • 6
    • 7
    • 8
    • 9
    • 10
    • 11
    • 12
    • 13
    • 14
    • 15
    • 16
    • 17
    • 18
    • 19
    • 20
    • 21
    • 22
    • 23
    • 24
    • 25
    • 26
    • 27
    • 28

    光看这个函数,没啥感觉,函数逻辑比较简单,就是基本的计算和索引,那么我们讲上述的案例2带入,来计算一下

    此处很明确,当前旧的切片的 cap 为 5

    也就是 growslice 函数 中 old.cap 为 5,传入的 cap 为 11,因此 cap > 2*old.cap

    因此 newcap 此处等于 11

    开始计算字节对齐之后的结果

    • roundupsize(uintptr(newcap) * sys.PtrSize) ,其中 newcap = 11,sys.PtrSize = 8,则 roundupsize 参数传入 88 ,此环境指针占用 8 字节
    • 按照如下逻辑进行计算

      • divRoundUp(88, 8) = 11
      • size_to_class8[11] = 8
      • class_to_size[8] = 96

    此处环境我们的 int 类型是占用 8 个字节,因此最终的 newcap = 96/8 = 12

    经过上述源码的处理,最终我们就可以正常的得到最终切片容量被扩容到 12 ,xdm 可以去看实际的源码

    小结

    使用 append 进行切片扩容的时候,先会按照基本的逻辑来计算 newcap 的大小

    • 如果当前传入的cap是比原有切片cap的2倍还要大,那么就会按照当前传入的cap来作为新切片的容量,否则去校验原有切片的容量是否小于 1024

    • 若小于1024,则按照原有的切片容量的2倍进行扩容

    • 若大于等于 1024,那么就按照原有切片的 1.25 倍继续扩容
      最终再进行字节对齐

    那么实际上,最终的切片容量一般是会等于或者大于原有的 2倍 或者是 1.25 倍的

    我们一般使用切片的时候可以如何避免频繁的扩容?

    一般在使用切片的时候,尽量避免频繁的去扩容,我们可以对已知数据量的数据,进行一次性去分配切片的容量

    例如,数据量有 1000 个,那么我们就可以使用 make 的方式来进行初始化

    sli := make([]int, 0, 1000)
    
    • 1

    本次就是这样,如果对源码还挺感兴趣的话,xdm 可以去实际查看一下源码哦,希望对你有帮助

    感谢阅读,欢迎交流,点个赞,关注一波 再走吧

    欢迎点赞,关注,收藏

    朋友们,你的支持和鼓励,是我坚持分享,提高质量的动力

    技术是开放的,我们的心态,更应是开放的。拥抱变化,向阳而生,努力向前行。

    我是阿兵云原生,欢迎点赞关注收藏,下次见~

    文中提到的技术点,感兴趣的可以查看这些文章:

  • 相关阅读:
    IOS安全测试学习-DVIA-v2
    自动化测试框架-----unittest篇
    【GitHub主页】优化简历
    2022年SQL经典面试题总结(带解析)
    世界杯“引爆”东南亚电商狂潮,电商人如何选品和营销?
    jsp192ssm驾校报名预约管理系统
    Keil 厂商DFP pack实现原理
    使用pyqtgraph生成一个能够播放图片的窗口
    Android_JNI编程入门
    python 实现等声值线图绘制
  • 原文地址:https://blog.csdn.net/m0_37322399/article/details/133757694