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

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

golang中struct和[]byte的轉(zhuǎn)換方法，其實(shí)就是用到了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)
} 

運(yùn)行結(jié)果：

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

因?yàn)閇]byte底層的數(shù)據(jù)結(jié)構(gòu)為：

struct { 
 addr uintptr 
 len int 
 cap int 
 } 

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

轉(zhuǎn)換的時(shí)候，需要定義一個(gè)和[]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)
} 

運(yùn)行結(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視為指針，所以要使用其地址進(jìn)行轉(zhuǎn)換，由于[]byte底層存放的是指向數(shù)據(jù)的地址。用[]byte的地址就需要使用雙層指針轉(zhuǎn)換，然后再指向其內(nèi)容，得出來的就是轉(zhuǎn)換對應(yīng)struct的指針了。

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