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

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

go 复制代码
func Copy(dst Writer, src Reader) (written int64, err error)

io.Copy() 的核心价值在于它提供了一个高效、简洁、通用 的方式来处理数据流的复制,而无需关心具体的数据类型或底层实现,只要它们实现了 io.Readerio.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.Bufferstrings.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)
}()
相关推荐
java坤坤12 分钟前
GoLand 项目从 0 到 1:第四天 —— 技术选型落地与方案设计
golang·jwt
liulanba16 分钟前
八股取士-go
golang
CodeUp.19 分钟前
基于SpringBoot的OA办公系统的设计与实现
spring boot·后端·mybatis
小醉你真好24 分钟前
Spring Boot + ShardingSphere 分库分表实战
java·spring boot·后端·mysql
YY_TJJ1 小时前
8.4 Java Web(Maven P50-P57)
java·开发语言·maven
Jacob02341 小时前
Node.js 性能瓶颈与 Rust + WebAssembly 实战探索
后端·rust·node.js
王中阳Go1 小时前
分库分表之后如何使用?面试可以参考这些话术
后端·面试
知其然亦知其所以然1 小时前
ChatGPT太贵?教你用Spring AI在本地白嫖聊天模型!
后端·spring·ai编程
kinlon.liu2 小时前
内网穿透 FRP 配置指南
后端·frp·内网穿透
kfyty7252 小时前
loveqq-mvc 再进化,又一款分布式网关框架可用
java·后端