我正在做一個小演示，試圖解釋一個基本的 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 有點陌生，想了解更多。
1 回答
青春有我
TA貢獻1784條經驗 獲得超8個贊
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"))
- 1 回答
- 0 關注
- 143 瀏覽
添加回答
舉報
0/150
提交
取消