web-dev-qa-db-ja.com

Golangでバッファリングされていない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")
}

クライアントの観点からは、2つの回線が同時に送信されます。どんな提案も大歓迎です:)

@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のメーリングリストでヘルプを見つけました。これを実現する2つの方法があります。 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

httpstreamgoflush
36
rmonjo

ドキュメント で暗示されているように、一部の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")
}

ネットワークまたはクライアント側の他の多くの場所でバッファリングが発生する可能性があることに注意してください。

29
Denys Séguret

あなたの質問を誤解した場合は申し訳ありませんが、以下のようなことがうまくいきますか？

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&param2=value2" http://localhost:8080

戻り値：

param1 = value1＆param2 = value2

すべてのデータを書き出す前に、bodyに必要なデータを常に追加するか、他の場所からバッファにさらにバイトを読み取ることができます。

1
Intermernet