当前位置：首页 > news >正文

Golang中的`io.Copy()`使用场景

news 2025/8/5 12:20:19

io.Copy() 是 Go 语言标准库 io 包中一个非常核心和常用的函数。它的基本功能是从一个源（src）读取数据并写入到一个目标（dst），直到读取到 EOF 或发生错误。

func Copy(dst Writer, src Reader) (written int64, err error)

io.Copy() 的核心价值在于它提供了一个高效、简洁、通用的方式来处理数据流的复制，而无需关心具体的数据类型或底层实现，只要它们实现了 io.Reader 和 io.Writer 接口。

1、文件操作

// 复制文件
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

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 代理或中间件中转发请求/响应体

// 在反向代理中，将后端服务的响应原样返回给客户端
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 之间或父子进程间传递数据。

// 创建管道
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）之间复制数据。

// 简单的 TCP 代理/转发
func proxyConn(srcConn, dstConn net.Conn) {defer srcConn.Close()defer dstConn.Close()// 同时双向复制var wg sync.WaitGroupwg.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 进行内存中的数据操作。

// 将字符串内容复制到 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 等包结合使用。

// 压缩文件
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. 日志处理

将程序的标准输出/错误重定向到日志文件。

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)
}()