Golang如何使用io.Pipe在管道中传输数据-Golang-PHP中文网

Golang如何使用io.Pipe在管道中传输数据

P粉602998670

发布： 2025-10-06 21:54:02

原创

411人浏览过

io.Pipe用于Go中goroutine间同步通信，提供无缓冲的读写端，需在不同协程中使用以避免死锁，写端关闭后读端收到EOF，适用于流式数据传输场景。

golang如何使用io.pipe在管道中传输数据

在Go语言中，io.Pipe 提供了一种在并发场景下通过管道传输数据的机制。它返回一个同步的内存管道（pipe），一端用于写入，另一端用于读取，常用于 goroutine 之间的通信，尤其是在需要满足 io.Reader 和 io.Writer 接口的场景中。

io.Pipe 基本原理

注意：io.Pipe 返回的是一个 *io.PipeReader 和 *io.PipeWriter。它们是线程安全的，但必须在不同的 goroutine 中使用，否则可能造成死锁。

Pipe 的读写操作是阻塞的：当没有数据可读时，读操作会等待；当缓冲区满（虽然 io.Pipe 没有内部缓冲，完全依赖读写协程配合）时，写操作也会等待。

基本用法示例

下面是一个简单的例子，演示如何使用 io.Pipe 在两个 goroutine 之间传输字符串数据：

立即学习“go语言免费学习笔记（深入）”；

代码示例：

Devv

Devv是一个专为程序员打造的新一代AI搜索引擎

140

查看详情

package main

import (
    "io"
    "log"
    "os"
)

func main() {
    // 创建一个 pipe
    r, w := io.Pipe()

    // 启动一个 goroutine 写入数据
    go func() {
        defer w.Close()
        _, err := w.Write([]byte("hello from the other side\n"))
        if err != nil {
            log.Fatal(err)
        }
        _, err = w.Write([]byte("this is the second message\n"))
        if err != nil {
            log.Fatal(err)
        }
    }()

    // 主 goroutine 读取数据并输出到标准输出
    defer r.Close()
    _, err := io.Copy(os.Stdout, r)
    if err != nil && err != io.EOF {
        log.Fatal(err)
    }
}

登录后复制

在这个例子中：

io.Pipe() 创建了一个管道。
一个 goroutine 向 w 写入数据。
主 goroutine 使用 io.Copy 将 r 中的数据复制到 os.Stdout。
写入端关闭后，读取端会收到 EOF，io.Copy 结束。

常见应用场景

io.Pipe 常用于以下场景：

模拟文件或网络流：测试函数接收 io.Reader 时，可以用 pipe 模拟数据输入。
与命令行工具交互：将程序输出通过 pipe 传递给另一个命令的输入。
HTTP 请求体流式生成：在发送大文件时，一边生成数据一边上传，避免内存溢出。

HTTP 流式上传示例：

req, _ := http.NewRequest("POST", "http://example.com/upload", r)
go func() {
    w.Write(dataChunk1)
    w.Write(dataChunk2)
    w.Close()
}()
client.Do(req)

登录后复制