Go不同切片头共享数据时逃逸分析失效，如何避免数据拷贝？

可行的零拷贝解决方案 #

核心逻辑是让原[]int32的底层数组分配到堆上，堆内存不会随函数返回被释放，因此共享底层数组的[]byte可以安全使用，全程没有额外的数据拷贝。

方案1：使用runtime.KeepAlive强制延长生命周期 #

只需要在pack函数返回前调用runtime.KeepAlive标记原切片仍然存活即可，逃逸分析会自动将ints分配到堆上，避免栈释放导致的野指针问题：

import "runtime"

func pack() []byte {
   ints := []int32{1,2,3,4,5}
   bytes := intsToBytes(ints)
   runtime.KeepAlive(ints) // 保证ints在返回前不会被回收，强制逃逸到堆
   return bytes 
}

方案2：用unsafe指针骗过逃逸分析 #

如果不想引入runtime包的依赖，可以用空的unsafe指针引用标记原切片地址被泄露，同样可以触发堆分配：

import "unsafe"

func pack() []byte {
   ints := []int32{1,2,3,4,5}
   _ = unsafe.Pointer(&ints) // 标记切片地址被外部引用，强制逃逸到堆
   bytes := intsToBytes(ints)
   return bytes 
}

更安全的转换实现 #

Go 1.17之后官方推荐使用unsafe.Slice代替直接操作reflect.SliceHeader，避免临时切片头被回收导致的野指针问题，更安全的intsToBytes实现如下：

import "unsafe"

const SizeOfInt32 = 4

func intsToBytes(i []int32) []byte {
    if len(i) == 0 {
        return nil
    }
    // 直接基于原切片的首指针生成[]byte，不操作切片头拷贝
    return unsafe.Slice((*byte)(unsafe.Pointer(&i[0])), len(i)*SizeOfInt32)
}

注意事项 #

• 这种零拷贝转换得到的[]byte的字节序和当前CPU的原生字节序一致，如果需要跨设备/跨架构传输数据，需要自行处理字节序转换。
• 所有基于unsafe的操作都不符合Go的内存安全保证，升级Go版本时需要做好兼容性测试。

内容的提问来源于stack exchange，提问作者Braden Shill