http响应的内容类型在使用外部客户端时会发生变化，但在单元测试中是正确的

Question

我遇到了一个奇怪的情况。我想从http处理程序返回内容类型application/json; charset=utf-8。

func handleTest() http.HandlerFunc {
return func(w http.ResponseWriter, r *http.Request) {
if r.Header.Get("Accept") != "application/json" {
w.WriteHeader(http.StatusNotAcceptable)
return
}
w.WriteHeader(http.StatusOK)
w.Header().Set("Content-Type", "application/json; charset=utf-8")
json.NewEncoder(w).Encode(map[string]string{"foo": "bar"})
}
}

当我在单元测试中检查这一点时，它是正确的。这个测试没有失败。

func TestTestHandler(t *testing.T) {
request, _ := http.NewRequest(http.MethodGet, "/test", nil)
request.Header.Set("Accept", "application/json")
response := httptest.NewRecorder()
handleTest().ServeHTTP(response, request)
contentType := response.Header().Get("Content-Type")
if contentType != "application/json; charset=utf-8" {
t.Errorf("Expected Content-Type to be application/json; charset=utf-8, got %s", contentType)
return
}
}

但当我尝试使用curl(和其他客户端(时，结果是text/plain; charset=utf-8。

$ curl -H 'Accept: application/json' localhost:8080/test -v
*   Trying 127.0.0.1:8080...
* TCP_NODELAY set
* Connected to localhost (127.0.0.1) port 8080 (#0)
> GET /test HTTP/1.1
> Host: localhost:8080
> User-Agent: curl/7.68.0
> Accept: application/json
> 
* Mark bundle as not supporting multiuse
< HTTP/1.1 200 OK
< Date: Tue, 28 Dec 2021 13:02:27 GMT
< Content-Length: 14
< Content-Type: text/plain; charset=utf-8
< 
{"foo":"bar"}
* Connection #0 to host localhost left intact

我试过用这个治疗卷发、失眠和蟒蛇。在所有三种情况下，内容类型均为text/plain; charset=utf-8。

是什么导致了这个问题，我该如何解决？

Answer 1

从http包文档：

WriteHeader发送一个HTTP响应标头，其中包含所提供的状态代码。

和

调用WriteHeader(或Write(后更改标头映射无效，除非修改的标头是尾部。

所以您正在设置；内容类型"；标头之后的标头已经发送到客户端。虽然嘲讽可能会起作用，因为存储头的缓冲区可以在WriteHeader调用后进行修改。但当实际使用TCP连接时，您无法做到这一点。

所以只需移动w.WriteHeader(http.StatusOK)，使其发生在w.Header().Set(...)之后