Dig101:Go之灵活的slice

newbmiao · · 98 次点击 · · 开始浏览    

Dig101: dig more, simplified more and know more

Slice作为go常用的数据类型,在日常编码中非常常见。 相对于数组的定长不可变,slice使用起来就灵活了许多。

也可以移步 微信版 阅读

0x01 slice 到底是什么?

首先我们看下源码中slice结构的定义

// src/runtime/slice.go
type slice struct {
	array unsafe.Pointer
	len   int
	cap   int
}
复制代码

slice数据结构如上,Data指向底层引用的数组内存地址, len是已用长度,cap是总容量。 为验证如上所述,我们尝试声明一个slice a,获取 a的sliceHeader头信息,并用%p获取&a, sh, a, a[0]的地址 看看他们的地址是否相同。

a := make([]int, 1, 3)
//reflect.SliceHeader 为 slice运行时数据结构
sh := (*reflect.SliceHeader)(unsafe.Pointer(&a))
fmt.Printf("slice header: %#v\naddress of a: %p &a[0]: %p |  &a: %p sh:%p ", 
    sh, a, &a[0],&a, sh)

//slice header: &reflect.SliceHeader{Data:0xc000018260, Len:1, Cap:3}
//address of a: 0xc000018260 &a[0]: 0xc000018260 | &a: 0xc00000c080 sh:0xc00000c080 
复制代码

结果发现a和&a[0]地址相同。 这个好理解,切片指向地址即对应底层引用数组首个元素地址 而&a和sh及sh.Data指向地址相同。这个是因为这三个地址是指slice自身地址。 这里【slice自身地址不同于slice指向的底层数据结构地址】, 清楚这一点对于后边的一些问题会更容易判断。

这里作为一个小插曲,我们看下当fmt.Printf("%p",a)时发生了什么 内部调用链 fmtPointer -> Value.Pointer 然后根据Pointer方法对应slice的注释如下

// If v's Kind is Slice, the returned pointer is to the first
// element of the slice. If the slice is nil the returned value
// is 0.  If the slice is empty but non-nil the return value is non-zero.
复制代码

发现没,正是我们上边说的,slice不为空时,返回了第一个元素的地址 有点迷惑性是不是,但其实作为使用slice的我们,更关心的是底层指向的数据不是么。

再一点就是,基于go中所有赋值和参数传递都是值传递,对于大数组而言,拷贝一个指向他的slice就高效多了 上一篇Go之for-range排坑指南 有过描述, 详见 0x03 对大数组这样遍历有啥问题?

总结下, slice是一个有底层数组引用的结构里,有长度,有容量。

就这么简单? 不,光这样还不足以让它比数组更好用。 slice还支持非常方便的切片操作和append时自动扩容,这让他更加flexible

0x02 slice能比较么?

答案是【只能和nil比较】

s := make([]int, 5)
a := s
println(s == a) 
//invalid operation: s == a (slice can only be compared to nil)
复制代码

这个也其实好理解,当你比较两个slice,你是想比较他们自身呢?(必然不同啊,因为有值拷贝) 还是比较他们底层的数组?(那长度和容量也一起比较么) 确实没有什么意义去做两个slice的比较。

0x03 花样的切片操作

slice通过三段参数来操作:x[from:len:cap] 即对x从from索引位置开始,截取len长度,cap大小的新切片返回 但是len和cap不能大于x原来的len和cap 三个参数都可省略,默认为x[0:len(x):cap(x)] 切片操作同样适用于array 如下都是通过src[:]常规对切片(指向的底层数组)或数组的引用

s:=make([]int,5)
x:=s[:]

arr:=[5]int{}
y:=arr[:]
复制代码

配合copy和append,slice的操作还有很多,官方wikiSlice Tricks 有更丰富的例子 比如更通用的拷贝 b = append(a[:0:0], a...) 比如cut或delete时增加对不使用指针的nil标记释放(防止内存泄露)

//Cut
copy(a[i:], a[j:])
for k, n := len(a)-j+i, len(a); k < n; k++ {
	a[k] = nil // or the zero value of T
}
a = a[:len(a)-j+i]

//Delete
if i < len(a)-1 {
  copy(a[i:], a[i+1:])
}
a[len(a)-1] = nil // or the zero value of T
a = a[:len(a)-1]
复制代码

不熟悉的话,建议好好练习一下去感受

0x04 append 时发生了什么?

总的来说,append时会按需自动扩容

  • 容量足够,无扩容则直接拷贝待 append 的数据到原 slice 底层指向的数组 之后(原slice的len之后),并返回指向该数组首地址的新slice(len改变)
  • 容量不够,有扩容则拷贝原有 slice 所指向部分数据到新开辟的数组,并对待 append 的数据附加到其后,并返回新数组首地址的新slice​(底层数组,len,cap均改变)

如下代码所示,容量不够时触发了扩容重新开辟底层数组,x 和 s 底层指向的数组已不是同一个

s := make([]int, 5)
x := append(s, 1)
fmt.Printf("x dataPtr: %p len: %d cap: %d\ns dataPtr: %p len: %d cap: %d", 
    x, len(x), cap(x), 
    s, len(s), cap(s))
