bl/logic/controller/Controller.go

// 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
}