51 lines
1.2 KiB
Go
51 lines
1.2 KiB
Go
|
// socket_stick/proto/proto.go
|
|||
|
|
package proto
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"bufio"
|
|||
|
|
"bytes"
|
|||
|
|
"encoding/binary"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// Encode 将消息编码
|
|||
|
|
func Encode(message string) ([]byte, error) {
|
|||
|
|
// 读取消息的长度，转换成int32类型（占4个字节）
|
|||
|
|
var length = int32(len(message))
|
|||
|
|
var pkg = new(bytes.Buffer)
|
|||
|
|
// 写入消息头
|
|||
|
|
err := binary.Write(pkg, binary.LittleEndian, length)
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, err
|
|||
|
|
}
|
|||
|
|
// 写入消息实体
|
|||
|
|
err = binary.Write(pkg, binary.LittleEndian, []byte(message))
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, err
|
|||
|
|
}
|
|||
|
|
return pkg.Bytes(), nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// Decode 解码消息
|
|||
|
|
func Decode(reader *bufio.Reader) (string, error) {
|
|||
|
|
// 读取消息的长度
|
|||
|
|
lengthByte, _ := reader.Peek(4) // 读取前4个字节的数据
|
|||
|
|
lengthBuff := bytes.NewBuffer(lengthByte)
|
|||
|
|
var length int32
|
|||
|
|
err := binary.Read(lengthBuff, binary.LittleEndian, &length)
|
|||
|
|
if err != nil {
|
|||
|
|
return "", err
|
|||
|
|
}
|
|||
|
|
// Buffered返回缓冲中现有的可读取的字节数。
|
|||
|
|
if int32(reader.Buffered()) < length+4 {
|
|||
|
|
return "", err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 读取真正的消息数据
|
|||
|
|
pack := make([]byte, int(4+length))
|
|||
|
|
_, err = reader.Read(pack)
|
|||
|
|
if err != nil {
|
|||
|
|
return "", err
|
|||
|
|
}
|
|||
|
|
return string(pack[4:]), nil
|
|||
|
|
}
|