io.Copy()
是 Go 语言标准库 io
包中一个非常核心和常用的函数。它的基本功能是从一个源(src
)读取数据并写入到一个目标(dst
) ,直到读取到 EOF
或发生错误。
go
func Copy(dst Writer, src Reader) (written int64, err error)
io.Copy()
的核心价值在于它提供了一个高效、简洁、通用 的方式来处理数据流的复制,而无需关心具体的数据类型或底层实现,只要它们实现了 io.Reader
和 io.Writer
接口。
1、文件操作
go
// 复制文件
func copyFile(src, dst string) error {
sourceFile, err := os.Open(src)
if err != nil {
return err
}
defer sourceFile.Close()
destFile, err := os.Create(dst)
if err != nil {
return err
}
defer destFile.Close()
_, err = io.Copy(destFile, sourceFile)
return err
}
这里需要注意的是,如果destFile
不是APPEND
模式,那么第二次调用io.Copy
会覆盖原先的内容。
2. HTTP 请求与响应处理
在 net/http
包中,io.Copy()
非常常见。
将 HTTP 响应体写入文件或另一个 Writer
go
resp, err := http.Get("https://example.com/data")
if err != nil {
// 处理错误
}
defer resp.Body.Close()
// 将响应体直接写入文件
file, _ := os.Create("downloaded_data")
defer file.Close()
io.Copy(file, resp.Body)
// 或者写入 bytes.Buffer
var buf bytes.Buffer
io.Copy(&buf, resp.Body)
在 HTTP 代理或中间件中转发请求/响应体
go
// 在反向代理中,将后端服务的响应原样返回给客户端
resp, err := backendClient.Do(req)
if err != nil {
// 处理错误
}
defer resp.Body.Close()
// 将后端响应头复制到客户端响应
for k, v := range resp.Header {
w.Header()[k] = v
}
w.WriteHeader(resp.StatusCode)
// 使用 io.Copy 将后端响应体流式传输给客户端
io.Copy(w, resp.Body) // w 是 http.ResponseWriter (实现了 io.Writer)
3. 进程间通信 (IPC)
通过管道 (io.Pipe
) 在 goroutine 之间或父子进程间传递数据。
go
// 创建管道
reader, writer := io.Pipe()
// 在一个 goroutine 中写入数据
go func() {
defer writer.Close()
fmt.Fprintln(writer, "Hello from goroutine!")
}()
// 在主 goroutine 中读取并通过 io.Copy 输出到 stdout
io.Copy(os.Stdout, reader)
4. 网络编程
在网络连接(net.Conn
)之间复制数据。
go
// 简单的 TCP 代理/转发
func proxyConn(srcConn, dstConn net.Conn) {
defer srcConn.Close()
defer dstConn.Close()
// 同时双向复制
var wg sync.WaitGroup
wg.Add(2)
go func() {
defer wg.Done()
io.Copy(dstConn, srcConn) // 客户端 -> 服务端
}()
go func() {
defer wg.Done()
io.Copy(srcConn, dstConn) // 服务端 -> 客户端
}()
wg.Wait()
}
5. 数据缓冲与转换
结合 bytes.Buffer
或 strings.Reader
进行内存中的数据操作。
go
// 将字符串内容复制到 Buffer
str := "Hello, World!"
reader := strings.NewReader(str)
var buf bytes.Buffer
io.Copy(&buf, reader)
data := buf.Bytes()
// 将 Buffer 内容复制到另一个 Writer
io.Copy(someWriter, &buf)
6. 压缩与解压缩
与 compress/gzip
等包结合使用。
go
// 压缩文件
func compressFile(inputFile, outputFile string) error {
inFile, _ := os.Open(inputFile)
defer inFile.Close()
outFile, _ := os.Create(outputFile)
defer outFile.Close()
gzipWriter := gzip.NewWriter(outFile)
defer gzipWriter.Close()
// 将原始文件内容通过 gzip 压缩器写入输出文件
_, err := io.Copy(gzipWriter, inFile)
return err
}
7. 日志处理
将程序的标准输出/错误重定向到日志文件。
go
logFile, _ := os.OpenFile("app.log", os.O_CREATE|os.O_WRONLY|os.O_APPEND, 0666)
defer logFile.Close()
// 将标准错误重定向到日志文件 (需要在 goroutine 中运行)
go func() {
io.Copy(logFile, os.Stderr)
}()