Go语言无需C.memcpy实现字节数组转uint32数组的高效方案

你之前直接调用copy报错是因为copy内置函数仅支持相同元素类型的切片拷贝，[]uint32和[]byte类型不匹配所以编译失败。以下是两种合法的替代方案：

方案1：使用标准库unsafe包（无CGO，性能最优） #

该方案不需要引入CGO，纯Go原生实现，性能和C.memcpy几乎一致，和你原C代码逻辑一致，默认适配小端序运行环境：

package main

import (
    "fmt"
    "unsafe"
)

func main() {
    // 注意原代码dest长度声明错误，16字节对应4个uint32，不是16个
    dest := make([]uint32, 4)
    src := []byte{0x11, 0x12, 0x13, 0x14,
        0x22, 0x22, 0x22, 0x22,
        0x33, 0x33, 0x33, 0x33,
        0x44, 0x44, 0x44, 0x44}

    // Go 1.17+ 可用unsafe.Slice实现安全转换，单条语句完成拷贝
    copy(dest, unsafe.Slice((*uint32)(unsafe.Pointer(&src[0])), len(src)/4))

    fmt.Printf("%04X %04X %04X %04X\n", dest[0], dest[1], dest[2], dest[3])
}

如果是低于1.17版本的Go，替换为以下写法即可：

// 低版本Go兼容写法
tmp := (*[1<<30 - 1]uint32)(unsafe.Pointer(&src[0]))[:len(src)/4:len(src)/4]
copy(dest, tmp)

方案2：纯安全Go实现（无unsafe，跨平台兼容） #

如果不想使用unsafe包担心内存安全问题，可以封装通用转换函数，现代Go编译器会对该类固定步长循环做高度优化，性能损耗极低，同时显式指定小端序可以做到跨大小端平台兼容：

package main

import (
    "fmt"
    "encoding/binary"
)

func bytesToUint32LittleEndian(src []byte) []uint32 {
    if len(src)%4 != 0 {
        panic("字节数组长度必须是4的整数倍")
    }
    dest := make([]uint32, len(src)/4)
    for i := 0; i < len(src); i += 4 {
        dest[i/4] = binary.LittleEndian.Uint32(src[i:i+4])
    }
    return dest
}

func main() {
    src := []byte{0x11, 0x12, 0x13, 0x14,
        0x22, 0x22, 0x22, 0x22,
        0x33, 0x33, 0x33, 0x33,
        0x44, 0x44, 0x44, 0x44}
    dest := bytesToUint32LittleEndian(src)
    fmt.Printf("%04X %04X %04X %04X\n", dest[0], dest[1], dest[2], dest[3])
}

注意事项 #

• unsafe方案需要确保输入字节数组长度是4的整数倍，否则会出现越界风险
• 若需要兼容大端序运行环境，仅需将方案2的binary.LittleEndian替换为binary.BigEndian即可

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