Go中ResponseController的SetWriteDeadline未生效，如何取消超时写入？

你的代码里SetWriteDeadline没生效的核心原因是标准库的http.ResponseWriter默认会缓冲响应数据：每次Write的1字节远小于默认缓冲区大小（约4KB），所以数据只是被放到内存缓冲区，并没有真正执行网络写操作，截止时间自然不会触发。直到循环结束后缓冲区才会被刷新，但此时客户端已经断开连接（curl返回空回复），但服务端的Write操作因为没触及网络层，不会立即返回错误。

方案一：强制刷新缓冲区（推荐，兼容性好） #

通过http.Flusher接口强制将缓冲区数据写入网络，让SetWriteDeadline作用于实际的网络写操作。每次Write后调用Flush()，并可以用CheckWriteTimeout()主动检查超时状态。

修改后的代码：

package main

import (
    "log"
    "net/http"
    "time"
)

func main() {
    http.HandleFunc("/", func(rw http.ResponseWriter, r *http.Request) {
        log.Println(r.Method, r.URL.Path)

        rc := http.NewResponseController(rw)
        dl := time.Now().Add(3 * time.Second)
        if err := rc.SetWriteDeadline(dl); err != nil {
            log.Println(err)
            return
        }
        log.Println("write deadline:", dl)

        // 断言获取Flusher接口，确保能强制刷新缓冲区
        flusher, ok := rw.(http.Flusher)
        if !ok {
            log.Println("response writer不支持刷新操作")
            http.Error(rw, "服务器内部错误", http.StatusInternalServerError)
            return
        }

        var total int
        for _, b := range []byte("What!\n") {
            time.Sleep(time.Second)

            n, err := rw.Write([]byte{b})
            if err != nil {
                log.Println("写入错误:", err)
                return
            }

            total += n
            log.Println("已写入字节数", n)

            // 强制将缓冲区数据写入网络
            flusher.Flush()

            // 主动检查是否超时或连接已关闭
            if err := rc.CheckWriteTimeout(); err != nil {
                log.Println("写入超时或连接关闭:", err)
                return
            }
        }

        log.Println(r.Method, r.URL.Path, "总写入字节数", total)
    })

    http.ListenAndServe(":8080", nil)
}

运行后，当超过3秒截止时间，后续的Flush或CheckWriteTimeout会触发超时错误，循环立即终止。

方案二：用Context超时+主动取消请求 #

通过context.WithTimeout创建超时上下文，超时后调用ResponseController.CancelRequest()主动终止请求，让后续Write操作返回错误。

代码示例：

package main

import (
    "context"
    "log"
    "net/http"
    "time"
)

func main() {
    http.HandleFunc("/", func(rw http.ResponseWriter, r *http.Request) {
        log.Println(r.Method, r.URL.Path)

        rc := http.NewResponseController(rw)
        // 创建3秒超时的上下文
        ctx, cancel := context.WithTimeout(r.Context(), 3*time.Second)
        defer cancel()

        // 启动goroutine监听超时，超时后取消请求
        go func() {
            <-ctx.Done()
            if ctx.Err() == context.DeadlineExceeded {
                log.Println("写入超时，取消请求")
                _ = rc.CancelRequest()
            }
        }()

        var total int
        for _, b := range []byte("What!\n") {
            time.Sleep(time.Second)

            n, err := rw.Write([]byte{b})
            if err != nil {
                log.Println("写入错误:", err)
                return
            }

            total += n
            log.Println("已写入字节数", n)

            // 检查上下文是否已取消，提前退出循环
            select {
            case <-ctx.Done():
                log.Println("上下文已取消，退出")
                return
            default:
            }
        }

        log.Println(r.Method, r.URL.Path, "总写入字节数", total)
    })

    http.ListenAndServe(":8080", nil)
}

这个方案不需要依赖缓冲区刷新，通过主动取消请求来终止写入流程，适合无法使用Flusher的场景。

方案三：直接操作底层连接（不推荐，兼容性差） #

如果能获取到HTTP服务器的底层net.Conn，可以直接给连接设置写截止时间，绕过http.ResponseWriter的缓冲区。但这个方法仅适用于标准库的http.Server，其他HTTP框架可能不支持。

代码示例：

package main

import (
    "log"
    "net"
    "net/http"
    "time"
)

func main() {
    http.HandleFunc("/", func(rw http.ResponseWriter, r *http.Request) {
        log.Println(r.Method, r.URL.Path)

        // 断言获取底层连接（仅标准库http.Server有效）
        resp, ok := rw.(interface{ Conn() net.Conn })
        if !ok {
            log.Println("无法获取底层连接")
            http.Error(rw, "服务器内部错误", http.StatusInternalServerError)
            return
        }
        conn := resp.Conn()

        dl := time.Now().Add(3 * time.Second)
        if err := conn.SetWriteDeadline(dl); err != nil {
            log.Println(err)
            return
        }
        log.Println("write deadline:", dl)

        var total int
        for _, b := range []byte("What!\n") {
            time.Sleep(time.Second)

            n, err := conn.Write([]byte{b})
            if err != nil {
                log.Println("写入错误:", err)
                return
            }

            total += n
            log.Println("已写入字节数", n)
        }

        log.Println(r.Method, r.URL.Path, "总写入字节数", total)
    })

    http.ListenAndServe(":8080", nil)
}

内容的提问来源于stack exchange，提问作者425nesp