Bagaimana untuk Mengatasi Pengepala HTTP dalam Go Middleware Tanpa Duplikasi?

Mengatasi Pengepala HTTP daripada External Go Middleware

Latar Belakang

Middleware dalam Go digunakan untuk memintas dan mengubah suai permintaan dan respons sebelum ia sampai kepada pengendali. Walau bagaimanapun, apabila berbilang komponen middleware mengubah suai pengepala yang sama, ia boleh menyebabkan berbilang kejadian pengepala itu ditetapkan dalam respons.

Pertimbangkan contoh middleware berikut:

func Server(h http.Handler, serverName string) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        w.Header().Set("Server", serverName)
        h.ServeHTTP(w, r)
    })
}

Apabila middleware ini digunakan bersama-sama dengan perisian tengah lain yang menetapkan pengepala "Pelayan", ia menghasilkan berbilang pengepala "Pelayan" dalam respons.

Custom ResponseWriter

Untuk menyelesaikan isu ini, kami boleh mencipta ResponseWriter tersuai yang memintas panggilan WriteHeader() dan menetapkan pengepala yang sesuai sebelum menghantar permintaan selanjutnya .

type serverWriter struct {
    w           http.ResponseWriter
    name        string
    wroteHeader bool
}

func (s serverWriter) WriteHeader(code int) {
    if s.wroteHeader == false {
        s.w.Header().Set("Server", s.name)
        s.wroteHeader = true
    }
    s.w.WriteHeader(code)
}

func (s serverWriter) Write(b []byte) (int, error) {
    return s.w.Write(b)
}

func (s serverWriter) Header() http.Header {
    return s.w.Header()
}

Dengan menggunakan ResponseWriter tersuai ini dalam perisian tengah Pelayan kami, kami boleh memastikan bahawa pengepala "Pelayan" ditetapkan sekali sahaja.

func Server(h http.Handler, serverName string) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        sw := serverWriter{
            w:           w,
            name:        serverName,
            wroteHeader: false,
        }
        h.ServeHTTP(sw, r)
    })
}

Penyelesaian ini membolehkan kami mengawal pengepala HTTP daripada perisian tengah luar tanpa melanggar semantik ServeHTTP.

Atas ialah kandungan terperinci Bagaimana untuk Mengatasi Pengepala HTTP dalam Go Middleware Tanpa Duplikasi?. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!