Golang如何使用io.Pipe在协程间传输数据_Golang Pipe协程数据传输实践

package main

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

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

    // 写入协程
    go func() {
        defer w.Close()
        for i := 0; i < 5; i++ {
            _, err := w.Write([]byte("hello world\n"))
            if err != nil {
                log.Printf("写入错误: %v", err)
                return
            }
        }
    }()

    // 读取并输出到标准输出
    _, err := io.Copy(os.Stdout, r)
    if err != nil && err != io.EOF {
        log.Printf("读取错误: %v", err)
    }

    r.Close()
}

package main

import (
    "compress/gzip"
    "io"
    "log"
    "os"
)

func main() {
    pr, pw := io.Pipe()
    gzFile, _ := os.Create("output.gz")

    // 启动压缩协程
    go func() {
        defer pw.Close()
        writer := gzip.NewWriter(pw)
        defer writer.Close()

        for i := 0; i < 1000; i++ {
            _, err := writer.Write([]byte("data line: hello golang\n"))
            if err != nil {
                log.Printf("压缩写入失败: %v", err)
                return
            }
        }
    }()

    // 主协程将管道中的压缩数据写入文件
    _, err := io.Copy(gzFile, pr)
    if err != nil && err != io.EOF {
        log.Printf("文件写入失败: %v", err)
    }

    pr.Close()
    gzFile.Close()
}

go func() {
    _, err := w.Write(data)
    if err != nil {
        w.CloseWithError(err)
        return
    }
    w.Close()
}()