[Pitaya Demo解读笔记]5.pipeline demo

项目目录:examples/demo/custom_metrics

本例展示了如何使用结构体验证 (struct validation)和处理函数钩子(handler hooks)

运行

服务器开起来

sh 复制代码
 go run main.go

服务监听了 3251 端口,使用命令行跑一下基本功能

有一点需要注意一下,之前的 demo 都设置了 component.WithNameFunc(strings.ToLower),所以之前的 router 都是全小写,这个示例程序并没有设置,访问的时候是大小写敏感的,否则会找不到服务器或 handler:

sh 复制代码
 > pitaya-cli
 Pitaya REPL Client
 >>> request metagameDemo.metagamehandler.createplayercheat {"name":"test", "email":"test@gmail.com", "softCurrency":10, "hardCurrency":20}
 >>> sv->{"code":"PIT-404","msg":"pitaya/handler: metagameDemo.metagamehandler.createplayercheat not found"}
 >>> request metagameDemo.metagameHandler.CreatePlayerCheat {"name":"test", "email":"test@gmail.com", "softCurrency":10, "hardCurrency":20}
 >>> sv->{"msg":"ok"}

服务器端输出日志如下:

sh 复制代码
 time="2023-10-16T16:13:59+08:00" level=info msg="Simple Before exec" requestId=BJIiSpBZ1Pue7CIr9md6iz route=metagameDemo.metagameHandler.CreatePlayerCheat source=pitaya userId=
 time="2023-10-16T16:13:59+08:00" level=info msg="Name: test" requestId=BJIiSpBZ1Pue7CIr9md6iz route=metagameDemo.metagameHandler.CreatePlayerCheat source=pitaya userId=
 time="2023-10-16T16:13:59+08:00" level=info msg="Email: test@gmail.com" requestId=BJIiSpBZ1Pue7CIr9md6iz route=metagameDemo.metagameHandler.CreatePlayerCheat source=pitaya userId=
 time="2023-10-16T16:13:59+08:00" level=info msg="SoftCurrency: 10" requestId=BJIiSpBZ1Pue7CIr9md6iz route=metagameDemo.metagameHandler.CreatePlayerCheat source=pitaya userId=
 time="2023-10-16T16:13:59+08:00" level=info msg="HardCurrency: 20" requestId=BJIiSpBZ1Pue7CIr9md6iz route=metagameDemo.metagameHandler.CreatePlayerCheat source=pitaya userId=
 time="2023-10-16T16:13:59+08:00" level=debug msg="SID=4, Data={"name":"test","email":"test@gmail.com","softCurrency":10,"hardCurrency":20}" requestId=BJIiSpBZ1Pue7CIr9md6iz route=metagameDemo.metagameHandler.CreatePlayerCheat source=pitaya userId=
 time="2023-10-16T16:13:59+08:00" level=info msg="CreatePlayerChest called" requestId=BJIiSpBZ1Pue7CIr9md6iz route=metagameDemo.metagameHandler.CreatePlayerCheat source=pitaya userId=
 time="2023-10-16T16:13:59+08:00" level=info msg="Simple After exec - response: &{ok} , error: <nil>" requestId=BJIiSpBZ1Pue7CIr9md6iz route=metagameDemo.metagameHandler.CreatePlayerCheat source=pitaya userId=

钩子方法输出了 Simple Before 和 Simple After...

如果我们给一个不满足验证的值,hardCurrency = 2000,服务器会验证输入,返回错误信息:

sh 复制代码
 >>> request metagameDemo.metagameHandler.CreatePlayerCheat {"name":"test", "email":"test@gmail.com", "softCurrency":10, "hardCurrency":2000}
 >>> sv->{"code":"PIT-000","msg":"Key: 'CreatePlayerCheatArgs.HardCurrency' Error:Field validation for 'HardCurrency' failed on the 'lte' tag"}

代码分析

结构体验证

开启结构体验证:

go 复制代码
 // main.go:101
 config.DefaultPipelines.StructValidation.Enabled = true

使用 tag validate 设置验证规则:

go 复制代码
 type CreatePlayerCheatArgs struct {
     Name         string `json:"name"`
     Email        string `json:"email" validate:"email"`
     SoftCurrency int    `json:"softCurrency" validate:"gte=0,lte=1000"`
     HardCurrency int    `json:"hardCurrency" validate:"gte=0,lte=200"`
 }

Pitaya 默认的验证器是这个包:github.com/go-playgrou...

这个功能很强大啊,在游戏开发里,默认客户端是不可信的,需要对大部分上行参数做数据验证,自己手动写就很麻烦,在 pitaya 框架里,我们可以使用 StructValidation 来实现,优雅!

结构体验证的实现用到了我们接下来要说的钩子方法,开启验证,就是在 BeforeHandler 添加了验证方法:

go 复制代码
 // builder.go: 314
 func configureDefaultPipelines(handlerHooks *pipeline.HandlerHooks) {
     handlerHooks.BeforeHandler.PushBack(defaultpipelines.StructValidatorInstance.Validate)
 }

钩子方法 BeforeHandler AfterHandler

pitaya 提供了钩子方法集:

  • BeforeHandler 在消息处理方法前执行,结构体验证就是在这里调用的
  • AfterHandler 在消息处理方法后执行

这两种钩子方法集在 ProcessHandlerMessage 里被调用:

go 复制代码
 // handler_pool.go:43
 func (h *HandlerPool) ProcessHandlerMessage(
     ctx context.Context,
     rt *route.Route,
     serializer serialize.Serializer,
     handlerHooks *pipeline.HandlerHooks,
     session session.Session,
     data []byte,
     msgTypeIface interface{},
     remote bool,
 ) ([]byte, error) {
     ...
     ctx, arg, err = handlerHooks.BeforeHandler.ExecuteBeforePipeline(ctx, arg)
     if err != nil {
         return nil, err
     }
     ...
     resp, err := util.Pcall(handler.Method, args)
     ...
     resp, err = handlerHooks.AfterHandler.ExecuteAfterPipeline(ctx, resp, err)
     ...
 }

ExecuteBeforePipelineExecuteAfterPipeline 就是遍历切片,依次调用注册的方法

相关推荐
德育处主任8 分钟前
在亚马逊云上,如何基于 VPC IPAM 的 ALB 公网 IP 预测分配?
后端
不吃肉的羊19 分钟前
PHP设置文件上传最大值
后端·php
专注物联网全栈开发22 分钟前
ESP32的IRAM用完了怎么优化
后端
雨落倾城夏未凉23 分钟前
7.QObject定时器和QTimer定时器的区别
后端·qt
洗澡水加冰25 分钟前
RAG系统工程化
后端·aigc
paopaokaka_luck1 小时前
智能推荐社交分享小程序(websocket即时通讯、协同过滤算法、时间衰减因子模型、热度得分算法)
数据库·vue.js·spring boot·后端·websocket·小程序
程序员NEO1 小时前
Spring AI 对话记忆大揭秘:服务器重启,聊天记录不再丢失!
人工智能·后端
用户21411832636021 小时前
惊爆!国内轻松白嫖 Claude Code,编程效率狂飙
后端
iccb10131 小时前
我是如何实现在线客服系统的极致稳定性与安全性的
前端·javascript·后端
M1A11 小时前
Java 面试系列第一弹:基础问题大盘点
java·后端·mysql