http.ResponseWriter.WriteHeader() 导致死锁

Question

我试图在 golang 中使用 httptest 包。我发现了一些我不明白的东西。这是code：

package main

import (
    "fmt"
    "io/ioutil"
    "log"
    "net/http"
    "net/http/httptest"
)

func main() {
    ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        w.Write([]byte("Hello1"))
    }))
    ts.Close()
    ts = httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        w.Write([]byte("Hello2"))
    }))
    ts.Close()
    ts = httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        w.WriteHeader(100)
        w.Write([]byte("Hello3"))
    }))

    res, err := http.Get(ts.URL)
    if err != nil {
        log.Fatal(err)
    }
    greeting, err := ioutil.ReadAll(res.Body)
    res.Body.Close()
    if err != nil {
        log.Fatal(err)
    }

    ts.Close()
    fmt.Printf("%s", greeting)
}

在这个代码示例中，我多次尝试打开和关闭httptest 服务器。不知何故，它在 The Go Playground 中造成了僵局。我在自己的环境（Go 版本：go1.7.4 darwin/amd64）上尝试过，它导致挂起，根本没有响应。

我的问题是：为什么w.WriteHeader(100) 会导致死锁而w.WriteHeader(200) 不会？是Golang核心库的错误还是我误解了一些用法？谢了！

Answer 1

如果你稍微修改代码：

package main

import (
    "fmt"
    "io/ioutil"
    "log"
    "net/http"
    "net/http/httptest"
)

func main() {
    ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        w.Write([]byte("Hello1"))
    }))
    ts.Close()
    ts = httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        w.Write([]byte("Hello2"))
    }))
    ts.Close()
    ts = httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        w.WriteHeader(100)
        w.Write([]byte("Hello3"))
    }))

    fmt.Println("before get")   ///// . <----
    res, err := http.Get(ts.URL)
    fmt.Println("after get")    ///// . <----
    if err != nil {
        log.Fatal(err)
    }
    greeting, err := ioutil.ReadAll(res.Body)
    res.Body.Close()
    if err != nil {
        log.Fatal(err)
    }

    ts.Close()
    fmt.Printf("%s", greeting)
}

并运行它 - 你只会看到第一行。

这意味着 Go http 客户端需要您提供更多数据。所以就挂了

    res, err := http.Get(ts.URL)

并且无法访问下面的ts.Close()。

接下来 - 让我们修改一个测试，所以它会关闭一个连接，这样会释放一个客户端等待锁：

ts = httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
    w.WriteHeader(100)
    w.Write([]byte("Hello3"))
    hj, _ := w.(http.Hijacker)
    conn, _, _ := hj.Hijack()
    conn.Close()
}))

我明确地关闭了连接，但这样你就可以得到检查字符串和测试完成。试试看。

当然这是违反 HTTP 协议的，所以我收到错误：

Get http://127.0.0.1:54243: net/http: HTTP/1.x transport connection broken: unexpected EOF