mux路由中間件的使用
代碼示例：
package main
import (
"encoding/json"
"fmt"
"github.com/gorilla/mux"
"net/http"
"strings"
)
// 定義Fish的結構體
type Fish struct {
Name string `json:"name"`
}
var fishlist []Fish
// 獲取Fish列表
func getFishListHandler(w http.ResponseWriter, r *http.Request) {
// Header設置Content-Type為json格式绽媒，輸出json格式的結果，看起來更方便
w.Header().Set("Content-Type", "application/json")
w.WriteHeader(http.StatusOK)
json.NewEncoder(w).Encode(fishlist)
}
func logMiddleware(h http.Handler) http.Handler {
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
// 打印請求路徑
fmt.Println(r.RequestURI)
//fmt.Fprintln(w, "%s\r\n", r.URL)
h.ServeHTTP(w, r)
})
}
// 定義權限的中間件結構體
type AuthMiddleware struct {
token map[string]string
}
func (amw *AuthMiddleware) Populate() {
amw.token = make(map[string]string)
amw.token["000"] = "user0"
amw.token["aaa"] = "userA"
amw.token["05ft"] = "randomUser"
amw.token["deadbeef"] = "user0"
}
func (amw *AuthMiddleware) authMiddleware(next http.Handler) http.Handler {
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
token := strings.Trim(r.Header.Get("X-Session-Token"), " ")
if token == "" {
fmt.Fprintf(w, "token is error \r\n")
}
if user, found := amw.token[token]; found {
//we found the token in out map
fmt.Printf("Authenticated user: %s\n", user)
fmt.Fprintf(w, "Authenticated user: %s\n", user)
// Pass down the request to the next middleware (or final handler)
next.ServeHTTP(w, r)
} else {
// Write an error and stop the handler chain
http.Error(w, "Forbidden", http.StatusForbidden)
}
})
}
func main() {
// 初始化fishlist的數(shù)據
fishlist = append(fishlist, Fish{Name: "金魚"})
fishlist = append(fishlist, Fish{Name: "鯉魚"})
// 初始化一個新的路由
r := mux.NewRouter()
r.HandleFunc("/fish/list", getFishListHandler).Methods("GET")
// 使用中間件
//r.Use(logMiddleware)
amw := AuthMiddleware{}
amw.Populate()
r.Use(amw.authMiddleware)
// 監(jiān)聽8002端口
http.ListenAndServe(":8002", r)
}
運行程序，通過postman或其它http請求工具發(fā)送請求获三，下面是測試的結果。
1棺聊、GET請求：http://localhost:8002/fish/list贞谓，headers不加參數(shù)，直接請求返回：
token is error
Forbidden
2犀暑、GET請求：http://localhost:8002/fish/list烁兰，headers增加X-Session-Token：000，結果：
Authenticated user: user0
[{"name":"金魚"},{"name":"鯉魚"}]
image.png
3广辰、GET請求：http://localhost:8002/fish/list主之，headers增加X-Session-Token：0000(或者隨便輸入)，結果：（返回狀態(tài)為403）
Forbidden
