Не буферизирован http.ResponseWritter в Голанге
Я пишу простое веб-приложение в Go, и я хочу, чтобы мои ответы были переданы клиенту (т.е. не буферизованы и отправлены в блоках после полной обработки запроса):
func handle(res http.ResponseWriter, req *http.Request) {
fmt.Fprintf(res, "sending first line of data")
sleep(10) //not real code
fmt.Fprintf(res, "sending second line of data")
}
С клиентской точки зрения две строки будут отправляться одновременно. Любые предложения приветствуются:)
Изменить после ответа @dystroy
Это возможно для очистки после каждой записи, которую я лично делаю, но в моем случае использования этого недостаточно:
cmd := exec.Command("a long command that outputs lots of lines")
cmd.Stdout = res //where res is a http.ResponseWritter
cmd.Stderr = res
err := cmd.Run()
Я хочу, чтобы вывод моего cmd
был также очищен. В любом случае, чтобы "автозапускать" ResponseWritter?
Решение
Я нашел помощь в списке рассылки golang. Существует два способа достичь этого: используя hijacker, которые позволяют взять на себя базовое TCP-соединение HTTP или передать stdout и stderr команда в рутине, которая будет писать и стираться:
pipeReader, pipeWriter := io.Pipe()
cmd.Stdout = pipeWriter
cmd.Stderr = pipeWriter
go writeCmdOutput(res, pipeReader)
err := cmd.Run()
pipeWriter.Close()
//---------------------
func writeCmdOutput(res http.ResponseWriter, pipeReader *io.PipeReader) {
buffer := make([]byte, BUF_LEN)
for {
n, err := pipeReader.Read(buffer)
if err != nil {
pipeReader.Close()
break
}
data := buffer[0:n]
res.Write(data)
if f, ok := res.(http.Flusher); ok {
f.Flush()
}
//reset buffer
for i := 0; i < n; i++ {
buffer[i] = 0
}
}
}
Последнее обновление
Еще лучше: http://play.golang.org/p/PpbPyXbtEs
Ответы
Ответ 1
Как подразумевается в документации, некоторые ResponseWriter
могут реализовать интерфейс Flusher
.
Это означает, что вы можете сделать что-то вроде этого:
func handle(res http.ResponseWriter, req *http.Request) {
fmt.Fprintf(res, "sending first line of data")
if f, ok := res.(http.Flusher); ok {
f.Flush()
} else {
log.Println("Damn, no flush");
}
sleep(10) //not real code
fmt.Fprintf(res, "sending second line of data")
}
Будьте осторожны, что буферизация может произойти во многих других местах сети или на стороне клиента.
Ответ 2
Извините, если я неправильно понял ваш вопрос, но что-то вроде нижеследующего делает трюк?
package main
import (
"bytes"
"fmt"
"net/http"
)
func handler(w http.ResponseWriter, r *http.Request) {
body := make([]byte, int(r.ContentLength))
b := bytes.NewBuffer(body)
if _, err := b.ReadFrom(r.Body); err != nil {
fmt.Fprintf(w, "%s", err)
}
if _, err := b.WriteTo(w); err != nil {
fmt.Fprintf(w, "%s", err)
}
}
func main() {
http.HandleFunc("/", handler)
if err := http.ListenAndServe(":8080", nil); err != nil {
panic(err)
}
}
$ curl --data "param1=value1¶m2=value2" http://localhost:8080
возвращает:
param1 = значение1 & param2 = значение2
Вы всегда можете добавлять любые данные, которые вы хотели бы body
, или читать больше байтов в буфер из других источников, прежде чем все это снова записать.