GoFrame 如何优雅的共享变量?Context的使用

开发 后端
今天就为大家介绍一下Context的使用:告诉大家Context是什么?怎么用?为什么要用Context以及使用中的小技巧和注意问题。

前言

昨天merge代码,发现了好多冲突,原因是同事在review项目,做链路追踪,发现老项目有不少方法传参不规范,没有传入Context,不方便做链路追踪。

所以把这些方法和调用进行了修改,导致了大量冲突,修复冲突又花了好长时间

所以:传参规范还是要在项目启动时就确定好呀,一定要搞清楚Context怎么用呀!

今天就为大家介绍一下Context的使用:

告诉大家Context是什么?怎么用?为什么要用Context以及使用中的小技巧和注意问题。

Context是什么?

Context​指的是标准库的context.Context​,是一个接口对象,常用于异步IO控制以及上下文流程变量的传递。

本文将要介绍的是Context如何优雅的在业务流程中进行变量的传递,以及为什么需要要进行变量的传递。

为什么需要Context?

在Go​的执行流程中,特别是HTTP/RPC​执行流程中,没有通过”全局变量”获取请求参数的方式,只能通过上下文Context变量,传递到后续执行流程的方法中。

如何使用?

Context上下文变量,包含了所有需要传递的共享变量。

并且Context中的共享变量是需要事先约定的,并且往往存储为对象指针形式。

通过Context上下文,共享变量非常简单,下面通过示例带大家了解一下如何传递和使用通用的共享变量。

一、结构定义

上下文对象中往往存储一些需要共享的变量,这些变量通常使用结构化的对象来存储,以方便维护。

例如,我们在model定义一个上下文中的共享变量:

const (
// 上下文变量存储键名,前后端系统共享
ContextKey = "ContextKey"
)

// 请求上下文结构
type Context struct {
Session *ghttp.Session // 当前Session管理对象
User *ContextUser // 上下文用户信息
Data g.Map // 自定KV变量,业务模块根据需要设置,不固定
}

// 请求上下文中的用户信息
type ContextUser struct {
Id uint // 用户ID
Passport string // 用户账号
Nickname string // 用户名称
Avatar string // 用户头像
}

介绍

  • model.ContextKey​常量表示存储在context.Context​上下文变量中的键名,该键名用于从传递的context.Context变量中存储/获取业务自定义的共享变量。
  • model.Context​结构体中的Session​表示当前请求的Session​对象,在GoFrame​框架中每个HTTP​请求对象中都会有一个空的Session对象,该对象采用了懒初始化设计,只有在真正执行读写操作时才会初始化。
  • model.Context​结构体中的User​表示当前登录的用户基本信息,只有在用户登录后才有数据,否则是nil。
  • model.Context​结构体中的Data​,用于存储自定义的KV​变量,因此一般来说开发者无需再往context.Context​上下文变量中增加自定义的键值对,而是直接使用model.Context​对象的这个Data属性即可。

二、逻辑封装

由于该上下文对象也是和业务逻辑相关的,因此我们需要通过service对象将上下文变量封装起来以方便其他模块使用。

// 上下文管理服务
var Context = new(contextService)

type contextService struct{}

// 初始化上下文对象指针到上下文对象中,以便后续的请求流程中可以修改。
func (s *contextService) Init(r *ghttp.Request, customCtx *model.Context) {
r.SetCtxVar(model.ContextKey, customCtx)
}

// 获得上下文变量,如果没有设置,那么返回nil
func (s *contextService) Get(ctx context.Context) *model.Context {
value := ctx.Value(model.ContextKey)
if value == nil {
return nil
}
if localCtx, ok := value.(*model.Context); ok {
return localCtx
}
return nil
}

// 将上下文信息设置到上下文请求中,注意是完整覆盖
func (s *contextService) SetUser(ctx context.Context, ctxUser *model.ContextUser) {
s.Get(ctx).User = ctxUser
}

Tips

在架构设计中,在哪个场景下设置Context是非常关键的。

上下文的变量必须在请求一开始便注入到请求流程中,以便于其他方法调用,所以在中间件中来实现是非常优雅的选择。

我们来看下面的介绍:

三、上下文变量注入

在HTTP​请求中我们可以使用GoFrame的中间件来实现。

在GRPC请求中我们也可以使用拦截器来实现。

在service​层的middleware管理对象中,我们可以这样来定义:

