AWS SDK for iOS 中 S3 多部分上传回调丢失问题的分析与解决

在 iOS 应用开发中使用 AWS SDK 进行大文件上传时，开发者经常会遇到需要实现断点续传功能的需求。AWS SDK for iOS 提供了 AWSS3TransferUtility 类来处理文件上传任务，特别是通过 multipartUpload 方法支持大文件分块上传。然而，在 iOS 18 环境下，开发者发现了一个关键问题：当应用被终止后重新启动时，虽然上传任务能够自动恢复并完成，但进度回调和完成回调却无法正常触发。

问题现象

开发者在使用 AWSS3TransferUtility 进行大文件（如 10GB）上传时，按照官方推荐的方式实现了断点续传功能。具体实现是通过调用 transferUtility.enumerateToAssign(blocks:) 方法来重新关联上传任务的回调块。在 iOS 17.5.1 及之前版本中，这一方案工作正常，但当升级到 iOS 18 后，发现虽然上传任务能够成功恢复并完成，但 multiPartUploadProgress 和 multiPartUploadCompleted 这两个关键回调却不再被触发。

问题根源

经过深入分析，发现问题出在回调块的注册时机上。在 iOS 18 中，系统对后台任务的处理机制有所调整，导致回调块的注册与任务恢复之间存在竞态条件。具体来说：

1. 当应用重新启动时，AWSS3TransferUtility 需要先完成对之前未完成上传任务的恢复和重新链接
2. 开发者通常在应用启动时就立即调用 enumerateToAssign 方法注册回调
3. 在 iOS 18 中，任务恢复过程可能尚未完成时回调就已经注册，导致回调无法正确关联到恢复的任务上

解决方案

正确的做法是确保在 AWSS3TransferUtility 完全初始化并恢复所有任务后，再注册回调块。具体实现步骤如下：

1. 使用带 completionHandler 的注册方法初始化 AWSS3TransferUtility
2. 在 completionHandler 中确保所有任务都已恢复
3. 然后再调用 enumerateToAssign 方法注册回调块

示例代码如下：

let transferUtilityKey = "your_unique_key"
let configuration = AWSS3TransferUtilityConfiguration()

AWSS3TransferUtility.register(
    with: configuration,
    forKey: transferUtilityKey
) { error in
    guard error == nil else {
        print("Transfer Utility 注册失败: \(error!)")
        return
    }
    
    guard let transferUtility = AWSS3TransferUtility.s3TransferUtility(forKey: transferUtilityKey) else {
        print("无法获取 Transfer Utility 实例")
        return
    }

    let blocks = AWSS3TransferUtilityBlocks(
        uploadProgress: nil,
        multiPartUploadProgress: { task, progress in
            print("上传进度: \(progress.fractionCompleted)")
        },
        downloadProgress: nil,
        uploadCompleted: nil,
        multiPartUploadCompleted: { task, error in
            if let error = error {
                print("上传失败: \(error)")
            } else {
                print("上传成功完成")
            }
        },
        downloadCompleted: nil
    )
    
    transferUtility.enumerateToAssign(blocks: blocks)
}

最佳实践建议

1. 唯一标识符：始终为每个 TransferUtility 实例使用唯一的 key，特别是在需要管理多个上传会话时
2. 错误处理：完善注册过程和回调中的错误处理逻辑
3. 进度更新：在主线程更新 UI，避免在回调中直接操作界面
4. 任务管理：对于特别大的文件，考虑实现额外的本地状态管理机制作为备份
5. 兼容性考虑：虽然问题在 iOS 18 上表现明显，但建议在所有版本 iOS 上都采用这种安全的回调注册方式

总结

通过调整回调注册时机，开发者可以确保在 iOS 18 及更早版本上都能可靠地接收到上传进度和完成通知。这一解决方案不仅修复了回调丢失的问题，还提供了更健壮的任务恢复机制，特别适合需要处理大文件上传的 iOS 应用场景。

对于使用 Amplify Storage 的开发者，需要注意的是 Amplify 框架目前对应用终止后的自动恢复支持有限，如果应用被系统终止，上传任务可能无法自动恢复。在这种情况下，开发者需要考虑实现自己的任务持久化和恢复逻辑，或者直接使用底层的 AWSS3TransferUtility 来实现更精细的控制。