Go Gin 从中间件设置和访问上下文值

Question

我正在尝试在中间件中设置我的用户上下文，然后尝试检查用户是否在其他处理程序函数中具有权限。但是由于某种原因，当我尝试从上下文访问用户时，它会以 nils 的形式返回。中间件代码似乎正在工作，当我传递一个有效的 jwt 令牌时，它显示用户正在中间件函数的上下文中设置。但是一旦我点击 getCurrentUser 函数，它就会说它为零。这是代码：中间件// Middleware wraps the request with auth middlewarefunc Middleware(path string, sc *cfg.Server, orm *orm.ORM) gin.HandlerFunc {    logger.Info("[Auth.Middleware] Applied to path: ", path)    return gin.HandlerFunc(func(c *gin.Context) {        t, err := ParseToken(c, sc)        if err != nil {            authError(c, err)        } else {            if claims, ok := t.Claims.(jwt.MapClaims); ok {                if claims["exp"] != nil {                    issuer := claims["iss"].(string)                    userid := claims["jti"].(string)                    email := claims["email"].(string)                    if claims["aud"] != nil {                        audiences := claims["aud"].(interface{})                        logger.Warnf("\n\naudiences: %s\n\n", audiences)                    }                    if claims["alg"] != nil {                        algo := claims["alg"].(string)                        logger.Warnf("\n\nalgo: %s\n\n", algo)                    }                    if user, err := orm.FindUserByJWT(email, issuer, userid); err != nil {                        authError(c, ErrForbidden)                    } else {                        if user != nil {                            c.Request = addToContext(c, consts.ProjectContextKeys.UserCtxKey, user)                            logger.Debug("User: ", user.ID)                        }                        c.Next()                    }                } else {                    authError(c, ErrMissingExpField)                }            } else {                authError(c, err)            }        }    })}

Answer 1

问题是您将用户存储在一个上下文中，但随后您尝试从另一个上下文中检索用户。value*gin.Context和 value*gin.Context.Request.Context是两个独立的上下文值。

您正在使用请求的上下文来存储用户：

c.Request.WithContext(context.WithValue(c.Request.Context(), key, value))

然后您使用 gin 上下文来检索用户：

func getCurrentUser(ctx context.Context) *dbm.User {

    cu := ctx.Value(utils.ProjectContextKeys.UserCtxKey).(*dbm.User)

    // ...

func Get(orm *orm.ORM) gin.HandlerFunc {

    return func(ctx *gin.Context) {

        cu := getCurrentUser(ctx) // here you're passing *gin.Context to the function.

        // ...

因此，要解决这个问题，将传递给getCurrentUser调用的值更改为：

func Get(orm *orm.ORM) gin.HandlerFunc {

    return func(ctx *gin.Context) {

        cu := getCurrentUser(ctx.Request.Context())

        if ok, err := cu.HasPermission(consts.Permissions.Create, consts.EntityNames.Users); !ok || err != nil {

            ctx.String(http.StatusUnauthorized, "BAD")

        }

    }

}