go - 如何连接 io.Reader 和 io.Writer？

Question

我正在编写一个长时间运行的任务，它多次从 mongodb(使用 mgo)获取。然后使用 this module 将其写入 xlsx 文件.然后使用 os.Open 再次读取它，然后将其存储到我的 ftp 服务器。

Stor 函数非常消耗我的内存，所以我认为应该有一种方法不保存文件而是将我的数据从 xlsx.Write 直接传递到 ftp.Store。 (如果我可以同时流式传输就完美了，因为我不必在将所有文档发送到 Stor 函数之前将它们保存在服务器的内存中)

这些是函数的原型(prototype)

func (f *File) Write(writer io.Writer) (err 错误) xlsl

func (ftp *FTP) Stor(path string, r io.Reader) (err 错误) ftp

Answer 1

您想使用 io.Pipe .你可以这样做:

reader, writer := io.Pipe()
errChan := make(chan error)
go func() {
    errChan <- myFTP.Stor(path, reader)
}()
err := myXLS.Write(writer)
// handle err
err = <-errChan
// handle err

如果 xlsx.Write 在不关闭编写器的情况下返回错误，您可能想要 writer.CloseWithError(err)。