如何将参数传递给 MiddlewareFunc？

Question

我正在做一个小演示，试图解释一个基本的 HTTP 处理程序是如何工作的，我找到了以下示例：package mainfunc 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 authfunc 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 middlewarefunc 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"))