Files
bl/logic/controller/controller.go
昔念 bf72b91fc6 refactor(rpc): 重构 RPC 客户端并添加重连机制
- 更新了 RPC 客户端的初始化和重连逻辑
- 添加了重连函数和最大重试次数的配置
- 优化了与服务器的连接管理
- 调整了端口相关的数据类型
2025-07-17 05:20:30 +08:00

244 lines
6.1 KiB
Go

package controller
import (
"blazing/common/data/entity"
"blazing/common/socket/handler"
"blazing/cool"
"blazing/logic/service"
"os"
"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, 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())
if entity.ConutPlayer() <= 0 {
//执行退出逻辑
os.Exit(1)
}
<-time.After((50000))
}
}()
//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").Uint16()
// 获取对象的反射值和类型
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).Elem())
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().Addr()
// 设置 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
// }
// }
// if cmdlister.Type().In(0).Implements(reflect.TypeOf((*Blazingservice)(nil)).Elem()) {
// fmt.Printf("参数 %d 是结构体且实现了 MyInterface 接口\n" )
// tt1
// } else {
// fmt.Printf("参数 %d 是结构体但未实现 MyInterface 接口\n", i)
// }
aa := ptrValue1.Elem().Field(0).Addr().Interface()
t := aa.(Blazingservice)
//st, ok := aa.(Blazingservice)
//fmt.Println(t.Ret())
if t.Ret() != nil {
err := r.SendPack(t.Ret())
if err != nil {
fmt.Println(err)
}
//t.Ret()
}
//r.SendPack(ret[0].Interface().([]byte))
}
type Blazingservice interface {
Ret() []byte
}