1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
|
package znet
import (
"bytes"
"encoding/binary"
"errors"
"zinx/utils"
"zinx/ziface"
)
type DataPack struct{}
func NewDataPack() *DataPack {
return &DataPack{}
}
func (dp *DataPack) GetDataLen() uint32 {
//DataLen uint32:4bit, ID uint32 4bit
return 8
}
func (dp *DataPack) Pack(msg ziface.IMessage) ([]byte, error) {
//创建一个缓存
dataBuf := bytes.NewBuffer([]byte{})
//将dataLen 写入databuf
if err := binary.Write(dataBuf, binary.LittleEndian, msg.GetMsgLen()); err != nil {
return nil, err
}
//将MsgId 写入databuf
if err := binary.Write(dataBuf, binary.LittleEndian, msg.GetMsgId()); err != nil {
return nil, err
}
//将data 写入databuf
if err := binary.Write(dataBuf, binary.LittleEndian, msg.GetData()); err != nil {
return nil, err
}
return dataBuf.Bytes(), nil
}
func (dp *DataPack) Unpack(data []byte) (ziface.IMessage, error) {
//创建一个缓存
dataBuf := bytes.NewBuffer(data)
//只解压head信息，得到dataLen和msgId
msg := &Message{}
//读dataLen
if err := binary.Read(dataBuf, binary.LittleEndian, &msg.DataLen); err != nil {
return nil, err
}
//读msgId
if err := binary.Read(dataBuf, binary.LittleEndian, &msg.Id); err != nil {
return nil, err
}
//判断datalen是否已经超出我们允许的最大包长度
if utils.GlobalObject.MaxPackageSize > 0 && msg.DataLen > utils.GlobalObject.MaxPackageSize {
return nil, errors.New("Too large msg data recv")
}
return msg, nil
}
|