golang游戏服务器 - tgf系列课程04

用户登录

介绍了如何使用tgf自带的登录功能进行用户的登录操作,并且编写机器人客户端的一个模拟请求代码

需求描述

用户请求登录,登录成功之后请求HelloWorld接口.

Common

接口定义和生成

接口定义

新增登录接口

type IHallService interface {Login(ctx context.Context, args *rpc.Args[*pb.LoginRequest], reply *rpc.Reply[*pb.LoginResponse]) (err error)HelloWorld(ctx context.Context, args *rpc.Args[*pb.HelloWorldRequest], reply *rpc.Reply[*pb.HelloWorldResponse]) (err error)
}

生成api文件

查看项目文档,在kit项目中创建生成api接口文件脚本

func main() {//设置导出的golang目录util.SetAutoGenerateAPICodePath("../common/api")//根据接口生成对应的api结构util.GeneratorAPI[services.IHallService](hall.ModuleName, hall.Version, "api")
}

Service

逻辑端

完善接口逻辑

使用 rpc.UserLogin函数登录

func (s *service) Login(ctx context.Context, args *rpc.Args[*pb.LoginRequest], reply *rpc.Reply[*pb.LoginResponse]) (err error) {var userId stringvar pbData *pb.LoginResponseif args.GetData().GetAccount() == "admin" && args.GetData().GetPassword() == "123" {userId = util.GenerateSnowflakeId()rpc.UserLogin(ctx, userId)pbData = &pb.LoginResponse{Success: true}} else {pbData = &pb.LoginResponse{Success: false}}reply.SetData(pbData)return
}

Robot

请求代码

监听登录和HelloWorld接口的回调,并发起请求

func TestHelloWorld(t *testing.T) {rb := CreateRobot()rb.RegisterCallbackMessage(api.HelloWorld.MessageType, func(i robot.IRobot, bytes []byte) {resp := &pb.HelloWorldResponse{}proto.Unmarshal(bytes, resp)t.Log(resp.Message)}).RegisterCallbackMessage(api.Login.MessageType, func(i robot.IRobot, bytes []byte) {resp := &pb.LoginResponse{}proto.Unmarshal(bytes, resp)if resp.Success {i.Send(api.HelloWorld.MessageType, &pb.HelloWorldRequest{Name: "tgf"})} else {t.Log("login fail")}})//rb.Send(api.Login.MessageType, &pb.LoginRequest{Account:  "admin",Password: "123",})select {}
}