Go的io包通过Reader和Writer接口统一处理输入输出,支持文件、网络、内存等数据源;示例展示从字符串读取、写入缓冲区及使用io.Copy高效复制数据,强调流式处理与接口复用。

在Go语言中,io 包是处理输入输出操作的核心工具。它定义了统一的接口如 Reader 和 Writer,使得各种数据源(文件、网络、内存等)可以使用一致的方式进行读写。掌握这些基础能力,对构建高效、可复用的程序至关重要。
理解 io.Reader 与 io.Writer 接口
Go 的 io 包中最核心的是两个接口:
-
io.Reader:定义了
Read(p []byte) (n int, err error)方法,从数据源读取数据到字节切片中。 -
io.Writer:定义了
Write(p []byte) (n int, err error)方法,将字节切片中的数据写入目标位置。
只要一个类型实现了这两个接口中的任意一个,就可以参与通用的 I/O 操作。比如 *os.File、*bytes.Buffer、http.ResponseWriter 都实现了它们。
使用 io.Reader 读取数据
任何实现了 Read 方法的类型都可以用来读取数据。常见场景包括从文件、字符串或网络连接读取。
立即学习“go语言免费学习笔记(深入)”;
示例:从字符串中读取内容
package main
import (
"fmt"
"io"
"strings"
)
func main() {
reader := strings.NewReader("Hello, Golang!")
buf := make([]byte, 8)
for {
n, err := reader.Read(buf)
if err == io.EOF {
break
}
if err != nil {
fmt.Println("读取错误:", err)
return
}
fmt.Printf("读取 %d 字节: %q\n", n, buf[:n])
}
}
这段代码每次最多读取8个字节,直到遇到 io.EOF 结束。注意 Read 不保证一次性读完所有数据,需循环处理。
使用 io.Writer 写入数据
Write 方法用于向目标写入字节数据。常用于写文件、网络响应或内存缓冲区。
示例:写入数据到内存缓冲区
package main
import (
"bytes"
"fmt"
"io"
)
func main() {
var writer bytes.Buffer
data := []byte("学习 Go 的 io 操作")
n, err := writer.Write(data)
if err != nil {
fmt.Println("写入失败:", err)
return
}
fmt.Printf("成功写入 %d 字节\n", n)
fmt.Println("当前缓冲区内容:", writer.String())
}
这里使用 bytes.Buffer 实现了 io.Writer,适合临时拼接或测试输出。
组合使用 io 工具提升效率
Go 提供了许多辅助函数来简化常见操作,避免手动循环读写。
- io.Copy(dst Writer, src Reader):直接将一个 Reader 的数据复制到 Writer,自动处理分块读写。
-
io.ReadAll(reader):一次性读取所有数据,返回
[]byte。 - io.LimitReader(reader, n):限制最多读取 n 字节。
示例:使用 io.Copy 将字符串写入 buffer
package main
import (
"bytes"
"fmt"
"io"
"strings"
)
func main() {
var buf bytes.Buffer
reader := strings.NewReader("这是一段测试文本")
_, err := io.Copy(&buf, reader)
if err != nil {
fmt.Println("复制失败:", err)
return
}
fmt.Println("复制完成,内容为:", buf.String())
}
io.Copy 内部会自动分配缓冲区并循环调用 Read/Write,代码更简洁且性能良好。
基本上就这些。熟练使用 io.Reader 和 io.Writer 能让你写出更通用、解耦更强的代码。无论是处理文件、HTTP 请求还是自定义协议,这套接口都适用。关键是理解“按需读取”和“流式处理”的思想,避免一次性加载大文件导致内存溢出。不复杂但容易忽略。










