Go语言中不同类型切片之间的相互转换

chai2010 · · 6287 次点击 · · 开始浏览    
这是一个创建于 的文章,其中的信息可能已经有所发展或是发生改变。

[]T 切片转换为 []byte

类似C语言中将其他类型的数组转换为char数组:

func ByteSlice(slice interface{}) (data []byte) {
    sv := reflect.ValueOf(slice)
    if sv.Kind() != reflect.Slice {
        panic(fmt.Sprintf("ByteSlice called with non-slice value of type %T", slice))
    }
    h := (*reflect.SliceHeader)((unsafe.Pointer(&data)))
    h.Cap = sv.Cap() * int(sv.Type().Elem().Size())
    h.Len = sv.Len() * int(sv.Type().Elem().Size())
    h.Data = sv.Pointer()
    return
}

基于该函数,我们可以方便调用 []byte 类型参数的函数:

func SaveImageData(name string, data []color.RGBA) error {
    return ioutil.WriteFile(name, ByteSlice(data), 0666)
}

[]X 转换为 []Y 切片

类似C语言中将不同类型的数组转之间的相互转换:

func Slice(slice interface{}, newSliceType reflect.Type) interface{} {
    sv := reflect.ValueOf(slice)
    if sv.Kind() != reflect.Slice {
        panic(fmt.Sprintf("Slice called with non-slice value of type %T", slice))
    }
    if newSliceType.Kind() != reflect.Slice {
        panic(fmt.Sprintf("Slice called with non-slice type of type %T", newSliceType))
    }
    newSlice := reflect.New(newSliceType)
    hdr := (*reflect.SliceHeader)(unsafe.Pointer(newSlice.Pointer()))
    hdr.Cap = sv.Cap() * int(sv.Type().Elem().Size()) / int(newSliceType.Elem().Size())
    hdr.Len = sv.Len() * int(sv.Type().Elem().Size()) / int(newSliceType.Elem().Size())
    hdr.Data = uintptr(sv.Pointer())
    return newSlice.Elem().Interface()
}

转换时需要传入一个期望的目标切片类型,以 interface{} 形式返回转换后的切片。

比如,在图像处理中,转换 RGB 格式为 BGR 格式:

type RGB struct {
    R, G, B uint8
}
type BGR struct {
    B, G, R uint8
}

func RGB2BGR(data []RGB) []BGR {
    d := Slice(data, reflect.TypeOf([]BGR(nil)))
    return d.([]BGR)
}

类似于C语言中,将RGB指针转换为BGR指针的思路。 数据的底层结构并没有变化。

注意事项

该转换操作有一定的风险,用户需要自己保证安全。主要涉及以下几种类型:

  • 当结构体中含有指针时,转换会导致垃圾回收的问题。
  • 如果是 []byte[]T 可能会导致起始地址未对齐的问题 ([]byte 有可能从奇数位置切片)。
  • 该转换操作可能依赖当前系统,不同类型的处理器之间有差异。

该转换操作的优势是性能和类似void*的泛型,与cgo接口配合使用会更加理想。


有疑问加站长微信联系(非本文作者)

本文来自:开源中国博客

感谢作者:chai2010

查看原文:Go语言中不同类型切片之间的相互转换

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

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