在Go的执行流程中，特别是HTTP/RPC执行流程中，不存在”全局变量”获取请求参数的方式，只有将上下文Context变量传递到后续流程的方法中，而Context上下文变量即包含了所有需要传递的共享变量。并且该Context中的共享变量应当是事先约定的，并且往往存储为对象指针形式。

通过Context上下文共享变量非常简单，以下我们通过一个项目中的示例来展示如何在实战化项目中传递和使用通用的共享变量。

一、结构定义

上下文对象中往往存储一些需要共享的变量，这些变量通常使用结构化的对象来存储，以方便维护。例如，我们在model定义一个上下文中的共享变量：

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

二、逻辑封装

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

  1. // 上下文管理服务
  2. var Context = new(contextService)
  6. // 初始化上下文对象指针到上下文对象中，以便后续的请求流程中可以修改。
  7. func (s *contextService) Init(r *ghttp.Request, customCtx *model.Context) {
  8.     r.SetCtxVar(model.ContextKey, customCtx)
  10. // 获得上下文变量，如果没有设置，那么返回nil
  11. func (s *contextService) Get(ctx context.Context) *model.Context {
  12.     value := ctx.Value(model.ContextKey)
  13.     if value == nil {
  14.         return nil
  15.     }
  16.     if localCtx, ok := value.(*model.Context); ok {
  17.         return localCtx
  18.     }
  19.     return nil
  20. }
  22. // 将上下文信息设置到上下文请求中，注意是完整覆盖
  23. func (s *contextService) SetUser(ctx context.Context, ctxUser *model.ContextUser) {
  24.     s.Get(ctx).User = ctxUser
  25. }

三、上下文变量注入

上下文的变量必须在请求一开始便注入到请求流程中，以便于其他方法调用。在HTTP请求中我们可以使用GoFrame的中间件来实现。在GRPC请求中我们也可以使用拦截器来实现。在service层的middleware管理对象中，我们可以这样来定义：

该中间件初始化了用户执行流程共享的对象，并且存储到context.Context变量中的对象是指针类型*model.Context。这样任何一个地方获取到这个指针，既可以获取到里面的数据，也能够直接修改里面的数据。

四、上下文变量使用

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

  1. // 执行用户登录
  3. }
  5. // 查询内容列表
  6. func (s *contentService) GetList(ctx context.Context, r *define.ContentServiceGetListReq) (*define.ContentServiceGetListRes, error) {
  7.     ...
  8. }
  10. // 创建回复内容
  11. func (s *replyService) Create(ctx context.Context, r *define.ReplyServiceCreateReq) error {
  12.     ...
  13. }

此外，约定俗成的，方法的最后一个返回参数往往是error类型。如果您确定此方法内部永不会产生error，那么可以忽略。

通过service中封装的以下方法，将context.Context上下文变量传递进去即可。context.Context上下文变量在GoFrame框架的HTTP请求中可以通过r.Context()方法获取，在GRPC请求中，编译生成的pb文件中执行方法的第一个参数即固定是context.Context。

  1. // 设置自定义键值对
  2. service.Context.Get(ctx).Data[key] = value
  4. ...
  6. // 获取自定义键值对
  7. service.Context.Get(ctx).Data[key]

五、注意事项

  1. 上下文变量只传递必须的链路参数数据，不要什么参数都往里面塞。特别是一些方法参数传参的数据，别往里面塞，而应当显示传递方法参数。