//isSymbol表示有無符號
func BytesToInt(b []byte, isSymbol bool) (int, error){
if isSymbol {
return bytesToIntS(b)
}
return bytesToIntU(b)
}
//字節(jié)數(shù)(大端)組轉(zhuǎn)成int(無符號的)
func bytesToIntU(b []byte) (int, error) {
if len(b) == 3 {
b = append([]byte{0},b...)
}
bytesBuffer := bytes.NewBuffer(b)
switch len(b) {
case 1:
var tmp uint8
err := binary.Read(bytesBuffer, binary.BigEndian, tmp)
return int(tmp), err
case 2:
var tmp uint16
err := binary.Read(bytesBuffer, binary.BigEndian, tmp)
return int(tmp), err
case 4:
var tmp uint32
err := binary.Read(bytesBuffer, binary.BigEndian, tmp)
return int(tmp), err
default:
return 0,fmt.Errorf("%s", "BytesToInt bytes lenth is invaild!")
}
}
//字節(jié)數(shù)(大端)組轉(zhuǎn)成int(有符號)
func bytesToIntS(b []byte) (int, error) {
if len(b) == 3 {
b = append([]byte{0},b...)
}
bytesBuffer := bytes.NewBuffer(b)
switch len(b) {
case 1:
var tmp int8
err := binary.Read(bytesBuffer, binary.BigEndian, tmp)
return int(tmp), err
case 2:
var tmp int16
err := binary.Read(bytesBuffer, binary.BigEndian, tmp)
return int(tmp), err
case 4:
var tmp int32
err := binary.Read(bytesBuffer, binary.BigEndian, tmp)
return int(tmp), err
default:
return 0,fmt.Errorf("%s", "BytesToInt bytes lenth is invaild!")
}
}
//整形轉(zhuǎn)換成字節(jié)
func IntToBytes(n int,b byte) ([]byte,error) {
switch b {
case 1:
tmp := int8(n)
bytesBuffer := bytes.NewBuffer([]byte{})
binary.Write(bytesBuffer, binary.BigEndian, tmp)
return bytesBuffer.Bytes(),nil
case 2:
tmp := int16(n)
bytesBuffer := bytes.NewBuffer([]byte{})
binary.Write(bytesBuffer, binary.BigEndian, tmp)
return bytesBuffer.Bytes(),nil
case 3,4:
tmp := int32(n)
bytesBuffer := bytes.NewBuffer([]byte{})
binary.Write(bytesBuffer, binary.BigEndian, tmp)
return bytesBuffer.Bytes(),nil
}
return nil,fmt.Errorf("IntToBytes b param is invaild")
}
在tcp協(xié)議傳輸中,為了防止粘包,需要先發(fā)送消息頭,即先發(fā)送數(shù)據(jù)長度,在根據(jù)長度write真正的數(shù)據(jù),然e,由于網(wǎng)絡(luò)傳輸都是字節(jié)流,怎么將整型轉(zhuǎn)為字節(jié)流呢?
func IntToBytes(n int)[]byte{
data:=int64(n)//數(shù)據(jù)類型轉(zhuǎn)換
bytebuffer:=bytes.NewBuffer([]byte{})//字節(jié)集合
binary.Write(bytebuffer,binary.BigEndian,data)//按照二進(jìn)制寫入字節(jié)
return bytebuffer.Bytes()//返回字節(jié)結(jié)合
}
func BytesToInt(bs []byte)int{
bytebuffer:=bytes.NewBuffer(bs) //根據(jù)二進(jìn)制寫入二進(jìn)制結(jié)合
var data int64
binary.Read(bytebuffer,binary.BigEndian,data) //解碼
return int(data)
}