GO语言slice详解(结合源码)

一、GO语言中slice的定义

slice 是一种结构体类型,在源码中的定义为:

src/runtime/slice.go

type slice struct {
	array unsafe.Pointer
	len   int
	cap   int
}

从定义中可以看到,slice是一种值类型,里面有3个元素。array是数组指针,它指向底层分配的数组;len是底层数组的元素个数;cap是底层数组的容量,超过容量会扩容。

二、初始化操作

slice有三种初始化操作,请看下面代码:

package main
import "fmt"
func main() {
	//1、make
	a := make([]int32, 0, 5)
	//2、[]int32{}
	b := []int32{1, 2, 3}
	//3、new([]int32)
	c := *new([]int32)
	fmt.Println(a, b, c)
}

这几种初始化方式,在底层实现是不一样的。有一种了解底层实现好的方法,就是看反汇编的调用函数。运行下面命令即可看到代码某一行的反汇编:

go tool compile -S plan9Test.go | grep plan9Test.go:行号

1、make初始化

make函数初始化有三个参数,第一个是类型,第二个长度,第三个容量,容量要大于等于长度。slice的make初始化调用的是底层的runtime.makeslice函数。

func makeslice(et *_type, len, cap int) slice {
	// NOTE: The len > maxElements check here is not strictly necessary,
	// but it produces a 'len out of range' error instead of a 'cap out of range' error
	// when someone does make([]T, bignumber). 'cap out of range' is true too,
	// but since the cap is only being supplied implicitly, saying len is clearer.
	// See issue 4085.
	maxElements := maxSliceCap(et.size)
	if len < 0 || uintptr(len) > maxElements {
		panic(errorString("makeslice: len out of range"))
	}

	if cap < len || uintptr(cap) > maxElements {
		panic(errorString("makeslice: cap out of range"))
	}

	p := mallocgc(et.size*uintptr(cap), et, true)
	return slice{p, len, cap}
}

主要就是调用mallocgc分配一块 个数cap*类型大小 的内存给底层数组,然后返回一个slice,slice的array指针指向分配的底层数组。

2、[]int32{} 初始化

这种初始化底层是调用 runtime.newobject 函数直接分配相应个数的底层数组。

// implementation of new builtin
// compiler (both frontend and SSA backend) knows the signature
// of this function
func newobject(typ *_type) unsafe.Pointer {
	return mallocgc(typ.size, typ, true)
}

3、new([]int32) 初始化

这种初始化底层也是调用 runtime.newobject ,new是返回slice 的地址,所以要取地址里面内容才是真正的slice。

三、reSlice(切片操作)

所谓reSlice,是基于已有 slice 创建新的 slice 对象,以便在容量cap允许范围内调整属性。

data := []int32{0,1,2,3,4,5,6}
slice := data[1:4:5]  // [low:high:max]

切片操作有三个参数,low、high、max,新生成的 slice 结构体三个参数,指针array指向原slice 底层数组元素下标为low的位置, len = high - low, cap = max - low。如下图所示:

切片操作示意图

切片操作主要要注意的就是在原slice 容量允许范围,超出容量范围会报panic

四、append 操作

slice 的 append 操作是向底层数组尾部添加数据,返回 新的slice对象

请看下面一段代码:

package main
import (
	"fmt"
)
func main() {
	a := make([]int32, 1, 2)
	b := append(a, 1)
	c := append(a, 1, 2)
	fmt.Printf("a的地址:%p, 第一个元素地址:%p,容量:%v\n", &a, &a[0], cap(a))
	//a的地址:0xc42000a060, 第一个元素地址:0xc42001a090,容量:2
	fmt.Printf("b的地址:%p, 第一个元素地址:%p,容量:%v\n", &b, &b[0], cap(b))
	//b的地址:0xc42000a080, 第一个元素地址:0xc42001a090,容量:2
	fmt.Printf("c的地址:%p, 第一个元素地址:%p,容量:%v\n", &c, &c[0], cap(c))
	//c的地址:0xc42000a0a0, 第一个元素地址:0xc42001a0a0,容量:4
}

