io.copy是go中高效的数据流拷贝方式,内置32KB缓冲优化,适用于文件、网络、内存等Reader到Writer的复制,返回字节数和错误;需自定义缓冲时可用io.CopyBuffer控制大小,建议8KB–1MB;注意接口实现、资源关闭及context中断控制;高性能场景可选os.CopyFile或优化网络配置,避免小块循环拷贝。

使用 io.Copy 是 Go 中最简洁、高效的数据流拷贝方式,它内部已做缓冲优化,无需手动管理 buffer,适合大多数场景。
io.Copy 的基本用法
io.Copy 从源 io.Reader 读取数据,写入目标 io.Writer,直到读到 EOF 或发生错误。它返回实际复制的字节数和第一个遇到的错误。
- 函数签名:
func Copy(dst Writer, src Reader) (written int64, err Error) - 常见组合:文件到文件、网络连接到文件、内存 buffer 到 http 响应等
- 示例:将文件内容复制到标准输出
file, _ := os.Open(“input.txt”)
defer file.Close()
io.Copy(os.Stdout, file)
何时需要自定义缓冲区(io.CopyBuffer)
io.Copy 默认使用 32KB 缓冲区,对小量数据或受限内存环境可能不够灵活。此时可用 io.CopyBuffer 显式控制缓冲区大小。
- 适用于:嵌入式设备、流式处理高吞吐场景、避免大内存分配
- 缓冲区太小会增加系统调用次数;太大可能浪费内存或延迟响应
- 建议值:8KB–1MB,根据 I/O 类型和数据特征调整
buf := make([]byte, 64*1024) // 64KB
io.CopyBuffer(dst, src, buf)
避免常见陷阱
看似简单,但几个细节容易引发问题:
立即学习“go语言免费学习笔记(深入)”;
- 确保
src和dst都实现了对应接口,比如*os.File同时满足Reader和Writer,但strings.Reader只读,不能当dst - 复制后记得关闭资源(尤其是文件、网络连接),
io.Copy不负责关闭 - 若需精确控制中断(如超时、取消),应配合
context.Context使用io.CopyN或封装带 cancel 的 reader/writer
性能优化建议
- 对本地文件间拷贝,优先考虑
os.CopyFile(Go 1.19+),它可能调用操作系统零拷贝接口(如 linux 的copy_file_range) - 网络传输中,如果两端都支持,启用 TCP 的
SO_NOSIGPIPE和合理设置SetWriteDeadline可减少阻塞 - 避免在循环中反复调用
io.Copy小块数据;合并为一次大拷贝更高效
基本上就这些。io.Copy 设计克制而实用,理解它的边界和替代方案,比盲目优化更重要。