request.context不会关闭连接

时间:2017-12-01 10:26:42

标签: go http.client

我正在使用context.Context取消http请求

我发现虽然我得到了“上下文取消”,但底层套接字连接仍然可用,我可以在几秒钟后得到响应。一旦设置了请求,它是否以这种方式设计来读取响应?

这是代码

func SendRequest(ctx context.Context, url string) {
    req, err := http.NewRequest("GET", url, nil)
    if err != nil {
        fmt.Println(err)
    }
    req = req.WithContext(ctx)

    res, err := client.Do(req)
    select {
    case <-ctx.Done():
        fmt.Printf("%s Canceled\n", url)
        //client.Transport.(*http.Transport).CancelRequest(req)
        //client.Transport.(*http.Transport).CloseIdleConnections()
    }
    if res != nil {
        defer res.Body.Close()
    }
    if err != nil {
        fmt.Printf("Failed: %v\n", err)
    } else {
        io.Copy(ioutil.Discard, res.Body)
        fmt.Printf("return status: %d\n", url, res.StatusCode)
    }
}

我请求的URL将在几秒钟后返回,因此我仍然可以读取响应正文,并在进程退出后关闭连接。

这是一个重现问题的简单代码

func client() {
    ctx, cancel := context.WithCancel(context.Background())
    client := http.DefaultClient
    request, _ := http.NewRequest("GET", "http://127.0.0.1:9090", nil)
    req := request.WithContext(ctx)

    go func() {
        client.Do(req)
    }()

    time.Sleep(time.Duration(1) * time.Second)
    cancel()

    <-time.After(time.Duration(10) * time.Second)

}

func sayhelloName(w http.ResponseWriter, r *http.Request) {
    time.Sleep(time.Duration(10) * time.Second)
    fmt.Fprintf(w, "Hello world!")
}

func server() {
    http.HandleFunc("/", sayhelloName)
    err := http.ListenAndServe(":9090", nil)
    if err != nil {
        log.Fatal("ListenAndServe: ", err)
    }
}

连接在10秒后关闭 connection was closed after exactly 10 second

1 个答案:

答案 0 :(得分:4)

取消上下文时,您自己无需做任何取消请求。这已经由标准的http包as documented处理。

你需要的只是:

func SendRequest(ctx context.Context, url string) {
    req, err := http.NewRequest("GET", url, nil)
    if err != nil {
        panic(err)
    }
    req = req.WithContext(ctx)

    res, err := client.Do(req)
    if err != nil {
        panic(err)
    }
    defer res.Body.Close()
    io.Copy(ioutil.Discard, res.Body)
    fmt.Printf("return status: %d\n", url, res.StatusCode)
}

修改

您的TCP转储确认一切正常,并且正如我上面所描述的那样。

要打破它:

您的出站请求确实在一秒钟后被取消,因为您的TCP转储确认:

6 2017-12-04 06:10:21.864955 0.0993000      40 TCP       17503 → 9090 [FIN, ACK] Seq=96 Ack=1 Win=8192 Len=0
7 2017-12-04 06:10:21.864955 0.0000000      40 TCP       9090 → 17503 [ACK] Seq=1 Ack=97 Win-7936 Len=0

第6行的FIN表示HTTP客户端告诉服务器&#34;我已经和你说话了#34;。第7行的ACK表示服务器响应了请求,并关闭了连接。

但是,您的HTTP处理程序会忽略已取消的请求,并尝试响应,生成第8行:

8 2017-12-04 06:10:30.868955 9.004000     169 HTTP      HTTP/1.1 200 OK  (text/plain)

但是服务器收到连接无效的消息:

9 2017-12-04 06:10:30.868955  0.000000      40 TCP       17503 → 8080 [RST, ACK] Seq=97 Ack=130 Win=0 Len=0

RST表示连接已重置,目前无效。这是因为连接在9秒前终止了。

所以你看,你的请求会立即取消,完全按照原样取消。

您可以添加到代码中的唯一改进是让服务器中的HTTP处理程序通过在传入请求被取消时提前退出来实际检测并遵守此类取消。