如何将参数传递给 MiddlewareFunc？

Question

我正在做一个小演示，试图解释一个基本的 HTTP 处理程序是如何工作的，我找到了以下示例：

package main

func router() *mux.Router {

    router := mux.NewRouter()

    auth := router.PathPrefix("/auth").Subrouter()

    auth.Use(auth.ValidateToken)

    auth.HandleFunc("/api", middleware.ApiHandler).Methods("GET")

    return router

}

func main() {

    r := router()

    http.ListenAndServe(":8080", r)

}

package auth

func ValidateToken(next http.Handler) http.Handler {

    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {

        var header = r.Header.Get("secret-access-token")

        json.NewEncoder(w).Encode(r)

        header = strings.TrimSpace(header)

        if header == "" {

            w.WriteHeader(http.StatusForbidden)

            json.NewEncoder(w).Encode("Missing auth token")

            return

        }

        if header != "SecretValue" {

            w.WriteHeader(http.StatusForbidden)

            json.NewEncoder(w).Encode("Auth token is invalid")

            return

        }

        json.NewEncoder(w).Encode(fmt.Sprintf("Token found. Value %s", header))

        next.ServeHTTP(w, r)

    })

}

package middleware

func ApiHandler(w http.ResponseWriter, r *http.Request) {

    w.WriteHeader(http.StatusOK)

    w.Header().Set("Content-Type", "application/json")

    json.NewEncoder(w).Encode("SUCCESS!")

    return

}

一切都可以理解，但我想到了两个问题：

为什么不需要在此行中发送参数：secure.Use(auth.ValidateToken)？

如果我想向这个auth.ValidateToken函数发送一个额外的参数，（例如，一个字符串，应该比较这个头而不是"SecretValue"），怎么做？

在此先感谢，我对使用 golang 有点陌生，想了解更多。

Answer 1

auth.Use将函数作为参数，并且auth.ValidateToken是您传递的函数。如果要向 发送参数auth.ValidateToken，可以编写一个接受该参数的函数，并返回一个中间件函数，如下所示：func GetValidateTokenFunc(headerName string) func(http.Handler) http.Handler {  return func(next http.Handler) http.Handler {     return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {        var header = r.Header.Get(headerName)        ...     }  }}然后你可以这样做：    auth.Use(auth.GetValidateTokenFunc("headerName"))