你还在以为golang切片扩容是1/4的增加吗?其实早就已经变啦!

前言

1. 最近看go语言的切片原理时发现实际的扩容规则和一些博客和书上讲的不一致,所以看看各版本的源码中切片的扩容到底是怎么做的,看源码是最快捷且准确的方式,
   go语言的源码可以在仓库https://github.com/golang/go上找到,当然也可以在官网下载下来再去翻看,这里不再赘述

切片扩容对比

切片的扩容代码主要在各版本src/runtime/slice.go文件中,主要的扩容代码可以看growslice这个函数

Go1.17(这个版本中的扩容也是目前大部分书籍和博客中讲到的切片扩容内容)

func growslice(et *_type, old slice, cap int) slice {
	...

	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
			}
		}
	}

	...内存对齐...

	return slice{p, old.len, newcap}
}

1. 这里的主要扩容核心逻辑为:
    1. 如果新申请的容量(cap)大于旧容量(old.cap)的两倍,则最终容量(newcap)是新申请的容量(cap)
    2. 如果旧切片的长度小于1024,则最终容量是旧容量的2倍,即newcap=doublecap
    3. 如果旧切片的长度大于或者等于1024,则最终容量从旧容量开始循环增加原来的1/4,直到最终容量大于或等于新申请的容量
    4. 如果最终容量计算值溢出,即超过了int的最大范围,则最终容量就是新申请的容量

Go1.18(这个是目前实际的扩容规则,虽然在1.20中入参变了,但是核心的扩容逻辑没有变,所以还是以1.18举例)

func growslice(et *_type, old slice, cap int) slice {
	...

	newcap := old.cap
	doublecap := newcap + newcap
	if cap > doublecap {
		newcap = cap
	} else {
		const threshold = 256
		if old.cap < threshold {
			newcap = doublecap
		} else {
			// Check 0 < newcap to detect overflow
			// and prevent an infinite loop.
			for 0 < newcap && newcap < cap {
				// Transition from growing 2x for small slices
				// to growing 1.25x for large slices. This formula
				// gives a smooth-ish transition between the two.
				newcap += (newcap + 3*threshold) / 4
			}
			// Set newcap to the requested cap when
			// the newcap calculation overflowed.
			if newcap <= 0 {
				newcap = cap
			}
		}
	}

	...内存对齐...

	return slice{p, old.len, newcap}
}

1. 这里的扩容核心逻辑是:
    1. 如果新申请的容量(cap)大于旧容量(old.cap)的两倍,则最终容量(newcap)是新申请的容量(cap)
    2. 定义了一个阈值为256,如果旧容量小于256,那么最终容量(newcap)是旧容量(old.cap)的两倍
    3. 如果旧容量(old.cap)大于或等于256,那么最终容量开始从(旧容量+3*256)/4开始循环,直到最终容量大于等于申请的容量
    4. 如果最终容量计算值溢出,即超过了int的最大范围,则最终容量就是新申请的容量

结语

可以看到切片的扩容规则在1.18进行了修改,直到本文发出(截至1.21为止),当前的扩容逻辑和1.18是大致一致的
如果在使用过程中发现与书中或博客中不一致为问题,还是以源码为主,源码是不会骗人的!!!