将文件上传到 Google Storage 而不将其保存到内存

jz2*_*z22 4 upload file-upload go google-cloud-storage object-storage

我想从前端直接通过后端将文件上传到 Google Storage 存储桶中,而不是先将其完全保存在服务器的内存中。我添加了一个类似于Google 文档中的示例的端点,它可以工作。但是,我不确定这是否会先将整个文件保存到内存中,因为这可能会导致上传较大文件时出现问题。

如果它首先将文件保存到内存中,我该如何更改代码以便将上传内容直接流式传输到 Google Storage。类似问题的答案并没有澄清我的问题。

谢谢

func Upload(c *gin.Context) {
    file, _, _ := c.Request.FormFile("image")
    ctx := context.Background()

    client, err := storage.NewClient(ctx)
    if err != nil {
        fmt.Printf("Failed to create client with error: %v", err)
        return
    }

    bucket := client.Bucket("test-bucket")

    w := bucket.Object("testfile").NewWriter(ctx)

    w.ContentType = "image/jpeg"

    io.Copy(w, file)
    w.Close()
}
Run Code Online (Sandbox Code Playgroud)

I L*_*ion 5

正如彼得在对问题和答案的评论中所指出的,直接使用多部分阅读器来阅读请求正文。

func Upload(c *gin.Context) {
    mr, err :=  c.Request.MultipartReader()
    if err != nil {
        // handle error
        return
    }
    var foundImage bool
    for {
        p, err := mr.NextPart()
        if err == io.EOF {
            break
        }
        if err != nil {
            // handle error
            return
        }
        if p.FormName() == "image" {
            foundImage = true

            ctx := context.Background()
            client, err := storage.NewClient(ctx)
            if err != nil {
                // handle error
                return
            }
            bucket := client.Bucket("test-bucket")
            w := bucket.Object("testfile").NewWriter(ctx)
            w.ContentType = "image/jpeg"
            if _, err := io.Copy(w, p); err != nil {
              // handle error
              return
            }
            if err := w.Close(); err != nil {
              // handle error
              return
            }
        }
    }
    if !imageFound {
       // handle error
    }
}
Run Code Online (Sandbox Code Playgroud)

// handle error注释替换为以适当的错误状态响应客户端的代码。记录一些错误也可能很有用。