Go语言如何用io.Reader_Go语言io.Reader接口教程【详解】

在Go语言里处理IO流，`io.Reader`是绕不过去的核心接口。很多新手容易在这里踩坑，尤其是`Read`方法的边界情况和错误处理逻辑。说实话，在很多场景下，用手写`for`循环去读`io.Reader`，其实有点杀鸡用牛刀了。90%的情况下，`io.ReadAll`和`io.Copy`这两个工具就已经足够应对。手动循环，更多是留给那些需要流控、限速或逐帧解析的特殊场景来用的。

应优先使用 io.ReadAll 或 io.Copy：小数据一次性读取用 io.ReadAll，大数据流式传输用 io.Copy；手动 for 循环仅适用于流控、限速等特殊场景，且必须先处理 buf[:n] 再判 err。

### 为什么不能只看 `err` 就退出循环？ 这个问题是很多人的“噩梦”。核心原因在于，`io.Reader.Read` 方法不保证每调用一次，就把你传入的切片填满。它可能只读了一个字节，就返回 `n = 1, err = nil` —— 这完全符合接口规范，是合法的。一个常见的错误写法是： ```go for { n, err := r.Read(buf) if err != nil { // 错！这里会丢掉 buf[:n] 的数据 break } // 处理 buf，但没用 buf[:n] } ``` 这段代码的问题在于，一旦 `Read` 返回 `io.EOF`，它会直接跳出循环，而之前读到的最后一块数据（`buf[:n]`）就被扔掉了。正确的顺序永远是： * **先处理** `buf[:n]`。只要 `n > 0`，就立刻处理这些数据。 * **再根据** `err` 判断后续逻辑。`err == io.EOF` 表示正常读取结束；`err != nil && n > 0` 是“读了一半出错”，通常需要保留已读数据，并决定如何处理这个错误。 * 另外，`n == 0 && err == nil` 是一个非法状态，标准库的实现不会返回这个组合。 ### `io.ReadAll` 和 `io.Copy` 怎么选？ 这两个工具内部都帮你处理好了 `io.EOF`、分块读取和错误传播，完全不用你操心那些边界逻辑。选择的标准很简单： * **想一次性拿到全部内容**（比如一个JSON响应、一个小配置文件），就用 `io.ReadAll(r)`。它会返回一个 `[]byte`，你可以直接通过 `string(data)` 转成字符串。适合小数据。 * **想把数据从一个流倒进另一个流**（比如HTTP body转发、文件拷贝），就用 `io.Copy(dst, src)`。它内部会维护一个32KB大小的缓冲区，内存占用是恒定的，所以非常适合大数据量处理。 * 需要特别注意：`io.ReadAll` 会把整个流加载到内存里，如果文件很大（比如超过10MB），要谨慎使用，否则可能撑爆内存。`io.Copy` 则没有这个问题，它是一边读一边写，流式处理。 ### HTTP Body 读完不 `Close()` 会怎样？ `http.Response.Body` 是一个一次性的 `io.Reader`，底层是网络连接。 * **读完不调用 `resp.Body.Close()`**，这会直接导致连接无法复用。Go的HTTP客户端有连接池，如果连接不释放，连接池会慢慢被耗尽，后续的请求会越来越慢，甚至直接超时。 * **想多次读Body**（比如先看个header，再解析body），必须自己缓存。做法是：先用 `data, _ := io.ReadAll(resp.Body)` 把数据全部读出来，然后用 `bytes.NewReader(data)` 或 `bytes.NewBuffer(data)` 构造一个新的 `Reader`。别指望 `strings.NewReader("")` 能重读——它第一次调用 `Read` 就会返回 `n = 0, err = io.EOF`。 这里有个很容易被忽略的坑：当你自己实现一个自定义的 `io.Reader` 时，`Read` 方法必须是快速返回的。不能在 `Read` 里做耗时操作，比如解密、正则匹配、`time.Sleep` 这些。否则，下游如 `json.Decoder` 会误判为IO卡顿，触发超时或重试机制，导致整个系统行为异常。