// 自定义上下文对象
func (s *middlewareService) Ctx(r *ghttp.Request) {
// 初始化,务必最开始执行
customCtx := &model.Context{
Session: r.Session,
Data: make(g.Map),
}
service.Context.Init(r, customCtx)
if userEntity := Session.GetUser(r.Context()); userEntity != nil {
customCtx.User = &model.ContextUser{
Id: userEntity.Id,
Passport: userEntity.Passport,
Nickname: userEntity.Nickname,
Avatar: userEntity.Avatar,
}
}
// 将自定义的上下文对象传递到模板变量中使用
r.Assigns(g.Map{
"Context": customCtx,
})
// 执行下一步请求逻辑
r.Middleware.Next()
}

这个中间件,初始化了用户执行流程 共享的对象,并且存储到context.Context​变量中的对象是指针类型*model.Context。

这样做的好处是:任何一个地方获取到这个指针,不仅可以获取到里面的数据,而且能够直接修改里面的数据。

TIPS

如果Session​中存在用户登录后的存储信息,那么也会将需要共享的用户基本信息写入到*model.Context中。

四、上下文变量使用

方法定义

方法定义的第一个输入参数往往预留给context.Context​类型参数使用,以便接受上下文变量,特别是service层的方法。

例如:

// 执行用户登录
func (s *userService) Login(ctx context.Context, loginReq *define.UserServiceLoginReq) error {
...
}

// 查询内容列表
func (s *contentService) GetList(ctx context.Context, r *define.ContentServiceGetListReq) (*define.ContentServiceGetListRes, error) {
...
}

// 创建回复内容
func (s *replyService) Create(ctx context.Context, r *define.ReplyServiceCreateReq) error {
...
}

TIPS

另外一个好习惯是:方法的最后一个返回参数往往是error​类型。如果确定方法内部永不会产生error,那么可以忽略。

​Context​​对象获取

通过service​中封装的以下方法,将context.Context上下文变量传递进去即可。

context.Context​上下文变量在GoFrame​框架的HTTP​请求中可以通过r.Context()方法获取。

在GRPC​请求中,编译生成的pb​文件中执行方法的第一个参数即固定是context.Context。

service.Context.Get(ctx)

自定义​Key-Value​

我们可以通过以下方式设置/获取自定义的key-value键值对。

// 设置自定义键值对
service.Context.Get(ctx).Data[key] = value

// 获取自定义键值对
service.Context.Get(ctx).Data[key]

五、注意问题

上下文变量只传递必须的链路参数数据,不要什么参数都往里面塞。特别是一些方法参数、传参的数据,千万不能往上下文里面塞,而应当用显示的方式传递方法参数。

上下文变量仅用作运行时临时使用,不可做持久化存储长期使用。

总结

这篇文章详细的为大家介绍了GoFrame上下文对象Context的知识点:

Context的作用:在业务流程中进行变量的共享。

Context的结构定义、逻辑封装、如何在中间件中注入、如何通过Context设置值和取值、Context如何自定义key-value、以及在项目开发中使用的注意问题。

欢迎Star GoFrame:https://github.com/gogf/gf

本文转载自微信公众号「 程序员升级打怪之旅」,作者「王中阳Go」,可以通过以下二维码关注。

转载本文请联系「 程序员升级打怪之旅」公众号。

责任编辑:武晓燕 来源: 程序员升级打怪之旅
相关推荐

2023-06-16 09:08:39

ReactContextRFC

2021-08-10 07:41:24

ContextWaitGroupGoroutine

2017-07-26 11:32:50

NETRabbitMQ系统集成

2015-11-26 10:53:45

LinuxWindowsMac OS

2022-11-29 10:42:46

GoFrame技巧脚手架

2022-09-14 08:16:48

装饰器模式对象

2021-03-28 09:17:18

JVM场景钩子函数

2022-06-02 10:02:47

Kubectl更新应用Linux

2020-05-08 14:45:00

JS代码变量

2019-11-15 09:58:04

LinuxAsciinemapython

2022-02-28 09:19:12

MDK芯片软件

2022-01-21 09:45:42

Mozilla SOKubernetesLinux

2022-11-15 07:50:47

ORM链式操作删除

2022-11-11 07:48:56

ORM链式轮播图

2022-12-28 12:11:57

PHPGo框架

2021-01-19 10:35:49

JVM场景函数

2020-10-16 11:48:06

服务器系统运维

2022-10-26 10:15:53

GoFramePHP数组

2020-05-14 09:15:52

设计模式SOLID 原则JS

2017-09-25 16:39:51

移动端
点赞
收藏

51CTO技术栈公众号