如何在不额外分配内存的情况下展平Go中的数组切片？

问题背景 #

需要将[][N]T类型的数组切片转换为[]T类型的普通切片，且要求不分配新的底层数组内存，避免遍历复制的性能开销。示例需求为实现convert([][4]byte) []byte函数，且方案需适配任意类型、任意固定长度的数组切片。

解决方案（针对[][4]byte的具体实现） #

利用Go的unsafe包直接操作内存地址，共享原切片的底层内存：

import "unsafe"

func convert(sliceOfArrays [][4]byte) []byte {
    if len(sliceOfArrays) == 0 {
        return nil
    }
    // 获取底层连续byte序列的起始指针
    ptr := unsafe.Pointer(&sliceOfArrays[0][0])
    // 计算总元素长度：数组切片长度 × 单个数组的元素数
    totalLen := len(sliceOfArrays) * 4
    // 计算总容量：数组切片容量 × 单个数组的元素数
    totalCap := cap(sliceOfArrays) * 4
    // 构造新切片并设置正确的容量
    return unsafe.Slice(ptr, totalLen)[:totalLen:totalCap]
}

通用泛型实现（Go 1.18+） #

适配任意类型T和任意固定长度N的数组切片：

import "unsafe"

func ConvertArraySlice[T any, N int](sliceOfArrays [][N]T) []T {
    if len(sliceOfArrays) == 0 {
        return nil
    }
    ptr := unsafe.Pointer(&sliceOfArrays[0][0])
    totalLen := len(sliceOfArrays) * N
    totalCap := cap(sliceOfArrays) * N
    return unsafe.Slice(ptr, totalLen)[:totalLen:totalCap]
}

原理说明 #

Go语言中，[][N]T类型的数组切片底层内存是连续的：

• 每个[N]T数组在内存中是N个连续的T值；
• 数组切片的所有元素（即这些数组）也是连续存储的，因此整个内存块本质上就是一段连续的T类型序列。
  通过unsafe.Pointer获取这段连续序列的起始地址，再用unsafe.Slice构造新切片，即可实现零内存分配的转换。

注意事项 #

• unsafe包的兼容性风险：该方案依赖Go的内存布局实现，虽然当前版本稳定，但未来语言内存模型若有变更可能导致失效。
• 内存生命周期绑定：新切片与原数组切片共享底层内存，需确保原切片在新切片的使用周期内始终有效，否则会出现悬空指针问题。
• 数组长度限制：泛型中的N必须是编译期常量，符合Go语言对数组长度的要求。
• 空切片判断：必须先检查原切片是否为空，否则访问sliceOfArrays[0][0]会触发索引越界 panic。

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