refactor(common): 重构 Pack 函数并优化数据处理

- 重构了 Pack 函数,支持处理不同类型的 data 参数
- 优化了数据打包过程,使用反射处理结构体和切片
- 修改了 TomeeHeader 结构体,使用 EnumCommandID 类型
- 调整了 login 相关的结构体定义
- 优化了 main 函数中的数据处理逻辑
This commit is contained in:
2025-06-27 00:30:15 +08:00
parent feabc7153a
commit e25fe776eb
7 changed files with 76 additions and 90 deletions

View File

@@ -1,67 +1,71 @@
package core
import (
"blazing/common/serialize/bytearray"
"blazing/common/socket/cmd"
"blazing/common/socket/handler"
"bytes"
"fmt"
"reflect"
"github.com/lunixbochs/struc"
"github.com/panjf2000/gnet"
)
func SendError(cmd cmd.EnumCommandID, data any) { //TODO发送错误码的实现
func Pack(userid uint32, cmd cmd.EnumCommandID, data any, iserror uint32) []byte { //组包
//序列化结构体取错误码,然后构造错误结构
t := reflect.TypeOf(data)
tv := reflect.ValueOf(data)
var datar []byte
//SEND ERROR 方法通过error tag取到标签然后error库内取参数
// 处理指针类型
if t.Kind() == reflect.Ptr {
t = t.Elem() // 获取指针指向的类型
tv = tv.Elem() // 获取指针指向的值
}
//.Println(t.Kind())
//t1 := tv.Interface()
//fmt.Println(t1)
switch t.Kind() {
}
case reflect.Slice:
datar = data.([]byte)
//p.Conn.Write(p.pack(cmd, data.([]byte))) //写入数据
func SendPack(c gnet.Conn, cmd cmd.EnumCommandID, data any) { //TODO 待测试
case reflect.Struct:
var data1 bytes.Buffer
err := struc.Pack(&data1, data)
if err != nil {
fmt.Println(err)
}
//t := reflect.TypeOf(data)
// var datar []byte
// var iserror uint32 = 0
// switch t.Kind() {
// case reflect.Slice:
// datar = data.([]byte)
// //p.Conn.Write(p.pack(cmd, data.([]byte))) //写入数据
datar = data1.Bytes()
// fmt.Println(err, datar)
// p.Conn.Write(p.pack(cmd, data))
// case reflect.Struct:
// var data1 bytes.Buffer
// struc.Pack(&data1, &data)
// datar = data1.Bytes()
// // p.Conn.Write(p.pack(cmd, data))
// case reflect.Uint32: //说明是错误码
// iserror = data.(uint32)
// default:
// // 可选:处理 nil 数据情况,直接返回
// datar = nil
// return
// }
//c.Write(pack(packhead(userid uint32, cmd cmd.EnumCommandID, data []byte, iserror uint32)))
}
func Pack(userid uint32, cmd cmd.EnumCommandID, data []byte, iserror uint32) []byte { //组包
}
head1 := handler.TomeeHeader{
Len: uint32(len(data) + 17),
CMD: uint32(cmd),
Len: uint32(len(datar) + 17),
CMD: cmd,
Version: "7",
UserID: userid,
Result: iserror,
Data: data,
Data: datar,
}
if iserror != 0 { //如果存在错误码
head1.Data = nil
}
var data1 bytes.Buffer
struc.Pack(&data1, &head1)
return data1.Bytes()
//var data1 bytes.Buffer
//.Pack(&data1, &head1)
by := bytearray.CreateByteArray()
by.WriteUInt32(head1.Len)
by.WriteString(head1.Version)
by.WriteUInt32(uint32(head1.CMD))
by.WriteUInt32(head1.UserID)
by.WriteUInt32(head1.Result)
by.Write(datar)
return by.Bytes()
}