如何实现字符串和byte切片的零拷贝转换

这是一个非常精典的例子。实现字符串和 bytes 切片之间的转换，要求是 zero-copy。想一下，一般的做法，都需要遍历字符串或 bytes 切片，再挨个赋值。

完成这个任务，我们需要了解 slice 和 string 的底层数据结构：

1
type StringHeader struct {
2
  Data uintptr
3
  Len  int
4
}
5

6
type SliceHeader struct {
7
  Data uintptr
8
  Len  int
9
  Cap  int
10
}

上面是反射包下的结构体，路径：src/reflect/value.go。只需要共享底层 []byte 数组就可以实现 zero-copy。

1
func string2bytes(s string) []byte {
2
  stringHeader := (*reflect.StringHeader)(unsafe.Pointer(&s))
3

4
  bh := reflect.SliceHeader{
5
    Data: stringHeader.Data,
6
    Len:  stringHeader.Len,
7
    Cap:  stringHeader.Len,
8
  }
9

10
  return *(*[]byte)(unsafe.Pointer(&bh))
11
}
12

13
func bytes2string(b []byte) string{
14
  sliceHeader := (*reflect.SliceHeader)(unsafe.Pointer(&b))
15

16
  sh := reflect.StringHeader{
17
    Data: sliceHeader.Data,
18
    Len:  sliceHeader.Len,
19
  }
20

21
  return *(*string)(unsafe.Pointer(&sh))
22
}

代码比较简单，不作详细解释。通过构造 slice header 和 string header，来完成 string 和 byte slice 之间的转换。

相关阅读