昔念
741ef6ebd4
- 引入 idgenerator-go 库，实现全局唯一 ID 生成 - 重构 Pack 函数，使用接口参数提高灵活性 - 修改 Player 结构，增加 MainConn 字段用于主连接 - 更新 SocketHandler_Tomee 中的 Data 字段标记 - 优化 Recv 函数中的数据解包和参数处理逻辑
66 lines
1.3 KiB
Go
66 lines
1.3 KiB
Go
package core
|
|
|
|
import (
|
|
"blazing/common/serialize/bytearray"
|
|
"blazing/common/socket/handler"
|
|
"bytes"
|
|
"fmt"
|
|
"reflect"
|
|
|
|
"github.com/lunixbochs/struc"
|
|
)
|
|
|
|
func Pack(header1 handler.TomeeHeader, data any) []byte { //组包
|
|
header := header1
|
|
//t := reflect.TypeOf(data)
|
|
tv := reflect.ValueOf(data)
|
|
var datar []byte
|
|
|
|
// 处理指针类型
|
|
if tv.Kind() == reflect.Ptr {
|
|
//tv = t.Elem() // 获取指针指向的类型
|
|
tv = tv.Elem() // 获取指针指向的值
|
|
}
|
|
//.Println(t.Kind())
|
|
//t1 := tv.Interface()
|
|
//fmt.Println(t1)
|
|
switch tv.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()
|
|
|
|
default:
|
|
datar = []byte{}
|
|
// fmt.Println(err, datar)
|
|
// p.Conn.Write(p.pack(cmd, data))
|
|
|
|
}
|
|
header.Len = uint32(len(datar) + 17)
|
|
|
|
// if iserror != 0 { //如果存在错误码
|
|
// head1.Data = nil
|
|
// }
|
|
//var data1 bytes.Buffer
|
|
//.Pack(&data1, &head1)
|
|
by := bytearray.CreateByteArray()
|
|
by.WriteUInt32(header.Len)
|
|
by.WriteString(header.Version)
|
|
by.WriteUInt32(uint32(header.CMD))
|
|
by.WriteUInt32(header.UserID)
|
|
by.WriteUInt32(header.Result)
|
|
by.Write(datar)
|
|
|
|
return by.Bytes()
|
|
|
|
}
|