昔念
7de149d946
refactor(socket): 优化TCP数据包处理逻辑并增加调试日志 - 修复 handleTcp 中条件判断的空格格式问题 - 在解码失败时增加详细 Debug 日志输出 - 完善不完整数据包时手动唤醒连接的处理流程 refactor(pet): 重构宠物经验系统与升级逻辑 - 将经验计算函数移至 model 层统一管理 - 优化 AddPetExp 方法逻辑，避免直接修改原字段 - 升级过程中正确扣减经验池并防止溢出 - 抽离 Update 方法用于处理宠物进化和经验更新 refactor(model): 调整 PlayerInfo 结构体引用方式及相关初始化逻辑 - 修改 PlayerEX 中 Data 字段为值类型而非指针 - 更新 NewPlayerInfo 返回值为值类型 - 修正 defaults.Set 调用传参以适配结构体值类型 refactor(service): 统一 UserService 数据传递方式 - 修复 Person 方法返回值为指针类型 - 修复 Save 方法中赋值操作使用解引用方式 ```
82 lines
1.7 KiB
Go
82 lines
1.7 KiB
Go
package codec
|
|
|
|
import (
|
|
"encoding/binary"
|
|
"errors"
|
|
"io"
|
|
|
|
"github.com/panjf2000/gnet/v2"
|
|
)
|
|
|
|
var ErrIncompletePacket = errors.New("incomplete packet")
|
|
|
|
const maxBodyLen = 10 * 1024 * 1024 // 业务最大包体长度，按需调整
|
|
// TomeeSocketCodec 协议格式:
|
|
//
|
|
// * 0 4
|
|
// * +-----------+
|
|
// * | body len |
|
|
// * +-----------+
|
|
// * | |
|
|
// * + +
|
|
// * | body bytes|
|
|
// * + +
|
|
// * | ... ... |
|
|
// * +-----------+
|
|
type TomeeSocketCodec struct{}
|
|
|
|
var _ SocketCodec = (*TomeeSocketCodec)(nil)
|
|
|
|
func NewTomeeSocketCodec() *TomeeSocketCodec {
|
|
return &TomeeSocketCodec{}
|
|
|
|
}
|
|
|
|
func (codec TomeeSocketCodec) Encode(buf []byte) ([]byte, error) {
|
|
bodyLen := len(buf)
|
|
data := make([]byte, 4+bodyLen)
|
|
|
|
// 写入4字节的包长度
|
|
binary.BigEndian.PutUint32(data[:4], uint32(bodyLen))
|
|
// 写入包体
|
|
copy(data[4:], buf)
|
|
|
|
return data, nil
|
|
}
|
|
|
|
func (codec TomeeSocketCodec) Decode(c gnet.Conn) ([]byte, error) {
|
|
|
|
// handle(c)
|
|
// 先读取4字节的包长度
|
|
lenBuf, err := c.Peek(4)
|
|
if err != nil {
|
|
if errors.Is(err, io.ErrShortBuffer) {
|
|
return nil, ErrIncompletePacket
|
|
}
|
|
return nil, err
|
|
}
|
|
|
|
bodyLen := binary.BigEndian.Uint32(lenBuf)
|
|
if bodyLen > maxBodyLen {
|
|
return nil, errors.New("packet body exceeds max length")
|
|
}
|
|
|
|
// 检查整个包是否完整
|
|
buf, err := c.Peek(int(bodyLen))
|
|
if err != nil {
|
|
if errors.Is(err, io.ErrShortBuffer) {
|
|
return nil, ErrIncompletePacket
|
|
}
|
|
return nil, err
|
|
}
|
|
|
|
// 提取包体
|
|
body := make([]byte, bodyLen)
|
|
copy(body, buf)
|
|
|
|
// 从缓冲区中丢弃已读取的数据
|
|
_, _ = c.Discard(4 + int(bodyLen))
|
|
|
|
return body, nil
|
|
}
|