前言

• 最近在学习并使用 go 语言进行一些项目的开发，也测试了一下 http + 内置的 multipart.Writer 来实现 POST 上传文件到我以前的一个 Java 版本的文件服务器上.
• 网上的文章一般都很简单, 按照其规范编写了如下上传多个文件的测试代码:

    for _, uf := range uploadFiles {
        fileInfo, _ := os.Stat(uf)
        file, _ := os.Open(uf)
        fileWrite, _ := mpWrite.CreateFormFile("file", filepath.Base(uf))
        writeLength, _ := io.Copy(fileWrite, file)
        assert.Equal(t, fileInfo.Size(), writeLength, "拷贝文件的内容到内存缓冲区中")
        _ = file.Close()
    }

• 从语法 io.Copy(fileWrite, file) 来看, 需要把文件的所有内容都拷贝到 bytes.Buffer 实例中, 在上传大文件时必然占用大量内存. 实测验证果然, 占用的内存量为 2倍文件大小.
• 网上搜索了一下, 发现大家也都知道这个问题, 但解决方案一般都是 分成小文件块上传然后在服务端合并。这种方案也有一些问题：
  • 无法兼容老的只支持单文件上传的 Server
  • 上传时也要占用 “小文件块” 那么大的内存,而不像 C++/Java 等其他语言的小内存消耗.
  • 在 google, stackoverflow, chatgpt, github 等各个地方, 至今都没有找到任何类似 C++/Java 的小内存消耗的 multi part 解决方案, 如果有人知道的话, 也可以告知.

解决方案

• 参考go 源码中的 multipart.Writer , 重写一个 VirtualWriter , 实现按需读取文件内容并发送，从而可以支持超过 4G 的文件上传。
• 实现的功能:
  • 接口尽量参考 multipart.Writer, 支持多个 field, file ;
  • CreateFormFile 时不需要将文件内容都读入内存，而是保存信息，在 Read 时按需读取;
  • 增加 onProgressCallback 回调, 从而可以知道文件上传进度( 这个功能似乎不容易合并到 go sdk 中 )

源码

https://github.com/fishjam/go-library/blob/main/mime/multipart/virtual_writer.go
国内镜像: https://gitee.com/fishjam/go-library

UT(演示上传 4G+ 文件)

https://github.com/fishjam/go-library/blob/main/mime/multipart/virtual_writer_test.go

遗留问题

• 目前已经给 golang/go 提了 issue(#65203), 希望之后能合并到sdk中,从而方便大家的使用.