golang 仿python pack/unpack

写得不完善也不完美 尤其是高低位转换那（go和c 二进制高地位相反 需要转换，还有go int转[]byte长度是4位），希望牛人看后指导一下

项目需要通过socket调取 客户端是go ，服务器端是python，由于需要封包解包，就参照python写的

python 的pack/unpack 参考  Python使用struct处理二进制(pack和unpack用法)

package includes

import (
    "bytes"
    "encoding/binary"
    "fmt"
    _ "os"
    "strconv"
    "strings"
)

type Protocol struct {
    Format []string
}

//封包
func (p *Protocol) Pack(args ...interface{}) []byte {
    la := len(args)
    ls := len(p.Format)
    ret := []byte{}
    if ls >  && la >  && ls == la {
        for i := ; i < ls; i++ {
            if p.Format[i] == "H" {
                ret = append(ret, IntToBytes2(args[i].(int))...)
            } else if p.Format[i] == "I" {
                ret = append(ret, IntToBytes4(args[i].(int))...)
            } else if strings.Contains(p.Format[i], "s") {
                num, _ := strconv.Atoi(strings.TrimRight(p.Format[i], "s"))
                ret = append(ret, []byte(fmt.Sprintf("%s%s", args[i].(string), strings.Repeat("\x00", num-len(args[i].(string)))))...)
            }
        }
    }
    return ret
}

//解包
func (p *Protocol) UnPack(msg []byte) []interface{} {
    la := len(p.Format)
    ret := make([]interface{}, la)
    if la >  {
        for i := ; i < la; i++ {
            if p.Format[i] == "H" {
                ret[i] = Bytes4ToInt(msg[:])
                msg = msg[:len(msg)]
            } else if p.Format[i] == "I" {
                ret[i] = Bytes4ToInt(msg[:])
                msg = msg[:len(msg)]
            } else if strings.Contains(p.Format[i], "s") {
                num, _ := strconv.Atoi(strings.TrimRight(p.Format[i], "s"))
                ret[i] = string(msg[:num])
                msg = msg[num:len(msg)]

            }
        }
    }
    return ret
}

func (p *Protocol) Size() int {
    size :=
    ls := len(p.Format)
    if ls >  {
        for i := ; i < ls; i++ {
            if p.Format[i] == "H" {
                size = size +
            } else if p.Format[i] == "I" {
                size = size +
            } else if strings.Contains(p.Format[i], "s") {
                num, _ := strconv.Atoi(strings.TrimRight(p.Format[i], "s"))
                size = size + num
            }
        }
    }
    return size
}

//整形转换成字节
func IntToBytes(n int) []byte {
    m := int32(n)
    bytesBuffer := bytes.NewBuffer([]byte{})
    binary.Write(bytesBuffer, binary.BigEndian, m)

    gbyte := bytesBuffer.Bytes()

    return gbyte
}

//整形转换成字节4位
func IntToBytes4(n int) []byte {
    m := int32(n)
    bytesBuffer := bytes.NewBuffer([]byte{})
    binary.Write(bytesBuffer, binary.BigEndian, m)

    gbyte := bytesBuffer.Bytes()
    //c++ 高低位转换
    k :=
    x := len(gbyte)
    nb := make([]byte, k)
    for i := ; i < k; i++ {
        nb[i] = gbyte[x-i-]
    }
    return nb
}

//整形转换成字节2位
func IntToBytes2(n int) []byte {
    m := int32(n)
    bytesBuffer := bytes.NewBuffer([]byte{})
    binary.Write(bytesBuffer, binary.BigEndian, m)

    gbyte := bytesBuffer.Bytes()
    //c++ 高低位转换
    k :=
    x := len(gbyte)
    nb := make([]byte, k)
    for i := ; i < k; i++ {
        nb[i] = gbyte[x-i-]
    }
    return nb
}

//字节转换成整形
func BytesToInt(b []byte) int {
    bytesBuffer := bytes.NewBuffer(b)

    var x int32
    binary.Read(bytesBuffer, binary.BigEndian, &x)

    return int(x)
}

//4个字节转换成整形
func Bytes4ToInt(b []byte) int {
    xx := make([]byte, )
    if len(b) ==  {
        xx = []byte{b[], b[], , }
    } else {
        xx = b
    }

    m := len(xx)
    nb := make([]byte, )
    for i := ; i < ; i++ {
        nb[i] = xx[m-i-]
    }
    bytesBuffer := bytes.NewBuffer(nb)

    var x int32
    binary.Read(bytesBuffer, binary.BigEndian, &x)

    return int(x)
}

调用

p := new(Protocol)
p.Format = []string{"H", "H", "I", "16s", "I", "I", "I"}
h_byte := p.Pack(, , , "abc", , , len(request_buf))
conn.Write(h_byte)

附加：

int 转 二进制 fmt.Printf("%08b\n", ) //

二进制转int i, _ := strconv.ParseInt("", , ) //

int 转 []byte IntToBytes() // [0 0 3 232]  3*256+232=1000

func IntToBytes(n int) []byte {
m := int32(n)
bytesBuffer := bytes.NewBuffer([]byte{})
binary.Write(bytesBuffer, binary.BigEndian, m)

gbyte := bytesBuffer.Bytes()

return gbyte
}

[]byte 转int BytesToInt([]byte{,,,}) //

func BytesToInt(b []byte) int {
bytesBuffer := bytes.NewBuffer(b)

var x int32
binary.Read(bytesBuffer, binary.BigEndian, &x)

return int(x)
}

int 转 byte byte() //