从源码理解Go语言切片扩容机制之容量大小

切片的扩容时，新分配的切片容量由定义在Go源码./src/runtime/slice.go文件中的nextslicecap函数的newcap计算策略和growslice函数中的内存对齐策略共同决定。内存对齐策略较为复杂，我们这里仅根据nextslicecap分析大概的切片容量newcap是如何计算的，nextslicecap函数go1.22.2版本代码如下：

// newLen通常是你当前切片的长度（即已经包含的元素数量）加上你希望添加的元素数量。
func nextslicecap(newLen, oldCap int) int {
	newcap := oldCap
	doublecap := newcap + newcap
	if newLen > doublecap {
		return newLen
	}

	const threshold = 256
	if oldCap < threshold {
		return doublecap
	}
    // 这个循环确保newcap至少满足newLen的期望大小。
	for {
		// Transition from growing 2x for small slices
		// to growing 1.25x for large slices. This formula
		// gives a smooth-ish transition between the two.
        
        // 翻译：这个公式为从对小切片的小贝扩容切换到对大切片的1.25倍扩容提供了一个丝滑的切换。
		newcap += (newcap + 3*threshold) >> 2

		// We need to check `newcap >= newLen` and whether `newcap` overflowed.
		// newLen is guaranteed to be larger than zero, hence
		// when newcap overflows then `uint(newcap) > uint(newLen)`.
		// This allows to check for both with the same comparison.

        // 翻译：我们需要检查newcap >= newLen和newcap是否溢出。因为newLen保证大于0，所以当newcap溢出时，满足uint(newcap) > uint(newLen)。利用这个特性我们可以同时检查上面的两个条件是否满足。
        // 注：unit()会将int类型转换为uint，当newcap溢出时，会变为一个负数，使用unit会将其转换为一个非常大的正数，从而使满足而退出循环。
		if uint(newcap) >= uint(newLen) {
			break
		}
	}

	// Set newcap to the requested cap when
	// the newcap calculation overflowed.
    
    // 翻译：newcap溢出时，将newcap设置为newLen。
	if newcap <= 0 {
		return newLen
	}
	return newcap
}

由上面源码可以得出知道newcap的值取决于newLen和oldcap，不同情况的结果如下：

1. 当newLen大于oldcap的两倍时，newcap的值等于newLen。
2. newLen小于等于oldcap的两倍时，当oldcap小于256，newcap的值为oldcap的两倍。
3. newLen小于等于oldcap的两倍时，当oldcap大于25，使用公式newcap +=（newcap + 3 * threshold) >> 2循环计算直到newcap至少等于newLen，如果计算溢出则newcap等于newLen。

计算出newcap之后，会进行内存对齐得到实际的切片容量。如果想要了解这部分内容可以看一下Go源码./src/runtime/slice.go文件中的growslice函数源码。