如何等待 S3 上传

在开发过程中，我们常常需要将文件上传到云存储服务中，如 Amazon S3，但我们也经常会遇到等待上传完成的情况。下面是介绍如何等待 S3 上传的方法。

定义上传函数

首先，我们需要定义一个用于上传文件到 S3 的函数，以 Python 为例：

import boto3

def upload_to_s3(file_path, bucket_name, key):
    s3_client = boto3.client('s3')
    with open(file_path, 'rb') as file:
        s3_client.upload_fileobj(file, bucket_name, key)

使用 S3 Transfer 对象

然后，我们可以使用 S3 Transfer 对象，该对象可使您轻松完成在 S3 中上传和下载文件的单个文件传输。我们可以创建一个 Transfer 对象，并使用该对象调用 upload() 方法来上传文件，该方法默认与 Amazon S3 保持连接并等待传输完成：

import boto3
from boto3.s3.transfer import TransferConfig

def wait_s3_upload(file_path, bucket_name, key):
    s3_client = boto3.client('s3')
    config = TransferConfig(multipart_threshold=1024 * 25, max_concurrency=10,
                            multipart_chunksize=1024 * 25, use_threads=True)
    transfer = boto3.s3.transfer.S3Transfer(s3_client, config)
    transfer.upload(file_path, bucket_name, key)

请注意，TransferConfig 定义了 multipart_threshold、multipart_chunksize 和 max_concurrency，这些配置参数可使 S3 Transfer 使用多部分上传，增加代码的健壮性，并在上传大型文件时提高上传速度。

等待上传完成

要等待 S3 上传完成，我们可以使用 wait() 方法，该方法可使文件在上传完成前自动等待：

import boto3
from boto3.s3.transfer import TransferConfig

def wait_s3_upload(file_path, bucket_name, key):
    s3_client = boto3.client('s3')
    config = TransferConfig(multipart_threshold=1024 * 25, max_concurrency=10,
                            multipart_chunksize=1024 * 25, use_threads=True)
    transfer = boto3.s3.transfer.S3Transfer(s3_client, config)
    transfer.upload(file_path, bucket_name, key)
    transfer.wait()

现在，我们已经了解了如何使您的代码等待上传完成。欢迎试用！