巧用 Go 语言流式处理，高效完成 HTTP 大数据请求

今天我们来聊聊 Go 语言的流式处理特性。在开发中，我们常常需要处理体量巨大的数据，比如大文件上传或逐步生成的内容。这时，流式处理便成为了我们的好帮手。

流式处理的魔力：什么是流式写入？

在 Go 中，HTTP 请求的 body 通常需要将数据写入。当数据量不大时，我们可以一次性发送，但对于几个 G 的文件或动态生成的数据，直接加载到内存会导致服务崩溃。这时，流式写入的概念就显得尤为重要。

流式写入是一种边生成边发送数据的方式，Go 的 HTTP 请求体支持 io.Reader 接口，这意味着可以逐步将数据“喂”给 HTTP 请求，而不是一次性加载。

示例代码：流式写入的实战操作

以下是一个流式写入的简单示例，我们用一个大字符串来模拟大数据的上传：

package main

import (
  "bytes"
  "fmt"
  "io"
  "net/http"
  "strings"
)

func main() {
  largeData := strings.Repeat("Hello, World!", 1000000) // 大约13MB
  reader := strings.NewReader(largeData)

  req, err := http.NewRequest("POST", "http://example.com/upload", reader)
  if err != nil {
      fmt.Println("Error creating request:", err)
      return
  }

  req.Header.Set("Content-Type", "text/plain")

  client := &http.Client{}
  resp, err := client.Do(req)
  if err != nil {
      fmt.Println("Error sending request:", err)
      return
  }
  defer resp.Body.Close()

  body, err := io.ReadAll(resp.Body)
  if err != nil {
      fmt.Println("Error reading response body:", err)
      return
  }

  fmt.Println("Response status:", resp.Status)
  fmt.Println("Response body:", string(body))
}

这段代码展示了流式写入的强大之处。通过 strings.NewReader 将大字符串转换为 io.Reader，数据在请求发出时逐步读取并发送，无需一次性加载。

使用流式处理上传大文件

在实际开发中，我们更常需要上传大文件，以下是如何通过 io.Reader 从文件中读取数据并上传：

package main

import (
  "fmt"
  "io"
  "net/http"
  "os"
)

func main() {
  file, err := os.Open("largefile.txt")
  if err != nil {
      fmt.Println("Error opening file:", err)
      return
  }
  defer file.Close()

  req, err := http.NewRequest("POST", "http://example.com/upload", file)
  if err != nil {
      fmt.Println("Error creating request:", err)
      return
  }

  req.Header.Set("Content-Type", "text/plain")

  client := &http.Client{}
  resp, err := client.Do(req)
  if err != nil {
      fmt.Println("Error sending request:", err)
      return
  }
  defer resp.Body.Close()

  body, err := io.ReadAll(resp.Body)
  if err != nil {
      fmt.Println("Error reading response body:", err)
      return
  }

  fmt.Println("Response status:", resp.Status)
  fmt.Println("Response body:", string(body))
}

在这个例子中，我们直接从文件读取数据并作为 HTTP 请求的 body，这种方式避免了一次性加载文件到内存，特别适合大文件上传。