Go HTTP 响应写入前执行逻辑方法

本文介绍如何在 Go 的 http.ResponseWriter 写入结束（flush）前，通过包装处理器函数的方式注入额外响应内容，并规避 Content-Length 失效、错误响应污染等常见陷阱。

本文介绍如何在 Go 的 `http.ResponseWriter` 写入结束（flush）前，通过包装处理器函数的方式注入额外响应内容，并规避 `Content-Length` 失效、错误响应污染等常见陷阱。

在 Go 的 HTTP 服务开发中，http.ResponseWriter 并未提供类似 OnFlush 或 OnClose 的钩子接口，无法直接监听“响应即将发送”这一时机。但实际场景中，我们常需在主处理器执行完毕后、响应真正写出前，追加日志标记、埋点脚本、统一尾部数据（如 JSON API 的元信息字段）或调试头信息。最简洁、标准且符合 net/http 设计哲学的方案是：包装 Handler 函数，在其返回后立即操作 ResponseWriter。

以下是一个基础实现：

func myHandler(w http.ResponseWriter, r *http.Request) {
    w.WriteHeader(http.StatusOK)
    w.Write([]byte("Hello"))
}

func wrapper(w http.ResponseWriter, r *http.Request) {
    myHandler(w, r)                 // 执行原始业务逻辑
    w.Write([]byte(" World"))        // 追加内容 —— 此时响应尚未发出，仍可写入
}

func main() {
    http.HandleFunc("/", wrapper)
    http.ListenAndServe(":8080", nil)
}

访问 / 将返回完整响应体：Hello World。

⚠️ 然而，该模式在生产环境中需谨慎处理两个关键问题：

• 状态码与错误流干扰：若 myHandler 内部调用 http.Error() 或显式写入 500 状态码并输出错误页，后续 w.Write() 仍会追加内容，导致响应体结构破坏（例如 HTML 错误页后拼接纯文本）。因此，推荐在 wrapper 中检查已写入的状态码（可通过自定义 ResponseWriter 拦截 WriteHeader 调用），或约定业务 handler 返回 error，由 wrapper 统一决策是否追加：

  func safeWrapper(next http.HandlerFunc) http.HandlerFunc {
      return func(w http.ResponseWriter, r *http.Request) {
          // 使用 ResponseWriter 包装器捕获状态码与 header
          rw := &responseWriter{ResponseWriter: w, statusCode: http.StatusOK}
          next(rw, r)
          if rw.statusCode >= 200 && rw.statusCode < 400 {
              w.Write([]byte("\n<!-- appended by middleware -->"))
          }
      }
  }

• Content-Length 失效风险：一旦 handler 显式设置 Content-Length（如 w.Header().Set("Content-Length", "5")），后续 w.Write() 会导致实际响应体长度超过声明值，违反 HTTP 协议，可能触发客户端截断或代理拒绝转发。根本解法是避免手动设 Content-Length，交由 Go 标准库自动计算（即不调用 w.Header().Set("Content-Length", ...)）；若必须控制，应使用缓冲写入 + 重写头：

  func bufferedWrapper(next http.HandlerFunc) http.HandlerFunc {
      return func(w http.ResponseWriter, r *http.Request) {
          var buf bytes.Buffer
          bw := &bufferedResponseWriter{
              ResponseWriter: w,
              buffer:         &buf,
          }
          next(bw, r)
          // 追加内容到缓冲区
          buf.WriteString("\n[Generated at: " + time.Now().Format(time.RFC3339) + "]")
          // 清空原始 header 中的 Content-Length（如有）
          w.Header().Del("Content-Length")
          // 写入完整缓冲内容
          w.Write(buf.Bytes())
      }
  }

✅ 总结：Go 中实现“响应末尾钩子”的最佳实践不是监听 flush 事件（该事件不可控且无公开 API），而是利用 HTTP 处理链的同步执行特性——在 handler 函数返回后、ServeHTTP 完成前，对 ResponseWriter 进行最终操作。只要规避状态码误判与 Content-Length 冲突，该模式轻量、可靠、无需依赖第三方中间件，是构建可观测性增强、A/B 测试注入、合规水印等场景的理想基础。