Wie kann Go Middleware HTTP-Header zuverlässig und ohne Duplizierung steuern?

HTTP-Header von Outer Go Middleware steuern

HTTP-Middleware in Go bietet eine bequeme Möglichkeit, HTTP-Anfragen und -Antworten abzufangen und zu ändern. Allerdings kann die Steuerung von Headern von einer äußeren Middleware eine Herausforderung sein, da dabei vorhandene Header überschrieben werden müssen, ohne Duplikate einzuführen.

Betrachten Sie die folgende Server-Middleware, die den „Server“-Header festlegt:

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)
    })
}

Wenn diese Middleware zur Antwortkette hinzugefügt wird, setzt sie erfolgreich den „Server“-Header. Wenn jedoch ein anderer Handler in der Kette ebenfalls den „Server“-Header setzt, kommt es in der Antwort zu doppelten Headern.

Die Herausforderung entsteht, weil ServeHTTP das Schreiben in den ResponseWriter nach Abschluss der Anfrage ausdrücklich verbietet. Ein Ansatz besteht darin, einen benutzerdefinierten ResponseWriter zu erstellen, der Header-Schreibvorgänge abfängt und den „Server“-Header vor dem ersten Schreibvorgang einfügt.

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()
}

// Server attaches a Server header to the response.
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)
    })
}

Durch die Verwendung eines benutzerdefinierten ResponseWriter können wir sicherstellen, dass der „Server“-Header hinzugefügt wird nur einmal, unabhängig vom Verhalten anderer Handler. Dieser Ansatz führt eine zusätzliche Indirektionsebene ein, behält aber die gewünschte Funktionalität bei.

Das obige ist der detaillierte Inhalt vonWie kann Go Middleware HTTP-Header zuverlässig und ohne Duplizierung steuern?. Für weitere Informationen folgen Sie bitte anderen verwandten Artikeln auf der PHP chinesischen Website!