昔念
e25fe776eb
- 重构了 Pack 函数，支持处理不同类型的 data 参数 - 优化了数据打包过程，使用反射处理结构体和切片 - 修改了 TomeeHeader 结构体，使用 EnumCommandID 类型 - 调整了 login 相关的结构体定义 - 优化了 main 函数中的数据处理逻辑
72 lines
1.5 KiB
Go
72 lines
1.5 KiB
Go
package core
|
|
|
|
import (
|
|
"blazing/common/serialize/bytearray"
|
|
"blazing/common/socket/cmd"
|
|
"blazing/common/socket/handler"
|
|
"bytes"
|
|
"fmt"
|
|
"reflect"
|
|
|
|
"github.com/lunixbochs/struc"
|
|
)
|
|
|
|
func Pack(userid uint32, cmd cmd.EnumCommandID, data any, iserror uint32) []byte { //组包
|
|
|
|
t := reflect.TypeOf(data)
|
|
tv := reflect.ValueOf(data)
|
|
var datar []byte
|
|
|
|
// 处理指针类型
|
|
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))) //写入数据
|
|
|
|
case reflect.Struct:
|
|
var data1 bytes.Buffer
|
|
err := struc.Pack(&data1, data)
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
}
|
|
|
|
datar = data1.Bytes()
|
|
// fmt.Println(err, datar)
|
|
// p.Conn.Write(p.pack(cmd, data))
|
|
|
|
}
|
|
|
|
head1 := handler.TomeeHeader{
|
|
|
|
Len: uint32(len(datar) + 17),
|
|
CMD: cmd,
|
|
Version: "7",
|
|
UserID: userid,
|
|
Result: iserror,
|
|
Data: datar,
|
|
}
|
|
|
|
if iserror != 0 { //如果存在错误码
|
|
head1.Data = nil
|
|
}
|
|
//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()
|
|
}
|