refactor(common): 重构 Pack 函数并优化数据处理
- 重构了 Pack 函数,支持处理不同类型的 data 参数 - 优化了数据打包过程,使用反射处理结构体和切片 - 修改了 TomeeHeader 结构体,使用 EnumCommandID 类型 - 调整了 login 相关的结构体定义 - 优化了 main 函数中的数据处理逻辑
This commit is contained in:
@@ -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()
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user