// x dataPtr: 0xc000094000 len: 6 cap: 10
// s dataPtr: 0xc000092030 len: 5 cap: 5
复制代码

0x05 append内部优化

具体查阅源码,你会发现编译时将append分为三类并优化

除按需扩容外

  • x = append(y, make([]T, y)...) 使用memClr提高初始化效率
  • x = append(l1, l2...) 或者 x = append(slice, string) 直接复制l2
  • x = append(src, a, b, c) 确定待append数目下,直接做赋值优化

具体编译优化如下 注释有简化,详见internal/gc/walk.go: append

switch {
case isAppendOfMake(r):
// x = append(y, make([]T, y)...) will rewrite to

    // s := l1
    // n := len(s) + l2
    // if uint(n) > uint(cap(s)) {
    //   s = growslice(T, s, n)
    // }
    // s = s[:n]
    // lptr := &l1[0]
    // sptr := &s[0]
    // if lptr == sptr || !hasPointers(T) {
    //   // growslice did not clear the whole underlying array 
         // (or did not get called)
    //   hp := &s[len(l1)]
    //   hn := l2 * sizeof(T)
    //   memclr(hp, hn)
    // }

    //使用memClr提高初始化效率
    r = extendslice(r, init)
case r.IsDDD(): // DDD is ... syntax
// x = append(l1, l2...) will rewrite to

    // s := l1
    // n := len(s) + len(l2)
    // if uint(n) > uint(cap(s)) {
    //   s = growslice(s, n)
    // }
    // s = s[:n]
    // memmove(&s[len(l1)], &l2[0], len(l2)*sizeof(T))

    //直接复制l2
    r = appendslice(r, init) // also works for append(slice, string).
default:
// x = append(src, a, b, c) will rewrite to

    // s := src
    // const argc = len(args) - 1
    // if cap(s) - len(s) < argc {
    //     s = growslice(s, len(s)+argc)
    // }
    // n := len(s)
    // s = s[:n+argc]
    // s[n] = a
    // s[n+1] = b
    // ...

    //确定待append数目下,直接做赋值优化
    r = walkappend(r, init, n)
}
复制代码

这里关于append实现有几点可以提下

扩容的策略是什么?

答案是【总的来说是至少返回要求的长度n最大则为翻倍】 具体情况是:

  • len<1024时2倍扩容
  • 大于且未溢出时1.25倍扩容
  • 溢出则直接按申请大小扩容
  • 最后按mallocgc内存分配大小适配来确定len. (n-2n之间)

扩容留出最多一倍的余量,主要还是为了减少可能的扩容频率。 mallocgc内存适配实际是go内存管理做了内存分配的优化, 当然内部也有内存对齐的考虑。 雨痕Go学习笔记第四章内存分配,对这一块有很详尽的分析,值得一读。

至于为啥要内存对齐可以参见Golang 是否有必要内存对齐?,一篇不错的文章。

扩容判断中uint的作用是啥?

//n为目标slice总长度,类型int,cap(s)类型也为int
if uint(n) > uint(cap(s))
    s = growslice(T, s, n)
}
复制代码

答案是【为了避免溢出的扩容】

int有正负,最大值math.MaxInt64 = 1<<63 - 1 uint无负数最大值math.MaxUint64 = 1<<64 - 1 uint正值是int正值范围的两倍,int溢出了变为负数,uint(n)则必大于原s的cap,条件成立 到growslice内部,对于负值的n会panic,以此避免了溢出的扩容

内存清零初始化: memclrNoHeapPointers vs typedmemclr?

答案是【这个取决于待清零的内存是否已经初始化为type-safe(类型安全)状态,及类型是否包含指针】

具体来看,memclrNoHeapPointers使用场景是

  • 带清零内存是初始化过的,且不含指针
  • 带清零内存未初始化过的,里边内容是“垃圾值”(即非type-safe),需要初始化并清零

其他场景就是typedmemclr, 而且如果用于清零的Type(类型)包含指针,他会多一步WriteBarrier(写屏障),用于为GC(垃圾回收)运行时标记对象的内存修改,减少STW(stop the world)

所以memclrNoHeapPointers第一个使用场景为啥不含指针就不用解释了。

想了解更多可以看看zero-initialization-versus-zeroing 以及相关源码的注释memclrNoHeapPointerstypedmemclr

如有问题欢迎关注留言交流。

菜鸟Miao

文章首发地址:blog.newbmiao.com/2020/01/07/…

本文来自:掘金

感谢作者:newbmiao

查看原文:Dig101:Go之灵活的slice

入群交流(和以上内容无关):加入Go大咖交流群,或添加微信:muxilin131420 备注:入群;或加QQ群:729884609

98 次点击  
加入收藏 微博
暂无回复
添加一条新回复 (您需要 登录 后才能回复 没有账号 ?)
  • 请尽量让自己的回复能够对别人有帮助
  • 支持 Markdown 格式, **粗体**、~~删除线~~、`单行代码`
  • 支持 @ 本站用户;支持表情(输入 : 提示),见 Emoji cheat sheet
  • 图片支持拖拽、截图粘贴等方式上传