使用 Go 将文件上传到 AWS S3

概述

将大型文件直接上传到 AWS S3，同时最大限度地减少内存和磁盘占用空间是一项重要任务在云计算中。本指南将演示如何使用适用于 Go 的 AWS 开发工具包来实现此目的。

解决方案

要将文件上传直接流式传输到 S3，您可以使用 s3manager 包。以下是分步解决方案：

1. 配置 AWS 凭证和会话：

   • 设置您的 AWS 访问密钥和密钥，或使用默认凭证提供程序。
   • 使用指定的凭证初始化 AWS 会话配置。

2. 创建 S3 上传器：

   • 使用会话和可选配置设置初始化 S3 上传器。
   • 可以配置分片大小、并发数、最大上传数等参数parts.

3. 打开文件：

   • 使用 os.Open 函数打开要上传的文件。

4. 上传文件：

   • 使用上传器。上传方法以及适当的文件信息（存储桶、密钥和文件指针）。

示例代码

以下代码示例演示了如何使用流式传输大文件上传到 AWS S3 s3manager:

package main

import (
    "fmt"
    "os"

    "github.com/aws/aws-sdk-go/aws/credentials"

    "github.com/aws/aws-sdk-go/aws"
    "github.com/aws/aws-sdk-go/aws/session"
    "github.com/aws/aws-sdk-go/service/s3/s3manager"
)

const (
    filename      = "file_name.zip"
    myBucket      = "myBucket"
    myKey         = "file_name.zip"
    accessKey     = ""
    accessSecret  = ""
)

func main() {
    var awsConfig *aws.Config
    if accessKey == "" || accessSecret == "" {
        //load default credentials
        awsConfig = &aws.Config{
            Region: aws.String("us-west-2"),
        }
    } else {
        awsConfig = &aws.Config{
            Region:      aws.String("us-west-2"),
            Credentials: credentials.NewStaticCredentials(accessKey, accessSecret, ""),
        }
    }

    // The session the S3 Uploader will use
    sess := session.Must(session.NewSession(awsConfig))

    // Create an uploader with the session and default options
    //uploader := s3manager.NewUploader(sess)

    // Create an uploader with the session and custom options
    uploader := s3manager.NewUploader(sess, func(u *s3manager.Uploader) {
        u.PartSize = 5 * 1024 * 1024 // The minimum/default allowed part size is 5MB
        u.Concurrency = 2            // default is 5
    })

    //open the file
    f, err := os.Open(filename)
    if err != nil {
        fmt.Printf("failed to open file %q, %v", filename, err)
        return
    }
    //defer f.Close()

    // Upload the file to S3.
    result, err := uploader.Upload(&s3manager.UploadInput{
        Bucket: aws.String(myBucket),
        Key:    aws.String(myKey),
        Body:   f,
    })

    //in case it fails to upload
    if err != nil {
        fmt.Printf("failed to upload file, %v", err)
        return
    }
    fmt.Printf("file uploaded to, %s\n", result.Location)
}

通过执行这些步骤，您可以高效地将大型多部分/表单数据文件直接上传到 AWS S3，同时占用最少的内存。