从上面代码的打印结果中可以看出:a 是一个底层数组有一个元素,容量为2的slice;append 1个元素后,没有超出容量,产生了一个新的slice b,a 和 b 底层数组首元素相同地址,说明a,b共用底层数组;append 2个元素,超出了容量,产生一个新的slice c,c的底层数组地址变了,容量也翻倍了。
那么得出结论,append操作的运行过程:
1、如果添加数据后没有超过原始容量,新的slice对象 和原始slice共用底层数组,len 数据会变化,cap数据不变;
2、添加数据后超过了容量那就会扩容,重新分配一个新的底层数组,然后拷贝底层数组数据过去,那么append后产生的新slice对象和原始的slice就没有任何关系了。

扩容机制

看汇编代码可以知道,扩容调用的是底层函数 runtime.growslice
这个函数是这样定义的:
func growslice(et *_type, old slice, cap int) slice {}
这个函数传入三个参数:slice的原始类型,原始slice,期望的最小容量;返回一个新的slice,新slice 至少是拥有期望的最小容量,元素从原slice copy过来。

扩容规则主要是下面这段代码:

newcap := old.cap
	doublecap := newcap + newcap
	if cap > doublecap {
		newcap = cap
	} else {
		if old.len < 1024 {
			newcap = doublecap
		} else {
			for newcap < cap {
				newcap += newcap / 4
			}
		}
	}

扩容规则就是两点:

  1. 如果期望的最小容量大于原始的两倍容量时,那么新的容量就是等于期望的最小容量;
  2. 不满足第一种情况,那么判断原slice的底层数组元素长度是不是小于1024。小于1024,新容量是原来的两倍;大于等于1024 ,新容量是原来的1.25倍。

上面是扩容的基本规则判断,实际上扩容还要考虑到内存对齐情况:

var lenmem, newlenmem, capmem uintptr
	const ptrSize = unsafe.Sizeof((*byte)(nil))
	switch et.size {
	case 1:
		lenmem = uintptr(old.len)
		newlenmem = uintptr(cap)
		capmem = roundupsize(uintptr(newcap))
		newcap = int(capmem)
	case ptrSize:
		lenmem = uintptr(old.len) * ptrSize
		newlenmem = uintptr(cap) * ptrSize
		capmem = roundupsize(uintptr(newcap) * ptrSize)
		newcap = int(capmem / ptrSize)
	default:
		lenmem = uintptr(old.len) * et.size
		newlenmem = uintptr(cap) * et.size
		capmem = roundupsize(uintptr(newcap) * et.size)
		newcap = int(capmem / et.size)
	}
内存对齐之后,扩容的倍数就会 >= 2 或则 1.25 了。

为什么要内存对齐?

1.平台原因(移植原因):不是所有的硬件平台都能访问任意地址上的任意数据的;某些硬件平台只能在某些地址处取某些特定类型的数据,否则抛出硬件异常。
2.性能原因:数据结构(尤其是栈)应该尽可能地在自然边界上对齐。原因在于,为了访问未对齐的内存,处理器需要作两次内存访问;而对齐的内存访问仅需要一次访问。

五、函数调用中实参和形参的相互影响

1、slice值传递,在调用函数中直接操作底层数组

来看下面一段代码:

package main
import (
	"fmt"
)

func OpSlice(b []int32) {
	fmt.Printf("len: %d, cap: %d, data:%+v \n", len(b), cap(b), b)
	//len: 5, cap: 5, data:[1 2 3 4 5]
	fmt.Printf("b第一个元素地址:%p\n", &b[0])
	//b第一个元素地址:0xc420016120

	b[0] = 100
	fmt.Printf("len: %d, cap: %d, data:%+v \n", len(b), cap(b), b)
	//len: 5, cap: 5, data:[100 2 3 4 5]
	fmt.Printf("b第一个元素地址:%p\n", &b[0])
	//b第一个元素地址:0xc420016120
}

