如何覆盖 Go 中间件中冲突的服务器标头？

在 Go 中间件中重写服务器头

背景

在 Go 中，你可以使用中间件处理和修改 HTTP 请求和响应。但是，当添加服务器中间件来控制 HTTP 标头时，如果其他处理程序也设置了服务器标头，您可能会遇到响应中存在多个服务器标头的问题。

可能的解决方案

• 自定义 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 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)
    })
}

• 反转中间件顺序：您可以反转中间件的顺序，将服务器中间件放在最后，因此它对服务器标头有最终决定权。
• Inner-Most中间件： 或者，使服务器中间件成为最内部的中间件，允许其拦截并覆盖外部处理程序设置的其他标头。

附加说明

• 自定义 ResponseWriter 包装器方法引入了一种新类型，需要更多代码维护。
• 反转或将服务器中间件置于最内层可以改变中间件链的预期流程。
• 首选解决方案取决于应用程序的具体要求和复杂性。

以上是如何覆盖 Go 中间件中冲突的服务器标头？的详细内容。更多信息请关注PHP中文网其他相关文章！