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

在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语言免费学习笔记(深入)”;
代码示例:
第一步】:将安装包中所有的文件夹和文件用ftp工具以二进制方式上传至服务器空间;(如果您不知如何设置ftp工具的二进制方式,可以查看:(http://www.shopex.cn/support/qa/setup.help.717.html)【第二步】:在浏览器中输入 http://您的商店域名/install 进行安装界面进行安装即可。【第二步】:登录后台,工具箱里恢复数据管理后台是url/sho
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)
注意事项
使用 io.Pipe 时需注意以下几点:
- 读写操作必须在不同 goroutine 中进行,否则会死锁。
- 写入端必须调用
Close(),否则读取端无法得知数据结束,可能持续阻塞。 - 如果写入过程中发生错误,应调用
w.CloseWithError(err),这样读取端会收到同样的错误。 - io.Pipe 不带缓冲,性能依赖于读写速度匹配,必要时可考虑使用
bytes.Buffer或带缓冲的 channel 配合。
基本上就这些。io.Pipe 是处理流式数据传输的轻量级工具,关键在于合理使用 goroutine 配合读写,避免阻塞和死锁。