func main() {
	a := []int32{1, 2, 3, 4, 5}
	fmt.Printf("len: %d, cap: %d, data:%+v \n", len(a), cap(a), a)
	//len: 5, cap: 5, data:[1 2 3 4 5]
	fmt.Printf("a第一个元素地址:%p\n", &a[0])
	//a第一个元素地址:0xc420016120
	OpSlice(a)

	fmt.Printf("len: %d, cap: %d, data:%+v \n", len(a), cap(a), a)
	//len: 5, cap: 5, data:[100 2 3 4 5]
	fmt.Printf("a第一个元素地址:%p\n", &a[0])
	//a第一个元素地址:0xc420016120
}

从这段代码的打印中可以看到:
main函数中的slice a 是实参,值传递给调用函数时,要临时拷贝一份给b,所以a,b 的地址是不一样的,slice b 结构体中的三个元素都是a中的拷贝,但是元素array是指针,指针的拷贝还是指针,他们指向同一块底层数组,所以a,b底层数组的第一个元素地址是一样的。a,b共用同一块底层数组,在调用函数中,直接改变b的第一个元素内容,函数返回后a的第一个元素也变了,相当于改变了实参。

2、slice 指针传递

slice 指针传递就没什么说的了,在被调用函数中相当于操作的是实参中同一个slice,所有修改都会反映到实参。

3、slice 切片传递

不扩容的情况,来看下面一段代码:

package main
import (
	"fmt"
)
func OpSlice(b []int32) {
	fmt.Printf("len: %d, cap: %d, data:%+v \n", len(b), cap(b), b)
	//len: 3, cap: 9, data:[1 2 3]
	fmt.Printf("b第一个元素地址:%p\n", &b[0])
	//b第一个元素地址:0xc42007a064

	b = append(b, 100)
	fmt.Printf("len: %d, cap: %d, data:%+v \n", len(b), cap(b), b)
	//len: 4, cap: 9, data:[1 2 3 100]
	fmt.Printf("b第一个元素地址:%p\n", &b[0])
	//b第一个元素地址:0xc42007a064

}

func main() {
	a := []int32{0, 1, 2, 3, 4, 5, 6, 7, 8, 9}
	fmt.Printf("a第2个元素地址:%p\n", &a[1])
	//a第2个元素地址:0xc42007a064

	fmt.Printf("len: %d, cap: %d, data:%+v \n", len(a), cap(a), a)
	//len: 10, cap: 10, data:[0 1 2 3 4 5 6 7 8 9]
	fmt.Printf("a第一个元素地址:%p\n", &a[0])
	//a第一个元素地址:0xc42007a060
	OpSlice(a[1:4])

	fmt.Printf("len: %d, cap: %d, data:%+v \n", len(a), cap(a), a)
	//len: 10, cap: 10, data:[0 1 2 3 100 5 6 7 8 9]
	fmt.Printf("a第一个元素地址:%p\n", &a[0])
	//a第一个元素地址:0xc42007a060

}

前面已经讲过,切片和原slice是共用底层数组的。不扩容情况下,对切片产生的新的slice append 操作,新增加的元素会添加到底层数组尾部,会覆盖原有的值,反映到原slice中去;

总结

无论是slice的什么操作:拷贝,append,reSlice 等等都会产生新的slice,但是他们是共用底层数组的,不扩容情况,他们增删改元素都会影响到原来的slice底层数组;扩容情况下,产生的是一个“真正的”新的slice对象,和原来的完全独立开了,底层数组完全不会影响。

参考资料

  1. 深度解密Go语言之Slice.
  2. Go语言学习笔记.
posted @ 2020-02-21 15:29  程序员大天地  阅读(3896)  评论(3编辑  收藏  举报