昔念
ea4ca98e49
- 修复 `OnOpen` 中网络类型判断位置不正确的问题，提前过滤非 TCP 连接 - 移除 `OnTraffic` 中重复的网络类型判断 - 优化 `TomeeSocketCodec` 的解码逻辑，使用 `InboundBuffered` 和 `Next` 提高效率 - 调整 `ByteArray` 创建方法参数，避免可变参数带来的性能损耗 - 在 `ClientData` 中将 `IsCrossDomain` 改为 `sync.Once` 避免重复处理 - 使用 `AsyncWrite` 替代 `Write` 提升写入异步性 - 修复连接关闭流程，使用
275 lines
6.2 KiB
Go
275 lines
6.2 KiB
Go
package player
|
||
|
||
import (
|
||
"blazing/common/socket/errorcode"
|
||
"blazing/common/utils/bytearray"
|
||
"blazing/cool"
|
||
"sync"
|
||
|
||
"context"
|
||
|
||
"bytes"
|
||
"fmt"
|
||
"reflect"
|
||
|
||
"github.com/gobwas/ws"
|
||
"github.com/gobwas/ws/wsutil"
|
||
"github.com/gogf/gf/v2/os/glog"
|
||
"github.com/lunixbochs/struc"
|
||
"github.com/panjf2000/gnet/v2"
|
||
"github.com/panjf2000/gnet/v2/pkg/logging"
|
||
)
|
||
|
||
// TomeeHeader 结构体字段定义
|
||
type TomeeHeader struct {
|
||
Len uint32 `json:"len"`
|
||
Version byte `json:"version" struc:"[1]byte"`
|
||
CMD uint32 `json:"cmdId" struc:"uint32"`
|
||
UserID uint32 `json:"userId"`
|
||
//Error uint32 `json:"error" struc:"[0]pad"`
|
||
|
||
Result uint32 `json:"result"`
|
||
Data []byte `json:"data" struc:"skip"` //组包忽略此字段// struc:"[0]pad"
|
||
//Return []byte `struc:"[0]pad"` //返回记录
|
||
}
|
||
|
||
func NewTomeeHeader(cmd uint32, userid uint32) *TomeeHeader {
|
||
|
||
return &TomeeHeader{
|
||
CMD: cmd,
|
||
// Len: 0,
|
||
Version: 49,
|
||
Result: 0,
|
||
}
|
||
|
||
}
|
||
|
||
// Pack 将给定的数据打包成一个字节切片。
|
||
// 该方法处理的数据类型包括指针、切片和结构体。
|
||
// 对于指针类型，会解引用以获取实际值。
|
||
// 切片类型直接转换为字节切片。
|
||
// 结构体类型使用struc库进行序列化。
|
||
// 最后，将数据长度、版本号、命令码、用户ID和结果代码一并打包进返回的字节切片中。
|
||
func (h *TomeeHeader) Pack(data any) []byte { //组包
|
||
//h.Result = 0//默认置0
|
||
|
||
//t := reflect.TypeOf(data)
|
||
tv := reflect.ValueOf(data)
|
||
var datar []byte
|
||
|
||
// 处理指针类型
|
||
if tv.Kind() == reflect.Ptr {
|
||
//tv = t.Elem() // 获取指针指向的类型
|
||
tv = tv.Elem() // 获取指针指向的值
|
||
}
|
||
|
||
switch tv.Kind() {
|
||
case reflect.String:
|
||
datar = []byte(tv.String())
|
||
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))
|
||
|
||
}
|
||
h.Len = uint32(len(datar) + 17)
|
||
|
||
by := bytearray.CreateByteArray(nil)
|
||
by.WriteUInt32(h.Len)
|
||
by.WriteByte(h.Version)
|
||
by.WriteUInt32(uint32(h.CMD))
|
||
by.WriteUInt32(h.UserID)
|
||
by.WriteUInt32(h.Result)
|
||
by.Write(datar)
|
||
|
||
return by.Bytes()
|
||
|
||
}
|
||
|
||
// var _ Blazingservice = (*TomeeHeader)(nil)
|
||
|
||
// type Blazingservice interface {
|
||
// Ret() []byte
|
||
// }
|
||
|
||
// MergeBytes 将多个字节数组合并为一个
|
||
func MergeBytes(arrays ...[]byte) []byte {
|
||
// 计算所有数组的总长度
|
||
totalLen := 0
|
||
for _, arr := range arrays {
|
||
totalLen += len(arr)
|
||
}
|
||
|
||
// 创建结果切片
|
||
result := make([]byte, totalLen)
|
||
|
||
// 逐个复制数组内容
|
||
currentIndex := 0
|
||
for _, arr := range arrays {
|
||
copy(result[currentIndex:], arr)
|
||
currentIndex += len(arr)
|
||
}
|
||
|
||
return result
|
||
}
|
||
|
||
// 遍历结构体方法并执行RECV_cmd
|
||
func (h *ClientData) Recv(data TomeeHeader) {
|
||
|
||
cmdlister, ok := cool.CmdCache.Load(data.CMD)
|
||
if !ok {
|
||
|
||
glog.Debug(context.Background(), data.CMD, "cmd未注册")
|
||
return //TODO 待实现cmd未注册
|
||
}
|
||
|
||
params := []reflect.Value{}
|
||
|
||
//funct := cmdlister.Type().NumIn()
|
||
|
||
// 如果需要可设置的变量（用于修改值），创建指针并解引用
|
||
ptrValue := reflect.New(cmdlister.Type().In(0).Elem())
|
||
|
||
tt1 := ptrValue.Elem().Addr().Interface()
|
||
// fmt.Println(tt1)
|
||
err := struc.Unpack(bytes.NewBuffer(data.Data), tt1)
|
||
if err != nil {
|
||
cool.Loger.Error(context.Background(), data.UserID, data.CMD, "解包失败")
|
||
return
|
||
}
|
||
//fmt.Println(tt1)
|
||
ptrValue1 := ptrValue.Elem().Addr()
|
||
// 设置 Name 字段
|
||
nameField := ptrValue.Elem().Field(0) //首个为header
|
||
if nameField.IsValid() && nameField.CanSet() {
|
||
nameField.Set(reflect.ValueOf(data))
|
||
}
|
||
|
||
if data.CMD > 1001 { //if cmdlister.Type().In(1) == reflect.TypeOf(&Player{}) {
|
||
//t := GetPlayer(c, data.UserID)
|
||
|
||
// fmt.Println(data.CMD, "接收 变量的地址 ", &t.Info, t.Info.UserID)
|
||
|
||
params = append(params, ptrValue1, reflect.ValueOf(h.Conn.Context().(*ClientData).Player))
|
||
} else {
|
||
|
||
params = append(params, ptrValue1, reflect.ValueOf(h.Conn))
|
||
}
|
||
|
||
ret := cmdlister.Call(params)
|
||
|
||
if len(ret) <= 0 { //如果判断没有参数,那就说明这个包没有返回参数
|
||
return
|
||
}
|
||
|
||
aa, ok := ret[1].Interface().(errorcode.ErrorCode) //判断错误
|
||
data.Result = uint32(aa)
|
||
if aa == -1 {
|
||
return
|
||
|
||
}
|
||
t := h.Conn.Context().(*ClientData)
|
||
|
||
if ok && aa != 0 { //这里实现回复错误包
|
||
|
||
cool.Loger.Error(context.Background(), aa.Code())
|
||
t.SendPack(data.Pack(nil))
|
||
|
||
return
|
||
|
||
}
|
||
|
||
//data.Version = 49
|
||
t.SendPack(data.Pack(ret[0].Interface()))
|
||
|
||
}
|
||
|
||
type ClientData struct {
|
||
IsCrossDomain sync.Once //是否跨域过
|
||
Player *Player //客户实体
|
||
//Mu sync.RWMutex
|
||
ERROR_CONNUT int
|
||
Wsmsg *WsCodec
|
||
Conn gnet.Conn
|
||
SaveL sync.Once //保存锁
|
||
|
||
}
|
||
|
||
func NewClientData(c gnet.Conn) *ClientData {
|
||
// 创建事件处理器
|
||
|
||
cd := ClientData{
|
||
|
||
Player: nil,
|
||
Conn: c,
|
||
Wsmsg: &WsCodec{},
|
||
}
|
||
return &cd
|
||
|
||
}
|
||
|
||
func (h *ClientData) OnEvent(v []byte) {
|
||
|
||
header := TomeeHeader{}
|
||
|
||
tempdata := bytearray.CreateByteArray(v)
|
||
header.Len, _ = tempdata.ReadUInt32()
|
||
header.Version, _ = tempdata.ReadByte()
|
||
header.CMD, _ = tempdata.ReadUInt32()
|
||
//header.CMD = cmd.EnumCommandID(_CMD)
|
||
header.UserID, _ = tempdata.ReadUInt32()
|
||
|
||
header.Result, _ = tempdata.ReadUInt32()
|
||
header.Data = tempdata.BytesAvailable()
|
||
if header.CMD > 1001 {
|
||
if h.Conn.Context().(*ClientData).Player == nil {
|
||
cool.Loger.Error(context.TODO(), header.UserID, "账号未注册")
|
||
return
|
||
}
|
||
if h.Conn.Context().(*ClientData).Player.Info == nil {
|
||
cool.Loger.Error(context.TODO(), header.UserID, "未创建角色")
|
||
return
|
||
}
|
||
}
|
||
cool.Loger.Debug(context.TODO(), "接收数据", header.UserID, header.CMD)
|
||
h.Recv(header)
|
||
|
||
}
|
||
func (p *ClientData) SendPack(b []byte) error {
|
||
if _, ok := p.Conn.Context().(*ClientData); !ok {
|
||
return fmt.Errorf("链接错误,取消发包")
|
||
|
||
}
|
||
|
||
if p.Conn.Context().(*ClientData).Wsmsg.Upgraded {
|
||
// This is the echo server
|
||
err := wsutil.WriteServerMessage(p.Conn, ws.OpBinary, b)
|
||
if err != nil {
|
||
logging.Infof("conn[%v] [err=%v]", p.Conn.RemoteAddr().String(), err.Error())
|
||
return err
|
||
}
|
||
} else {
|
||
|
||
err := p.Conn.AsyncWrite(b, nil)
|
||
if err != nil {
|
||
glog.Debug(context.Background(), err)
|
||
|
||
}
|
||
}
|
||
|
||
return nil
|
||
}
|