http - Go 的 http.MaxBytesReader，为什么要传入 writer？

Question

直观地说，我认为当您创建 MaxByteReader 并传入 http.ResponseWriter 时，它会为您写出状态代码。但事实并非如此，作者实际上在做什么？
例子:

func maxBytesMiddleware(next http.Handler) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        r.Body = http.MaxBytesReader(w, r.Body, 1)

        next.ServeHTTP(w, r)
    })
}

func mainHandler(w http.ResponseWriter, r *http.Request) {
    var i interface{}
    err := json.NewDecoder(r.Body).Decode(&i)
    if err != nil {
        fmt.Println(err.Error())
    }
}

func TestMaxBytesMiddleware(t *testing.T) {
    handlerToTest := maxBytesMiddleware(http.HandlerFunc(mainHandler))

    req := httptest.NewRequest(http.MethodPost, "http://test.com", bytes.NewReader(json.RawMessage(`{"hello":"world"}`)))

    recorder := httptest.NewRecorder()
    handlerToTest.ServeHTTP(recorder, req)

    if recorder.Result().StatusCode != http.StatusRequestEntityTooLarge {
        t.Errorf("expected %d got %d", http.StatusRequestEntityTooLarge, recorder.Result().StatusCode)
    }
}

但是当这个测试运行时，我得到了这个:

http: request body too large
--- FAIL: TestMaxBytesMiddleware (0.00s)
    main_test.go:37: expected 413 got 200

如果我想要我认为这个函数所做的所需功能，我需要将我的 mainHandler 更改为如下所示:

func mainHandler(w http.ResponseWriter, r *http.Request) {
    var i interface{}
    err := json.NewDecoder(r.Body).Decode(&i)
    if err != nil {
        if err.Error() == "http: request body too large" {
            w.WriteHeader(http.StatusRequestEntityTooLarge)
            return
        }

        fmt.Println(err.Error())
    }
}

那么那个作家到底是为了什么？

Answer 1

如果 MaxBytesReader 在读取整个正文之前停止，它会在 writer 上设置一些标志，以确保在发送响应后关闭 HTTP 连接。通常服务器愿意从同一个连接中读取另一个请求(HTTP keepalive)，但是如果前一个请求的未读位仍在管道中，它就不能这样做，所以它必须关闭连接，强制客户端如果要发送更多请求，则建立新连接。
这是使用私有(private) requestTooLarge 完成的。 http.ResponseWriter的方法.