正确使用 Go 的多部分阅读功能

本篇文章向大家介绍《正确使用 Go 的多部分阅读功能》，主要包括，具有一定的参考价值，需要的朋友可以参考一下。

我一直在尝试使用 multipart.part 来帮助从 http 读取非常大的文件上传 (>20gb) - 所以我编写了下面的代码，看起来效果很好：

func ReceiveMultipartRoute(w http.ResponseWriter, r *http.Request) {
    mediatype, p, err := mime.ParseMediaType(r.Header.Get("Content-Type"))
    if err != nil {
       //...
    }
    if mediatype != "multipart/form-data" {
        //...
    }
    boundary := p["boundary"]
    reader := multipart.NewReader(r.Body, boundary)
    buffer := make([]byte, 8192)
    for {
        part, err := reader.NextPart()
        if err != nil {
            // ... 
        }
    
        f, err := os.CreateTemp("", part.FileName())
        if err != nil {
            // ...
        }
    
        for {
            numBytesRead, err := part.Read(buffer)
            // People say not to read if there's an err, but then I miss the last chunk?
            f.Write(buffer[:numBytesRead])
            if err != nil {
                if err == io.EOF {
                    break
                } else {
                    // error, abort ...
                    return
                }
            }
        }
    }
}

但是，在最里面的 for 循环中，我发现我必须在检查 eof 之前从 part.read 读取，因为我注意到如果我事先这样做并中断，我会错过最后一个块。但是，我注意到在许多其他文章/帖子中，人们检查错误/eof，并检查 break-ing（如果有）而不使用上次读取的内容。我是否错误/安全地使用了 multipart.part.read() ？

正确答案

您以正确的方式使用multipart.part。

multipart.part 是一个 特定 实现 go.dev/io#reader" rel="nofollow noreferrer">io.reader。因此，您应该遵循约定并遵循 io.reader 的建议。引用自文档：

调用者应始终在考虑错误 err 之前处理返回的 n > 0 字节。这样做可以正确处理读取一些字节后发生的 i/o 错误以及允许的 eof 行为。

另请注意，在示例中，您将数据从 io.reader 复制到 os.file。 os.file 实现 io.readerfrom 接口，因此您可以使用 file.readfrom() 方法复制数据。 p>

_, err := file.readfrom(part)
// non io.eof
if err != nil {
    return fmt.errorf("copy data: %w", err)
}

如果需要使用缓冲区，可以使用io.copybuffer()函数。但请注意，您需要隐藏 io.readerfrom 实现，否则缓冲区 将不会使用 执行复制。请参阅示例：1、2，3。

_, err := io.CopyBuffer(writeFunc(file.Write), part, buffer)
// non io.EOF
if err != nil {
    return fmt.Errorf("copy data: %w", err)
}

type writeFunc func([]byte) (int, error)

func (write writeFunc) Write(data []byte) (int, error) {
        return write(data)
}

理论要掌握，实操不能落！以上关于《正确使用 Go 的多部分阅读功能》的详细介绍，大家都掌握了吧！如果想要继续提升自己的能力，那么就来关注golang学习网公众号吧！