在Go語言中,中間件是指在請求處理過程中插入的一段代碼,用于執行一些額外的操作,如身份驗證、日志記錄、性能監控等。開發Go語言中間件時,需要掌握以下要點:
中間件函數通常接收兩個參數:http.ResponseWriter
和 *http.Request
。返回值可以是 error
,用于處理錯誤情況。
func MiddlewareFunc(next http.Handler) http.Handler {
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
// 中間件邏輯
// 調用 next.ServeHTTP(w, r) 處理請求
})
}
Go標準庫提供了一些內置的中間件,如日志中間件、恢復中間件等。
import (
"log"
"net/http"
)
func LoggingMiddleware(next http.Handler) http.Handler {
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
log.Printf("Request: %s %s", r.Method, r.URL.Path)
next.ServeHTTP(w, r)
})
}
可以根據需求自定義中間件,例如添加身份驗證、記錄請求時間等。
func AuthenticationMiddleware(next http.Handler) http.Handler {
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
// 身份驗證邏輯
token := r.Header.Get("Authorization")
if token == "" {
http.Error(w, "Unauthorized", http.StatusUnauthorized)
return
}
next.ServeHTTP(w, r)
})
}
可以使用多個中間件來組合成一個完整的請求處理流程。
func main() {
mux := http.NewServeMux()
mux.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
w.Write([]byte("Hello, World!"))
})
// 組合中間件
wrappedMux := LoggingMiddleware(AuthenticationMiddleware(mux))
log.Fatal(http.ListenAndServe(":8080", wrappedMux))
}
通過掌握以上要點,你可以有效地開發和應用Go語言中間件,提升應用程序的可維護性和擴展性。