快捷導(dǎo)航

golang中struct和[]byte的相互轉(zhuǎn)換示例

更新時間：2018年07月23日 10:34:18 作者：穩(wěn)健明

這篇文章主要介紹了golang中struct和[]byte的相互轉(zhuǎn)換示例,小編覺得挺不錯的，現(xiàn)在分享給大家，也給大家做個參考。一起跟隨小編過來看看吧

在網(wǎng)絡(luò)傳輸過程中，經(jīng)常會這樣處理：socket接收到數(shù)據(jù)，先獲取其消息頭，然后再做各種不同的業(yè)務(wù)處理。在解析消息頭的時候的方法有多種多樣。其中最為高效解析消息頭的方法就是直接把數(shù)據(jù)頭部分強制類型轉(zhuǎn)換為對應(yīng)的消息頭結(jié)構(gòu)體。這種做法在C/C++中非常的常見。而golang其實也是可以這樣子做的。類似這樣的應(yīng)用，直接類型轉(zhuǎn)換獲取消息對應(yīng)的解析方法其實效率會相對較高。

golang中struct和[]byte的轉(zhuǎn)換方法，其實就是用到了golang中的unsafe包加上類型轉(zhuǎn)換 , 約束：struct中不能有指針類型。

1、struct轉(zhuǎn)化為[]byte，轉(zhuǎn)換方法如下：

import (
  "fmt"
  "unsafe"
)
type TestStructTobytes struct {
  data int64
}
type SliceMock struct {
  addr uintptr
  len int
  cap int
}

func main() {

  var testStruct = &TestStructTobytes{100}
  Len := unsafe.Sizeof(*testStruct)
  testBytes := &SliceMock{
    addr: uintptr(unsafe.Pointer(testStruct)),
    cap: int(Len),
    len: int(Len),
  }
  data := *(*[]byte)(unsafe.Pointer(testBytes))
  fmt.Println("[]byte is : ", data)
}

運行結(jié)果：

[]byte is : [100 0 0 0 0 0 0 0]

因為[]byte底層的數(shù)據(jù)結(jié)構(gòu)為：

struct { 
 addr uintptr 
 len int 
 cap int 
 }

其中addr為數(shù)值的地址，len為當?shù)財?shù)值的長度，cap為數(shù)值的容量。

轉(zhuǎn)換的時候，需要定義一個和[]byte底層結(jié)構(gòu)一致的struct（如例子中的SliceMock），然后把結(jié)構(gòu)體的地址賦給addr，結(jié)構(gòu)體的大小賦給len和cap。最后將其轉(zhuǎn)換為[]byte類型。

2、將[]byte轉(zhuǎn)換為struct,轉(zhuǎn)換方法如下：

import (
  "fmt"
  "unsafe"
)
type TestStructTobytes struct {
  data int64
}
type SliceMock struct {
  addr uintptr
  len int
  cap int
}

func main() {

  var testStruct = &TestStructTobytes{100}
  Len := unsafe.Sizeof(*testStruct)
  testBytes := &SliceMock{
    addr: uintptr(unsafe.Pointer(testStruct)),
    cap: int(Len),
    len: int(Len),
  }
  data := *(*[]byte)(unsafe.Pointer(testBytes))
  fmt.Println("[]byte is : ", data)
  var ptestStruct *TestStructTobytes = *(**TestStructTobytes)(unsafe.Pointer(&data))
  fmt.Println("ptestStruct.data is : ", ptestStruct.data)
}

運行結(jié)果：

[]byte is : [100 0 0 0 0 0 0 0]
ptestStruct.data is : 100

從上面的例子中將[]byte轉(zhuǎn)換為struct的代碼片段為：

var ptestStruct *TestStructTobytes = *(**TestStructTobytes)(unsafe.Pointer(&data))

分析：

由于在golang中編譯器不將[]byte視為指針，所以要使用其地址進行轉(zhuǎn)換，由于[]byte底層存放的是指向數(shù)據(jù)的地址。用[]byte的地址就需要使用雙層指針轉(zhuǎn)換，然后再指向其內(nèi)容，得出來的就是轉(zhuǎn)換對應(yīng)struct的指針了。

以上就是本文的全部內(nèi)容，希望對大家的學(xué)習(xí)有所幫助，也希望大家多多支持腳本之家。

您可能感興趣的文章: