本次主要来聊聊关于切片的扩容是如何扩的,还请大佬们不吝赐教
切片,相信大家用了 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])
}
}
可以看到在这个案例的环境中,一个 int 类型的变量占用 8 个字节,自然对于 demoArray 数组中,地址是连续的,每一个元素占用的空间也是我们所期望的
那么切片的数据地址也是连续的吗??
如果有人问这个问题,实际上是想问切片的底层数组的地址是不是也是连续的
我们知道,切片 slice 在 Go 中是一个结构体,其中 array 字段是一个指针,指向了一块连续的内存地址,也就是底层数组
type slice struct {
array unsafe.Pointer
len int
cap int
}
其中 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])
}
}
自然,demoSli 中的元素打印出来,地址也是连续的,没有毛病
此处 xdm 模拟的时候,切勿去打印拷贝值的地址,例如下面这种方式是相当不明智的
现在简单的去给 切片追加一个元素
可以看到切片的容量变成了原来的两倍(容量从 5 扩容成 10),且切片中底层数组的元素地址自然也是连续的,不需要着急下结论,继续往下看,好戏在后头
总共是扩容了 14 次
可以看到切片容量小于 1024 时,触发扩容都是扩容到原来的 2 倍,但是 大于 1024 之后,有的是 1.25 倍,有的是 1.35 倍,有的大于 1.35 倍,那么这是为什么呢?后面统一看源码
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])
}
}
通过这一段代码,我们可以看到,讲一个 len 为 5,cap 为 5 的切片,追加数字 6 的时候,切片应该要扩容到 10,然后追加到数字 11 的时候,切片应该扩容到 20,可实际真的是这样吗?
xdm 可以将上述 demo 贴到自己环境试试,得到的结果仍然会是切片的容量 cap 最终是 12,并不是 20
那么这一切都是为什么呢?我们来查看源码一探究竟
查看公共库中 runtime/slice.go
的 growslice 函数就可以解开我们的疑惑
可以看出在我们使用 append 对切片追加元素的时候,实际上会调用到 growslice 函数, growslice 中的核心逻辑我们就可以理解为计算基本的 newcap 和进行字节对齐
// 省略部分
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
}
}
}
// 省略部分
此处逻辑可以知道
否则去校验原有切片的容量是否小于 1024
然后是否看到这里就就结束了呢?就下定论来呢?并不,我们切莫断章取义,需要看全整个流程
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, ...}
光看这个函数,没啥感觉,函数逻辑比较简单,就是基本的计算和索引,那么我们讲上述的案例2带入,来计算一下
此处很明确,当前旧的切片的 cap 为 5
也就是 growslice 函数 中 old.cap 为 5,传入的 cap 为 11,因此 cap > 2*old.cap
因此 newcap 此处等于 11
开始计算字节对齐之后的结果
按照如下逻辑进行计算
此处环境我们的 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)
本次就是这样,如果对源码还挺感兴趣的话,xdm 可以去实际查看一下源码哦,希望对你有帮助
感谢阅读,欢迎交流,点个赞,关注一波 再走吧
朋友们,你的支持和鼓励,是我坚持分享,提高质量的动力
技术是开放的,我们的心态,更应是开放的。拥抱变化,向阳而生,努力向前行。
我是阿兵云原生,欢迎点赞关注收藏,下次见~
文中提到的技术点,感兴趣的可以查看这些文章: