Go中介軟體如何有效控制HTTP頭？

使用外部Go 中間件控制HTTP 標頭

問題：覆蓋中間件中的伺服器標頭

您面臨的情況是Go 中有中間件嘗試設定自訂“伺服器”標頭。但是，應用程式或其他中間件設定的現有「伺服器」標頭仍然存在，導致回應中出現多個「伺服器」標頭。

不允許的標頭修改

根據 HTTP 語義，修改回應標頭ServeHTTP 方法返回後被禁止。此限制旨在在請求處理和回應最終確定之間建立明確的界限。

建議的解決方案

1.自訂 ResponseWriter：

定義一個自訂 ResponseWriter 包裝器，用於攔截標頭修改操作。在寫入任何標頭之前，包裝器會插入您的自訂「伺服器」標頭。這種方法增加了額外的間接層，但提供了對標頭管理的細粒度控制。

這是自訂 ResponseWriter 的範例實作：

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

在中介軟體中，您可以使用此自訂 ResponseWriter來控制標頭插入：

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

2.中間件順序反轉：

另一種方法是反轉中間件的順序，以便您的「伺服器」標頭設定中間件在所有其他中間件之後執行。這可確保您的標頭修改是回應最終確定之前的最後一個操作。

3.最內層中間件：

如果可能，請將「伺服器」標頭設定中間件放置為中間件的最內層。這消除了外部中間件在設定“Server”標頭後修改它的可能性。

4.回應修改：

作為最後的手段，您可以考慮使用自訂 http.Handler 來攔截回應並相應地修改「Server」標頭。這種方法需要小心處理，以避免破壞應用程式中的其他功能。

以上是Go中介軟體如何有效控制HTTP頭？的詳細內容。更多資訊請關注PHP中文網其他相關文章！