我正在使用中间件 (CheckToken) 来检查 JWT 并获取自定义声明 (Id)(它将是我数据库中用户的 ID)但是我需要将它传递给 campaign.Attack(这样我就可以知道谁是进行“攻击”的用户)，但我无法找到一种方法来做到这一点。

我试图将它作为参数传递给 token.go 中的 next(w, req, claim.id) 但我需要触摸 http.HandlerFunc 函数，所以这不是一个有效的选项。

关于如何将 claim.id 从 CheckToken 传递到 campaign.Attack() 有什么想法吗？

谢谢

         ***** main.go*****

            func main() {

                router := mux.NewRouter()

                router.HandleFunc("/attack", token.CheckToken(campaign.Attack)).Methods("GET", "OPTIONS")

               log.Fatal(http.ListenAndServe(":3000", handlers.CORS(handlers.AllowedOrigins([]string{"*"}),
                    handlers.AllowedHeaders([]string{"Content-Type", "authorization"}))(router)))
            }



            ******campaign.go*****

            package campaign

            import (
                "log"
                "net/http"

            )


            func init() {

            }

            func Attack(w http.ResponseWriter, req *http.Request) {
                log.Println("attack")

                //i need to get the claim.Id here
            }


            ****token.go****

type MyCustomClaims struct {
    Id int `json:"id"`  //the Id of the user 
    jwt.StandardClaims
}

            func CheckToken(next http.HandlerFunc) (MyCustomClaims, http.HandlerFunc) {
                return MyCustomClaims{}, http.HandlerFunc(func(w http.ResponseWriter, req *http.Request) {

                    authorizationHeader := req.Header.Get("authorization")
                    if authorizationHeader != "" {
                        bearerToken := strings.Split(authorizationHeader, " ")
                        if len(bearerToken) == 2 {
                            token, err := jwt.ParseWithClaims(bearerToken[1], &MyCustomClaims{}, func(token *jwt.Token) (interface{}, error) {
                                return []byte("magicword"), nil
                            })

                            if token.Valid {


                                if claims, ok := token.Claims.(*MyCustomClaims); ok && token.Valid {
                               //**************************
                              //***********i have the claims.id here and it works.*******
                              //**************************
                                    log.Println(claims.Id)

                              //but i need to pass it or find a way to read it in campaign.Attack()
                                    next(w, req)
                                } else {
                                    log.Println(err)
                                }
                            } else if ve, ok := err.(*jwt.ValidationError); ok {
                                if ve.Errors&jwt.ValidationErrorMalformed != 0 {


                                } else if ve.Errors&(jwt.ValidationErrorExpired|jwt.ValidationErrorNotValidYet) != 0 {
                                    // Token is either expired or not active yet

                                } 
                            } 
                        } 
                    } 


                })

            }

最佳答案

使用上下文的方法 WithValue。 取而代之的是 next(w, req) 写

   ctx := context.WithValue(r.Context(), "claim_id", claims.Id)
   next(w, req.WithContext(ctx))

和内部攻击:

   claim_id, ok := r.Context().Value("claim_id").(int)
   if !ok {
     return // I don't have context .. sorry
   }
   // use claim_id

我没有提到的，如何创建一个唯一的 key ......但有时在未来。

关于go - 将变量从中间件传递到结束函数，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/48225863/