Блог

http.ResponseWriter имеет только две функции: одна есть WriteHeader() , другая есть Write() . Поэтому я считаю, что его нужно настроить в других местах. Затем я нашел это: https://golang.org/src/net/http/server.go L1534

 // 1. *response (the ResponseWriter) ->
// 2. (*response).w, a *bufio.Writer of bufferBeforeChunkingSize bytes
// 3. chunkWriter.Writer (whose writeHeader finalizes Content-Length/Type)
//    and which writes the chunk headers, if needed.
// 4. conn.buf, a bufio.Writer of default (4kB) bytes, writing to ->
// 5. checkConnErrorWriter{c}, which notes any non-nil error on Write
//    and populates c.werr with it if so. but otherwise writes to:
// 6. the rwc, the net.Conn.
 

Основываясь на моем тесте, размер буфера по умолчанию составляет 4 КБ. Я вижу, что это должно быть настраиваемо, но как? Кто-нибудь знает?

Я видел аналогичное требование и соответствующие изменения Server struct , которые были предложены в прошлом https://github.com/golang/go/issues/7467. Но было рекомендовано добавить собственную буферизацию к функциям обработчика, которые должны иметь буфер пользовательского размера.

Вы можете настроить размер буфера для http.ResponseWriter , обернув текущий http.ResponseWriter с помощью буфера записи пользовательского размера, как рекомендовано в выпуске github выше.

 package main

import (
    "bufio"
    "fmt"
    "io"
    "log"
    "net/http"
    "strings"
)

func main() {
    h1 := func(w http.ResponseWriter, _ *http.Request) {
        pw := bufio.NewWriterSize(w, 10<<10) // Bigger writer of 10kb
        fmt.Println("Buffer size", pw.Size())

        strR := strings.Repeat("Hello from a Func1!n", 1000) //just to create a bigger response

        _, err := io.WriteString(pw, strR)

        if err != nil {
            fmt.Println(err)
        }

        if pw.Buffered() > 0 {
            fmt.Println("Bufferred", pw.Buffered())
            pw.Flush() // Important step read my note following this code snippet
        }
    }

    h2 := func(w http.ResponseWriter, _ *http.Request) {
        io.WriteString(w, "Hello from a Func2!n")
    }

    http.HandleFunc("/foo", h1)
    http.HandleFunc("/bar", h2)

    log.Fatal(http.ListenAndServe(":8080", nil))
}
 

Примечание: Поскольку мы завернули оригинал http.ResponseWriter , нам нужно будет управлять сбросом буфера самостоятельно pw.Flush() , т. е. Как также упоминалось в документах.

Пишущие строки ответов, которые позволяют обработчику HTTP сбрасывать буферизованные данные клиенту.

Реализации пишущей машинки HTTP/1.x и HTTP/2 по умолчанию поддерживают Flusher, но оболочки пишущей машинки могут этого не делать. Обработчики всегда должны проверять эту способность во время выполнения.