当请求具有正文(卷曲,邮递员)时,Go http 上下文无法捕获取消信号

lns*_*shi 3 curl http go postman

可以使用以下简单的 go 代码片段重现该问题吗?

简单的 http 服务器:

package main

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

func handler(w http.ResponseWriter, r *http.Request) {
    go func(done <-chan struct{}) {
        <-done
        fmt.Println("message", "client connection has gone away, request got cancelled")
    }(r.Context().Done())

    time.Sleep(30 * time.Second)
    fmt.Fprintf(w, "Hi there, I love %s!\n", r.URL.Path[1:])
}

func main() {
    http.HandleFunc("/", handler)
    log.Fatal(http.ListenAndServe(":8080", nil))
}
Run Code Online (Sandbox Code Playgroud)

从 http 服务器上方开始,如果我GET使用 curl(邮递员也)发送一个简单的请求,例如:

curl -X GET http://localhost:8080/
Run Code Online (Sandbox Code Playgroud)

然后按Ctrl+C终止请求,然后我就可以在服务器端看到打印的消息:

message client connection has gone away, request got cancelled
Run Code Online (Sandbox Code Playgroud)

以上是我期望的正确行为:模拟当客户端离开时服务器可以捕获它然后尽早取消所有不必要的工作的情况。

但是当我发送一个带有请求正文的 POST 请求时,这种预期的行为不会发生,<-done直到请求截止日期满足才会捕获信号。

curl -X POST http://localhost:8080/ -H 'Content-Type: application/json' -d '{}'
Run Code Online (Sandbox Code Playgroud)

总结我的问题:

  1. 为什么 curl(postman) GETPOST(有或没有请求正文)请求会产生如此大的不同?
  2. 我应该如何使用 go context 包正确处理这种情况,我的意思是尽快捕获客户端消失信号,从而进一步取消服务器端不必要的工作以尽早释放资源。

Cer*_*món 6

读取请求正文以检测客户端何时关闭连接:

func handler(w http.ResponseWriter, r *http.Request) {
    go func(done <-chan struct{}) {
        <-done
        fmt.Println("message", "client connection has gone away, request got cancelled")
    }(r.Context().Done())

    io.Copy(ioutil.Discard, r.Body) // <-- read the body
    time.Sleep(30 * time.Second)
    fmt.Fprintf(w, "Hi there, I love %s!\n", r.URL.Path[1:])
}
Run Code Online (Sandbox Code Playgroud)

net/http 服务器通过读取连接来检查关闭的连接。在应用程序开始读取请求正文(如果有)之前,不会开始读取。

  • 我不知道这方面有任何文档。该行为源于以下事实:读取或写入是 Go 中检测关闭连接的唯一方法。 (4认同)