Go ResponseController SetReadDeadline 怎样限制请求体读取时间

Go HTTP 服务端遇到慢上传或客户端迟迟不发完请求体时，可以在 Handler 中用 http.NewResponseController(w).SetReadDeadline 设置读取截止时间。它约束的是整次请求的读取，包括请求体；deadline 到期后，继续读取会返回错误。要注意，这不是响应写超时，也不是请求体大小限制。

官方地址：https://pkg.go.dev/net/http#ResponseController.SetReadDeadline

先分清 SetReadDeadline 控制的范围

SetReadDeadline 从服务端底层连接的读取边界生效，覆盖请求读取过程，不只是下一次 Request.Body.Read。因此它适合处理“客户端连接还在，但迟迟没有把请求发完”的场景。截止时间可以是绝对时间，也可以传零值表示不设置读取 deadline。

它解决的是时间边界，不替代 http.MaxBytesReader 之类的大小边界。生产接口通常需要把“最多读多少字节”和“最晚读到什么时候”分开设计：前者避免大请求占用内存，后者避免慢连接长期占用连接资源。

在原始 ResponseWriter 上设置截止时间

最小用法是创建控制器、设置一个绝对截止时间，然后正常读取请求体。ResponseController 会尝试调用原始 ResponseWriter 或其 Unwrap 返回值提供的能力；如果当前写入器不支持该方法，应把能力错误和读取错误分开处理。

package main

import (
    "errors"
    "fmt"
    "io"
    "net/http"
    "time"
)

func upload(w http.ResponseWriter, r *http.Request) {
    controller := http.NewResponseController(w)

    // 使用绝对时间，限制本次请求从服务端开始读取到完成的窗口。
    deadline := time.Now().Add(10 * time.Second)
    if err := controller.SetReadDeadline(deadline); err != nil {
        // 不支持该能力是部署或 ResponseWriter 包装边界，不等同于请求超时。
        if errors.Is(err, http.ErrNotSupported) {
            http.Error(w, "当前 ResponseWriter 不支持读取截止时间", http.StatusNotImplemented)
            return
        }
        http.Error(w, "设置读取截止时间失败", http.StatusInternalServerError)
        return
    }

    // 读取错误要单独判断；deadline 到期通常会在这里暴露出来。
    body, err := io.ReadAll(r.Body)
    if err != nil {
        http.Error(w, fmt.Sprintf("读取请求体失败: %v", err), http.StatusRequestTimeout)
        return
    }

    // 读取完成后按业务处理 body；示例只返回长度，不保存真实数据。
    fmt.Fprintf(w, "received=%d", len(body))
}

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

示例中的错误分层很重要：SetReadDeadline 返回错误，说明当前写入器没有实现所需能力或控制器无法调用它；io.ReadAll 返回错误，才是本次读取过程遇到了超时、连接中断或其他输入问题。真实服务还应限制读取大小，并避免把整个上传体一次性读入内存。

读取完成后清除或收紧策略

官方语义中，零值表示没有读取 deadline。若一个 Handler 会继续复用同一连接完成后续工作，可以在确定不再需要该读取限制时清除它：

func clearReadDeadline(w http.ResponseWriter) error {
    controller := http.NewResponseController(w)

    // 零值表示取消读取 deadline，不会把请求重新设置成一个新的未来窗口。
    if err := controller.SetReadDeadline(time.Time{}); err != nil {
        // 调用方可以把不支持能力记录为兼容性问题，而不是吞掉所有错误。
        return fmt.Errorf("清除读取 deadline: %w", err)
    }
    return nil
}

不要把“已经超过截止时间”理解成“再设置一个更晚的时间就能续命”。官方文档明确说明，deadline 过期后再设置更晚的时间不能延长它。更稳妥的做法是：在进入读取前一次性确定时间窗口；超时后关闭或终止当前请求；如果业务允许重试，由上层创建新的请求和新的生命周期。

另外，ResponseController 不能在 Handler 的 ServeHTTP 返回后继续使用。把 controller 或底层写入器交给后台 goroutine，试图在 Handler 结束后修改 deadline，会把清晰的请求生命周期变成竞态和不可预测的连接状态。

区分读取超时与写入超时

这几个边界可以组合，但不要用一个 deadline 代替全部策略。若 ResponseWriter 被中间件包装，优先保留原始 writer 或实现正确的 Unwrap，否则控制器可能得到 ErrNotSupported。若只需要限制整个 Handler 的执行时间，还应结合 context 或服务端超时配置，不要把连接读取 deadline 当作通用业务超时。

相关问题

SetReadDeadline 是每次 Read 单独计时吗？

不是。它设置的是读取整个请求的截止时间，包含请求体；时间点一旦到达，后续读取会失败。

能不能传 time.Time{} 延长已超时的请求？

零值表示没有 deadline，但不能把已经过期的读取恢复成可继续读取。超时请求应结束当前生命周期。

为什么调用 SetReadDeadline 返回不支持？

ResponseController 依赖原始 ResponseWriter 或 Unwrap 后的实现提供该能力。检查中间件是否正确暴露底层 writer，并把能力错误与请求读取错误分开记录。

SetReadDeadline 和 MaxBytesReader 要选哪个？

一个限制时间，一个限制大小。慢连接防护与大请求防护通常需要同时存在，但参数应根据接口类型、上传大小和客户端网络条件分别设定。