// Package controller 提供游戏逻辑的控制器层,负责处理各种游戏功能的请求分发。 // 包含用户、宠物、战斗、物品、地图、任务等各个模块的控制器方法。 // 通过反射和标签系统自动注册 cmd 处理方法,实现请求到处理函数的映射。 package controller import ( "blazing/cool" "blazing/logic/service/common" "bytes" "context" "fmt" "reflect" "strconv" "strings" "sync" "github.com/gogf/gf/v2/os/glog" "github.com/lunixbochs/struc" ) // Maincontroller 是控制器层共享变量。 var Maincontroller = &Controller{} //注入service // Controller 分发cmd逻辑实现 type Controller struct { UID uint32 RPCClient *struct { Kick func(uint32) error RegisterLogic func(uint32, uint32) error } } // ParseCmd 将字节数组数据解析到指定类型的变量中 // 该函数使用struc库进行数据解包操作 // 参数 data: 需要解析的字节数据 // 返回值: 解析后的指定类型实例 func ParseCmd[T any](data []byte) T { var result T // 使用struc.Unpack将字节数据解包到result变量中 struc.Unpack(bytes.NewReader(data), &result) return result } // Init 初始化控制器,注册所有cmd处理方法 // 参数 isGame: 标识是否为游戏服务器(true)或登录服务器(false) func Init(isGame bool) { // 获取控制器实例的反射值 controllerValue := reflect.ValueOf(Maincontroller) // 获取控制器类型 controllerType := controllerValue.Type() // 遍历控制器的所有方法 for i := 0; i < controllerType.NumMethod(); i++ { method := controllerType.Method(i) methodValue := controllerValue.MethodByName(method.Name) methodType := methodValue.Type() // 获取方法第一个参数的类型(请求结构体) if methodType.NumIn() == 0 { continue } reqArgType := methodType.In(0) if reqArgType.Kind() != reflect.Ptr || reqArgType.Elem().Kind() != reflect.Struct { glog.Warning(context.Background(), "方法首参必须为结构体指针", method.Name, "跳过注册") continue } reqType := reqArgType.Elem() // 解析请求结构体中的cmd标签 for _, cmd := range getCmd(reqType) { if cmd == 0 { // 说明不是有效的注册方法 glog.Warning(context.Background(), "方法参数必须包含CMD参数", method.Name, "跳过注册") continue } // 根据服务器类型过滤cmd // 登录服务器只处理小于1000的cmd if !isGame && cmd > 1000 { continue } // 游戏服务器只处理大于等于1000的cmd if isGame && cmd < 1000 { continue } // 注册命令处理函数 if cool.Config.ServerInfo.IsDebug != 0 { fmt.Println("注册方法", cmd, method.Name) } cmdInfo := cool.Cmd{ Func: methodValue, Req: reqType, // Res: , // TODO 待实现对不同用户初始化方法以取消全局cmdcache } // 预编译创建req实例的函数:返回结构体指针 reqTypeForNew := reqType cmdInfo.NewReqFunc = func() interface{} { return reflect.New(reqTypeForNew).Interface() } if _, exists := cool.CmdCache[cmd]; exists { // 方法已存在 panic(fmt.Sprintf("命令处理方法已存在,跳过注册 %d %s", cmd, method.Name)) } cool.CmdCache[cmd] = cmdInfo } } } var targetType = reflect.TypeOf(common.TomeeHeader{}) var cmdTypeCache sync.Map // 默认返回值(无匹配字段/解析失败时) const defaultCmdValue = 0 func normalizeStructType(typ reflect.Type) reflect.Type { for typ.Kind() == reflect.Ptr { typ = typ.Elem() } return typ } // getCmd 从结构体类型中提取绑定的cmd指令(递归查找嵌套结构体,支持值/指针类型的TomeeHeader) // 参数 typ: 待解析的结构体类型(支持多层指针) // 返回值: 解析到的cmd切片,无匹配/解析失败时返回[defaultCmdValue] func getCmd(typ reflect.Type) []uint32 { typ = normalizeStructType(typ) if cached, ok := cmdTypeCache.Load(typ); ok { return cached.([]uint32) } // 非结构体类型直接返回默认值 if typ.Kind() != reflect.Struct { return []uint32{defaultCmdValue} } if cmd, ok := findCmd(typ, make(map[reflect.Type]struct{})); ok { cmdTypeCache.Store(typ, cmd) return cmd } // 未找到目标字段/所有解析失败,返回默认值 defaultCmd := []uint32{defaultCmdValue} cmdTypeCache.Store(typ, defaultCmd) return defaultCmd } func findCmd(typ reflect.Type, visiting map[reflect.Type]struct{}) ([]uint32, bool) { typ = normalizeStructType(typ) if typ.Kind() != reflect.Struct { return nil, false } if _, seen := visiting[typ]; seen { return nil, false } visiting[typ] = struct{}{} defer delete(visiting, typ) // 遍历结构体字段,查找TomeeHeader字段并解析cmd for i := 0; i < typ.NumField(); i++ { field := typ.Field(i) // 尝试解析当前字段的cmd标签 cmdSlice, isHeader, err := parseCmdTagWithStructField(field) if isHeader && err == nil { // 解析成功,直接返回结果 return cmdSlice, true } // 递归处理嵌套结构体(值/指针类型) nestedTyp := normalizeStructType(field.Type) if nestedTyp.Kind() == reflect.Struct { // 递归查找,找到有效cmd则立即返回 if nestedCmd, ok := findCmd(nestedTyp, visiting); ok { return nestedCmd, true } } } return nil, false } // parseCmdTagWithStructField 校验字段是否为TomeeHeader(值/指针)并解析cmd标签 // 参数 field: 结构体字段元信息 // 返回值: 解析后的cmd切片,是否为目标类型,解析失败错误 func parseCmdTagWithStructField(field reflect.StructField) ([]uint32, bool, error) { // 判断字段类型是否为 TomeeHeader 或 *TomeeHeader if field.Type != targetType && !(field.Type.Kind() == reflect.Ptr && field.Type.Elem() == targetType) { return nil, false, nil } // 提取cmd标签 cmdStr := field.Tag.Get("cmd") if cmdStr == "" { return nil, true, fmt.Errorf("field %s cmd tag is empty", field.Name) } // 高性能解析标签为uint32切片(替代gconv,减少第三方依赖且可控) result := make([]uint32, 0, strings.Count(cmdStr, "|")+1) remain := cmdStr for idx := 0; ; idx++ { part, next, found := strings.Cut(remain, "|") // 去除空白字符(兼容标签中意外的空格) s := strings.TrimSpace(part) if s == "" { return nil, true, fmt.Errorf("field %s cmd tag part %d is empty", field.Name, idx) } // 手动解析uint32,比gconv更可控,避免隐式转换问题 num, err := strconv.ParseUint(s, 10, 32) if err != nil { return nil, true, fmt.Errorf("field %s cmd tag part %d parse error: %v (value: %s)", field.Name, idx, err, s) } result = append(result, uint32(num)) if !found { break } remain = next } return result, true, nil }