72 lines
1.7 KiB
Go
72 lines
1.7 KiB
Go
package znet
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/binary"
|
|
"errors"
|
|
"go-study/zinx/utils"
|
|
"go-study/zinx/ziface"
|
|
)
|
|
|
|
type DataPack struct{}
|
|
|
|
func NewDataPack() *DataPack {
|
|
return &DataPack{}
|
|
}
|
|
|
|
// GetHeadLen 获取包头长度方法
|
|
func (dp *DataPack) GetHeadLen() uint32 {
|
|
// ID uint32(4字节) + DataLen uint32(4字节)
|
|
return 8
|
|
}
|
|
|
|
// Pack 封包方法（压缩数据）
|
|
func (dp *DataPack) Pack(msg ziface.IMessage) ([]byte, error) {
|
|
// 创建一个存放二进制数据的缓冲
|
|
dataBuff := bytes.NewBuffer([]byte{})
|
|
|
|
// 写 ID
|
|
if err := binary.Write(dataBuff, binary.LittleEndian, msg.GetMsgID()); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// 写 DataLen
|
|
if err := binary.Write(dataBuff, binary.LittleEndian, msg.GetDataLen()); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// 写 Data
|
|
if err := binary.Write(dataBuff, binary.LittleEndian, msg.GetData()); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return dataBuff.Bytes(), nil
|
|
}
|
|
|
|
// Unpack 拆包方法（解压数据）
|
|
func (dp *DataPack) Unpack(binaryData []byte) (ziface.IMessage, error) {
|
|
// 创建一个从输入二进制数据的 ioReader
|
|
dataBuff := bytes.NewReader(binaryData)
|
|
|
|
// 只解压 head 信息，得到 ID 和 DataLen
|
|
msg := &Message{}
|
|
|
|
// 读 ID
|
|
if err := binary.Read(dataBuff, binary.LittleEndian, &msg.id); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// 读 DataLen
|
|
if err := binary.Read(dataBuff, binary.LittleEndian, &msg.dataLen); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// 判断 DataLen 是否超出我们允许的最大包长度
|
|
if utils.ConfigInstance.MaxPacketSize > 0 && msg.dataLen > utils.ConfigInstance.MaxPacketSize {
|
|
return nil, errors.New("too large msg data received")
|
|
}
|
|
|
|
// 这里只需把 head 信息拆包出来即可，然后通过 head 中的长度再从 conn 读取一次数据
|
|
return msg, nil
|
|
}
|