登录
首页 >  Golang >  Go问答

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

来源:stackoverflow

时间:2024-04-26 13:36:36 408浏览 收藏

亲爱的编程学习爱好者,如果你点开了这篇文章,说明你对《如何将上下文值从 Gin 中间件传播到 gqlgen 解析器?》很感兴趣。本篇文章就来给大家详细解析一下,主要介绍一下,希望所有认真读完的童鞋们,都有实质性的提高。

问题内容

我正在尝试在令牌身份验证中间件中提取 user_id 并将其传递给 gqlgen 的 graphql 解析器函数(以填充 graphql 架构的created_by 和updated_by 列)。身份验证部分工作没有任何问题。

gin 中间件:

    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 列中获得“虚拟”值。我知道不鼓励使用全局变量,并且我对其他想法持开放态度。谢谢


正确答案


使用 context.context 传播值。

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

因此,使用 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)
    // ...
}

还提供了一个示例(尽管没有杜松子酒)here

接受的答案对我不起作用。深入挖掘 Gin source code 告诉我,我需要使用 contextwithfallbacktrue 的引擎,或者简单地使用 c.set()。如果键是字符串,ctx.value() 将在 gin 的上下文 c.keys 中查找并返回它,即使作为 context.context 而不是 *gin.context 访问也是如此。

所以对我来说,这有效:

func TokenAuthMiddleware() gin.HandlerFunc {
    return func(c *gin.Context) {
        UID := // ... get the UID somehow
        
        c.Set("user_id", UID)
        c.Next()
    }
}

// ...
func (r *mutationResolver) CreateFarmer(ctx context.Context, input model.NewFarmer) (*model.Farmer, error) {
    UID, _ := ctx.Value("user_id").(string)
    // ...
}

我不确定为什么 contextwithfallback 对我来说是 false,但无论如何我更喜欢这个结果,因为它更易于阅读。

这适用于杜松子酒 1.8.1。

好了,本文到此结束,带大家了解了《如何将上下文值从 Gin 中间件传播到 gqlgen 解析器?》,希望本文对你有所帮助!关注golang学习网公众号,给大家分享更多Golang知识!

声明:本文转载于:stackoverflow 如有侵犯,请联系study_golang@163.com删除
相关阅读
更多>
最新阅读
更多>
课程推荐
更多>