Go
middle~~POS=TRUNC
Sök…
Introduktion
I Go kan Middleware användas för att köra kod före och efter hanteringsfunktion. Den använder kraften i enfunktionsgränssnitt. Kan introduceras när som helst utan att påverka den andra mellanprogrammet. För ex: autentiseringsloggning kan läggas till i senare utvecklingsstadier utan att störa den befintliga koden.
Anmärkningar
Mellanvarans signatur bör vara (http.ResponseWriter, * http.Request) dvs av typen http.handlerFunc .
Normal hanterarfunktion
func loginHandler(w http.ResponseWriter, r *http.Request) {
// Steps to login
}
func main() {
http.HandleFunc("/login", loginHandler)
http.ListenAndServe(":8080", nil)
}
Middleware Beräkna den tid som krävs för att hanterarenFunc ska kunna köras
// logger middlerware that logs time taken to process each request
func Logger(h http.Handler) http.Handler {
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
startTime := time.Now()
h.ServeHttp(w,r)
endTime := time.Since(startTime)
log.Printf("%s %d %v", r.URL, r.Method, endTime)
})
}
func loginHandler(w http.ResponseWriter, r *http.Request) {
// Steps to login
}
func main() {
http.HandleFunc("/login", Logger(loginHandler))
http.ListenAndServe(":8080", nil)
}
CORS Middleware
func CORS(h http.Handler) http.Handler {
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
origin := r.Header.Get("Origin")
w.Header().Set("Access-Control-Allow-Origin", origin)
if r.Method == "OPTIONS" {
w.Header().Set("Access-Control-Allow-Credentials", "true")
w.Header().Set("Access-Control-Allow-Methods", "GET,POST")
w.RespWriter.Header().Set("Access-Control-Allow-Headers", "Content-Type, X-CSRF-Token, Authorization")
return
} else {
h.ServeHTTP(w, r)
}
})
}
func main() {
http.HandleFunc("/login", Logger(CORS(loginHandler)))
http.ListenAndServe(":8080", nil)
}
Auth Middleware
func Authenticate(h http.Handler) http.Handler {
return CustomHandlerFunc(func(w *http.ResponseWriter, r *http.Request) {
// extract params from req
// post params | headers etc
if CheckAuth(params) {
log.Println("Auth Pass")
// pass control to next middleware in chain or handler func
h.ServeHTTP(w, r)
} else {
log.Println("Auth Fail")
// Responsd Auth Fail
}
})
}
Återställningshanterare för att förhindra att servern kraschar
func Recovery(h http.Handler) http.Handler {
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request){
defer func() {
if err := recover(); err != nil {
// respondInternalServerError
}
}()
h.ServeHTTP(w , r)
})
}
Modified text is an extract of the original Stack Overflow Documentation
Licensierat under CC BY-SA 3.0
Inte anslutet till Stack Overflow