优雅使用GoFrame共享变量Context示例详解
作者:王中阳Go 发布时间:2024-04-27 15:27:02
前言摘要
昨天和同事merge代码又遇到了很多冲突,发现之前有些方法写的参数不规范,没有传入Context,不方便进行链路追踪。他在review项目代码,基本把项目中的方法都加了Context参数。
今天就为大家介绍一下Context的使用,告诉大家Context是什么?怎么用?为什么要用Context以及使用中的小技巧和注意问题。
Context是什么?
Context
指的是标准库的context.Context
,是一个接口对象,常用于异步IO
控制以及上下文流程变量的传递。
本文将要介绍的是Context
如何优雅的在业务流程中进行变量的传递,以及为什么需要要进行变量的传递。
为什么需要Context?
在Go
的执行流程中,特别是HTTP/RPC
执行流程中,不存在”全局变量”获取请求参数的方式,只有通过上下文Context
变量传递到后续流程的方法中。
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
...go
// 获取自定义键值对
service.Context.Get(ctx).Data[key]
五、注意问题
上下文变量只传递必须的链路参数数据,不要什么参数都往里面塞。特别是一些方法参数、传参的数据,千万不能往上下文里面塞,而应当显示的传递方法参数。
上下文变量仅用作运行时临时使用,不可做持久化存储长期使用。
来源:https://juejin.cn/post/7113118741776793636
猜你喜欢
- 1,创建测试表CREATE TABLE `testsign` ( `userid` int(5) DEFAULT NULL, `user
- 引入依赖#?导入模块import?pymysqlimport?pandas?as?pdimport?numpy?as?npimport?ti
- 使用JS技术实现QQ阅读类似的点击展开、收起效果,具体内容如下一、定义展开函数showdiv(),实现点击"全文"按钮,
- 本文实例讲述了Python sqlite3事务处理方法。分享给大家供大家参考,具体如下:sqlite3事务总结:在connect()中不传入
- 容器与可迭代对象在正式开始前先补充一些基本概念在 Python 中存在容器 与 可迭代对象容器:用来存储多个元素的数据结构,例如 列表,元组
- 方案有很多种,我这里简单说一下:1. into outfileSELECT * FROM mytable  
- 一、节点的定义dom节点树图中可见节点HTML文档中的每个成分都是一个节点:整个文档是一个文档节点每个HTML标签是一个元素节点包含在HTM
- jenkins是什么? Jenkins是一个开
- urllib是python的一个获取url(Uniform Resource Locators,统一资源定址符)了,可以利用它来抓取远程的数
- MySql:MySQL数据库实现分页比较简单,提供了 LIMIT函数。一般只需要直接写到sql语句后面就行了。LIMIT子 句可以用来限制由
- 程序流Python 解释器在其最简单的级别,以类似的方式操作,即从程序的顶端开始,然后一行一行地顺序执行程序语句。例如,清单 1 展示了几个
- 使用一阶线性方程预测波士顿房价载入的数据是随sklearn一起发布的,来自boston 1993年之前收集的506个房屋的数据和价格。loa
- 什么是事件代理(Event Delegation)?如果不太了解的朋友,可详细阅读:《Event delegation in JavaScr
- PHP simplexml_import_dom() 函数实例获取 DOM 文档节点并转换为 SimpleXML 节点:<?php $
- 这一次将使用pymysql来进行一次对MySQL的增删改查的全部操作,相当于对前五次的总结:先查阅数据库:现在编写源码进行增删改查操作,源码
- 创建临时表,往临时表插入数据的时候报的错误。一开始提示没有打开主键,后来打开主键就提示上述错误异常。从网上查找资料没有找到,然后又到群里问各
- 1、上下文概念上下文,说白了就是所谓的语境,就是语言环境。比如单独拎出来一篇文章的某一句话,我们可能不能理解它的意思,但是通过这句话所处的语
- map()是python的一个内建函数, 他能够通过函数来处理序列,比如,我们相关一个数组[0,1,2,3,4,5]所有的数字都+2 , 当
- 一、开头匹配从字符串开头开始匹配返回匹配对象;如果找不到匹配,则为Noneimport reprint(re.match('飞兔小哥
- 在代码中添加以下两行可以解决:torch.backends.cudnn.enabled = Truetorch.backends.cudnn