รูปแบบมิดเดิลแวร์

มิดเดิลแวร์คืออะไร

ลายเซ็นของมิดเดิลแวร์

type Middleware func(http.Handler) http.Handler

มิดเดิลแวร์บันทึกเหตุการณ์อย่างง่าย

func Logger(next http.Handler) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        log.Printf("%s %s", r.Method, r.URL.Path)
        next.ServeHTTP(w, r)
        log.Printf("done %s", r.URL.Path)
    })
}

การเชื่อมมิดเดิลแวร์

handler := Logger(Auth(RateLimit(myHandler)))
http.ListenAndServe(":8080", handler)

ตัวช่วยสร้างโซ่มิดเดิลแวร์

func Chain(h http.Handler, m ...Middleware) http.Handler {
    for i := len(m)-1; i >= 0; i-- {
        h = m[i](h)
    }
    return h
}
// Usage: Chain(myHandler, Logger, Auth, RateLimit)

มิดเดิลแวร์ Auth

func Auth(next http.Handler) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        token := r.Header.Get("Authorization")
        if !isValid(token) {
            http.Error(w, "Unauthorized", http.StatusUnauthorized)
            return
        }
        next.ServeHTTP(w, r)
    })
}

มิดเดิลแวร์กู้คืนจากแพนิก

func Recovery(next http.Handler) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        defer func() {
            if err := recover(); err != nil {
                log.Printf("panic: %v", err)
                http.Error(w, "Internal Server Error", 500)
            }
        }()
        next.ServeHTTP(w, r)
    })
}

การส่งข้อมูลผ่าน Context

ctx := context.WithValue(r.Context(), userKey, user)
next.ServeHTTP(w, r.WithContext(ctx))

ตัวห่อการตอบกลับ

type statusRecorder struct {
    http.ResponseWriter
    status int
}
func (r *statusRecorder) WriteHeader(code int) {
    r.status = code
    r.ResponseWriter.WriteHeader(code)
}

ลำดับมีความสำคัญ

มิดเดิลแวร์เฉพาะเส้นทางเทียบกับมิดเดิลแวร์ส่วนกลาง

ตรวจสอบความเข้าใจ

ทบทวน: รูปแบบมิดเดิลแวร์