Golang で HTTP 応答をストリーミングし、デフォルトのバッファリングを回避するにはどうすればよいですか?-Golang-php.cn

Golang で HTTP 応答をストリーミングし、デフォルトのバッファリングを回避するにはどうすればよいですか?

Susan Sarandon

リリース： 2024-12-30 05:07:13

オリジナル

622 人が閲覧しました

How Can I Stream HTTP Responses in Golang and Avoid Default Buffering?

Golang での HTTP 応答のストリーミング: デフォルトのバッファリングのバイパス

このガイドでは、Go の HTTP 応答処理におけるバッファリングの問題に取り組みます。通常、応答はバッファリングされ、ブロックで送信されます。リクエストが処理された後のクライアント。ストリーミングシナリオの場合、これは望ましくない可能性があります。

提供されたコードスニペットは、2 行のデータがストリーミングされるはずであるが、バッファリングのために同時に送信されるという問題を示しています。

実装フラッシュ

これに対処する 1 つのアプローチは、応答バッファを明示的にフラッシュすることです。 Go の ResponseWriter は Flush メソッドを提供しますが、その可用性は特定の実装によって異なります。

たとえば、次の変更されたコードにはフラッシュが組み込まれています。

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()
  }
  sleep(10) //not real code
  fmt.Fprintf(res, "sending second line of data")
}

ログイン後にコピー

外部出力のパイプ処理

外部コマンドの出力のストリーミングなど、特定のシナリオでは、直接フラッシュでは不十分な場合があります。 Go のパイプメカニズムにより、コマンドの出力と応答ライター間の直接ストリーミングが可能になります。

実装例は次のとおりです:

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
    }
  }
}

ログイン後にコピー