1

我正在开发一个流式 XML 编码器，它将同时将 XML 写入本地文件和 S3 存储桶。但是，仅通过测试它写入两个本地文件，我可以看到其中一个文件每次都缺少结束标记。

这大致是我的做法（省略错误处理）：

func copyToFile (fileName string) {
    f, _ := os.Create(fileName)
    defer f.Close()
    io.Copy(f, pr)
}

func main () {
    pr, pw := io.Pipe()
    defer pw.Close()

    encoder := xml.NewEncoder(pw)

    go copyToFile("file1.xml")
    go copyToFile("file2.xml")

    encoder.EncodeToken(xml.StartElement{...})
    encoder.Encode(SomeStruct{})
    encoder.EncodeToken(xml.EndElement{...})
    encoder.Flush()
}

结果file1.xml符合预期，所有标签都正确关闭，但file2.xml结束标签（的调用encoder.EncodeToken(xml.EndElement{...})）缺失。

我究竟做错了什么？当我将阅读器复制到 S3 时，我可以期待相同的结果吗？

4

1 回答 1

1

返回的数据不能有多个阅读器io.PipeReader，所有阅读器的数据都不会重复。io.PipeReader只能“服务”一个阅读器，并且您启动 2 个 goroutine 来读取它。

要实现您想要的，请使用io.MultiWriter(). 它会将您返回io.Writer到您可以写入的位置，并将写入复制到您传递给它的所有写入器。

例如：

f1 := &bytes.Buffer{}
f2 := &bytes.Buffer{}

w := io.MultiWriter(f1, f2)

encoder := xml.NewEncoder(w)

encoder.EncodeToken(xml.StartElement{Name: xml.Name{Local: "test"}})
encoder.Encode(image.Point{1, 2})
encoder.EncodeToken(xml.EndElement{Name: xml.Name{Local: "test"}})
encoder.Flush()

fmt.Println(f1)
fmt.Println(f2)

这将输出（在Go Playground上尝试）：

<test><Point><X>1</X><Y>2</Y></Point></test>
<test><Point><X>1</X><Y>2</Y></Point></test>

上面的示例写入 2 个内存缓冲区。要写入 2 个文件，您可以将 2 os.Files传递给io.MultiWriter()（或任何其他实现io.Writer）：

f1, err := os.Create("file1.xml")
if err != nil {
    panic(err)
}
defer f1.Close()

f2, err := os.Create("file2.xml")
if err != nil {
    panic(err)
}
defer f2.Close()

w := io.MultiWriter(f1, f2)

// ...
于 2019-11-26T09:50:36.417 回答