bl/logic/controller/controller.go

package controller

import (
	"blazing/common/data/entity"
	"blazing/common/socket/handler"
	"blazing/cool"
	"blazing/logic/service"
	"time"

	"bytes"
	"context"
	"fmt"
	"reflect"

	"github.com/gogf/gf/v2/os/gcmd"
	"github.com/gogf/gf/v2/os/glog"
	"github.com/gogf/gf/v2/util/gconv"
	"github.com/lunixbochs/struc"
	"github.com/panjf2000/gnet/v2"
)

var Maincontroller = NewController() //注入service
// 分发cmd逻辑实现Controller
type Controller struct {
	Port      uint16
	RPCClient struct {
		Kick func(uint32) error

		RegisterLogic func(uint16) error
	}
}

func (h *Controller) KickPerson(a int) error {

	fmt.Println("检测到踢人请求", a)
	service.KickPlayer(uint32(a))
	return nil
}
func (h *Controller) QuitSelf(a int) error {
	//TODO 这里待退出
	fmt.Println("检测到退出请求")

	go func() {

		for {

			//entity.ConutPlayer()
			fmt.Println("当前在线人数", entity.ConutPlayer())
			<-time.After((1000))
		}
	}()
	//service.KickPlayer(uint32(a))
	return nil
}
func NewController() *Controller {
	return &Controller{}
}

func parseCmd[T any](a T, data []byte) T {
	// := info.NewLoginSidInfo()
	struc.Unpack(bytes.NewBuffer(data), &a)
	return a
	//fmt.Println(pinfo)
	//login.OnData_1001(pinfo, player)
	//fmt.Println(data)
}

func init() { //默认初始化扫描
	// 解析命令行参数
	cool.Config.PortBL = gcmd.GetOpt("port", "1").Uint32()
	// 获取对象的反射值和类型
	value := reflect.ValueOf(Maincontroller)

	// 获取类型
	typ := value.Type()

	for i := 0; i < typ.NumMethod(); i++ {
		method := typ.Method(i)

		// TODO 接收者 + 2个参数 一个是类型,一个是player
		//method.Type.In(1) == reflect.TypeOf([]byte{}

		// 获取方法值
		methodValue := value.MethodByName(method.Name)
		methodValue.Type().NumIn()
		var func_cmd uint32

		func_cmd = getcmd(methodValue.Type().In(0))
		if func_cmd == 0 { //说明不是注册方法
			glog.Warning(context.Background(), "方法参数必须是结构体", method.Name, "跳过注册")
			continue
		}
		// if !cmd.Isexist(cmd.CommandID, cmd.EnumCommandID(func_cmd)) {
		// 	fmt.Println("ID方法存在cmd未注册", func_cmd)

		// }

		if cool.Config.PortBL == 0 && func_cmd > 1000 { //判断login服务器
			break

		}

		if cool.Config.PortBL != 0 && func_cmd < 1000 { //判断login服务器
			continue

		}
		glog.Debug(context.Background(), "注册方法", func_cmd, method.Name)
		//	fmt.Println(methodValue.Interface().(func(gnet.Conn, handler.TomeeHeader)))
		_, ok := cool.CmdCache.LoadOrStore(func_cmd, methodValue) //TODO 待实现对不同用户初始化方法以取消全局cmdcache

		if ok { //方法已存在init
			glog.Error(context.Background(), "方法已存在init,不会初始化后面的方法", func_cmd)
		}
	}
}

func getcmd(t reflect.Type) uint32 {
	// 处理指针类型
	if t.Kind() == reflect.Ptr {
		t = t.Elem() // 获取指针指向的类型
	}

	// 确保是结构体
	if t.Kind() != reflect.Struct {

		return 0
	}

	// 遍历结构体字段
	//	fmt.Printf("结构体 %s 的字段信息：\n", t.Name())
	for i := 0; i < t.NumField(); i++ {
		field := t.Field(i)
		//fmt.Printf("- 字段名: %s\n", field.Name)
		//fmt.Printf("  类型: %v\n", field.Type)
		if field.Type == reflect.TypeOf(handler.TomeeHeader{}) {
			//	fmt.Println(reflect.ValueOf(field))

			return gconv.Uint32(field.Tag.Get("cmd"))

		}
		// fmt.Printf("  类型: %v\n", field.Type)
		// fmt.Printf("  标签: %q\n", field.Tag)

		// 递归处理嵌套结构体
		if field.Type.Kind() == reflect.Struct ||
			(field.Type.Kind() == reflect.Ptr && field.Type.Elem().Kind() == reflect.Struct) {
			fmt.Println("  嵌套结构体字段:")
			nestedType := field.Type
			if nestedType.Kind() == reflect.Ptr {
				nestedType = nestedType.Elem()
			}
			getcmd(nestedType)
		}
		//	fmt.Println()
	}
	return 0
}

// 遍历结构体方法并执行RECV_cmd
func Recv(c gnet.Conn, data handler.TomeeHeader) {

	tt, ok := cool.CmdCache.Load(data.CMD)
	if !ok {

		glog.Error(context.Background(), data.CMD, "cmd未注册")
		return //TODO 待实现cmd未注册
	}
	cmdlister := tt.(reflect.Value)
	//	fmt.Println(cmdlister)

	params := []reflect.Value{}

	//funct := cmdlister.Type().NumIn()

	// 如果需要可设置的变量（用于修改值），创建指针并解引用
	ptrValue := reflect.New(cmdlister.Type().In(0))

	tt1 := ptrValue.Elem().Addr().Interface()
	//	fmt.Println(tt1)
	err := struc.Unpack(bytes.NewBuffer(data.Data), tt1)
	if err != nil {
		fmt.Println(err)
	}
	//fmt.Println(tt1)
	ptrValue1 := ptrValue.Elem()
	// 设置 Name 字段
	nameField := ptrValue.Elem().Field(0) //首个为header
	if nameField.IsValid() && nameField.CanSet() {
		nameField.Set(reflect.ValueOf(data))
	}
	if cmdlister.Type().In(1) == reflect.TypeOf(entity.Player{}) {

		params = append(params, ptrValue1, reflect.ValueOf(service.GetPlayer(c, data.UserID)))
	} else {

		params = append(params, ptrValue1, reflect.ValueOf(c))
	}

	cmdlister.Call(params)

	r := entity.Conn{MainConn: c}
	// // 遍历结构体字段
	// //	fmt.Printf("结构体 %s 的字段信息：\n", t.Name())
	// for i := 0; i < ptrValue1.Type().NumField(); i++ {
	// 	field := ptrValue1.Type().Field(i)

	// 	if field.Name == "Ret" { //判断Ret

	// 	}

	// }
	t, ok := ptrValue1.Interface().(Blazingservice)

	if ok && t.Ret() != nil {
		r.SendPack(t.Ret())
		t.Ret()
	}
	//r.SendPack(ret[0].Interface().([]byte))

}

type Blazingservice interface {
	Ret() []byte
}