为了账号安全,请及时绑定邮箱和手机立即绑定

如何将上下文值从 Gin 中间件传播到 gqlgen 解析器?

如何将上下文值从 Gin 中间件传播到 gqlgen 解析器?

Go
达令说 2022-08-24 20:07:49
我正在尝试提取令牌身份验证中间件中的user_id,并将其传递给gqlgen的graphql解析器函数(以填充created_by并updated_by GraphQL模式的列)。身份验证部分工作没有问题。杜松子酒中间件:    var UID = "dummy"    func TokenAuthMiddleware() gin.HandlerFunc {        return func(c *gin.Context) {            err := auth.TokenValid(c.Request)            if err != nil {                c.JSON(http.StatusUnauthorized, "You need to be authorized to access this route")                c.Abort()                return            }            //            UID, _ = auth.ExtractTokenID(c.Request)            //c.Set("user_id", UID)                        c.Next()        }    }    func GetUID() string {        return UID    }graphql 解析器:    var ConstID = middleware.GetUID()        func (r *mutationResolver) CreateFarmer(ctx context.Context, input model.NewFarmer) (*model.Farmer, error) {        //Fetch Connection and close db        db := model.FetchConnection()        defer db.Close()            //var ConstID, _ = uuid.NewRandom()            log.Println(ctx)            farmer := model.Farmer{Name: input.Name, Surname: input.Surname, Dob: input.Dob, Fin: input.Fin, PlotLocLat: input.PlotLocLat, PlotLocLong: input.PlotLocLong, CreatedAt: time.Now(), UpdatedAt: time.Now(), CreatedBy: ConstID, UpdatedBy: ConstID}        db.Create(&farmer)        return &farmer, nil    }在这里,我尝试使用全局变量UID来执行此操作,但是UID的值在中间件中没有得到更新,因此,我在和列中获得了“虚拟”值。我知道不鼓励使用全局变量,我对其他想法持开放态度。谢谢CreatedByUpdatedBy
查看完整描述

1 回答

?
慕田峪9158850

TA贡献1794条经验 获得超7个赞

使用 传播值。context.Context


如果您使用的是 gqlgen,则必须记住传递给解析器函数的实例来自 (假设您按照 gqlgen 文档中的建议设置了集成)。context.Context*http.Request


因此,使用Go-Gin,您应该能够通过一些额外的管道来做到这一点:


func TokenAuthMiddleware() gin.HandlerFunc {

    return func(c *gin.Context) {

        UID := // ... get the UID somehow

        

        ctx := context.WithValue(c.Request.Context(), "user_id", UID)

        c.Request = c.Request.WithContext(ctx)

        c.Next()

    }

}

然后,您通常会在解析器中获得该值:


func (r *mutationResolver) CreateFarmer(ctx context.Context, input model.NewFarmer) (*model.Farmer, error) {

    UID, _ := ctx.Value("user_id").(string)

    // ...

}

一个例子(虽然没有杜松子酒)也可以在这里找到


查看完整回答
反对 回复 2022-08-24
  • 1 回答
  • 0 关注
  • 120 浏览
慕课专栏
更多

添加回答

举报

0/150
提交
取消
微信客服

购课补贴
联系客服咨询优惠详情

帮助反馈 APP下载

慕课网APP
您的移动学习伙伴

公众号

扫描二维码
关注慕课网微信公众号