Golang中io.Reader和io.Writer使用教程

io.Reader和io.Writer是Go语言核心I/O接口，分别通过Read和Write方法实现统一的数据读写，广泛用于文件、网络、内存等场景，支持高复用与解耦设计。

在Go语言中，io.Reader 和 io.Writer 是最核心的I/O接口。它们提供了统一的方式来处理各种数据源的读写操作，比如文件、网络、内存缓冲等。掌握这两个接口的使用，是编写高效、可复用Go代码的关键。

理解 io.Reader 和 io.Writer

io.Reader 接口定义了一个方法 Read(p []byte)，它从数据源读取数据并填充字节切片 p。返回读取的字节数和可能的错误（如 io.EOF 表示读取结束）。

io.Writer 接口定义了一个方法 Write(p []byte)，它将字节切片 p 中的数据写入目标位置。返回成功写入的字节数和错误。

这两个接口屏蔽了底层实现的复杂性，让上层代码可以一致地处理不同类型的输入输出。

常见使用场景与示例

实际开发中，很多类型都实现了这两个接口，可以直接配合使用。

1. 文件读写

os.File 同时实现了 io.Reader 和 io.Writer：

file, err := os.Open("input.txt")
if err != nil {
    log.Fatal(err)
}
defer file.Close()

writer, err := os.Create("output.txt")
if err != nil {
    log.Fatal(err)
}
defer writer.Close()

buf := make([]byte, 1024)
for {
    n, err := file.Read(buf)
    if err == io.EOF {
        break
    }
    if err != nil {
        log.Fatal(err)
    }
    _, err = writer.Write(buf[:n])
    if err != nil {
        log.Fatal(err)
    }
}

2. 使用 io.Copy 简化操作

标准库提供了 io.Copy(dst io.Writer, src io.Reader)，自动完成从Reader到Writer的复制：

inputFile, _ := os.Open("data.txt")
outputFile, _ := os.Create("copy.txt")
defer inputFile.Close()
defer outputFile.Close()

_, err := io.Copy(outputFile, inputFile)
if err != nil {
    log.Fatal(err)
}

这比手动循环读写更简洁，且内部使用了优化的缓冲机制。

3. 内存中的读写：bytes.Buffer

bytes.Buffer 是一个可变字节缓冲区，同时实现了 io.Reader 和 io.Writer，常用于内存中拼接或解析数据：

var buf bytes.Buffer
buf.WriteString("Hello, ")
buf.WriteString("World!")

var dest bytes.Buffer
io.Copy(&dest, &buf)
fmt.Println(dest.String()) // 输出: Hello, World!

构建可组合的程序

利用接口抽象，可以写出高度解耦的代码。例如，编写一个函数只依赖 io.Reader，就能处理文件、网络请求、压缩流等多种输入。

func process(r io.Reader) error {
    scanner := bufio.NewScanner(r)
    for scanner.Scan() {
        fmt.Println("Line:", scanner.Text())
    }
    return scanner.Err()
}

这个函数可以传入 *os.File、*bytes.Buffer，甚至是 http.Response.Body，无需修改。

同样，接收 io.Writer 的函数可以灵活输出到不同目标：

func writeTo(w io.Writer) {
    fmt.Fprintln(w, "Logging message")
}

可用于写文件、标准输出、网络连接等。

处理错误与边界情况

使用 io.Reader 时要注意判断返回的错误。只有当 err == io.EOF 且 n == 0 时才表示正常结束。如果 n > 0 时出现 err，说明部分数据已读取，应先处理这些数据再响应错误。

对于 io.Writer，要检查返回的字节数是否与预期一致，特别是在网络或磁盘满等异常情况下。

基本上就这些。熟练运用 io.Reader 和 io.Writer，能让代码更清晰、更通用，也更容易测试（比如用 strings.NewReader 构造测试输入）。Go标准库大量使用这种“小接口+组合”的设计思想，理解它是提升Go编程能力的重要一步。