我正在做一个小型演示,试图解释基本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
函数发送一个额外的参数(例如,一个字符串,应比较此 header 而不是"SecretValue"
的字符串),该怎么做? 在此先感谢您,我是使用golang的新手,并且想了解更多信息。
最佳答案
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"))
关于rest - 如何将参数传递给MiddlewareFunc?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/61069784/