From 84cb9807844cb97053bd6a693ab4fc6e268e0ea2 Mon Sep 17 00:00:00 2001
From: Harsh <6162866+harsh62@users.noreply.github.com>
Date: Thu, 3 Apr 2025 12:37:49 -0400
Subject: [PATCH] fix(storage): clean up cached files upon completion (#3967)

---
 .../Support/Internal/StorageMultipartUploadClient.swift       | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/AmplifyPlugins/Storage/Sources/AWSS3StoragePlugin/Support/Internal/StorageMultipartUploadClient.swift b/AmplifyPlugins/Storage/Sources/AWSS3StoragePlugin/Support/Internal/StorageMultipartUploadClient.swift
index ac1baacaa6..b6d547de0a 100644
--- a/AmplifyPlugins/Storage/Sources/AWSS3StoragePlugin/Support/Internal/StorageMultipartUploadClient.swift
+++ b/AmplifyPlugins/Storage/Sources/AWSS3StoragePlugin/Support/Internal/StorageMultipartUploadClient.swift
@@ -146,6 +146,10 @@ class DefaultStorageMultipartUploadClient: StorageMultipartUploadClient {
             Task {
                 do {
                     let partialFileURL = try result.get()
+                    defer {
+                        // Clean up the temporary file after we're done with it
+                        self.fileSystem.removeFileIfExists(fileURL: partialFileURL)
+                    }
 
                     let operation = AWSS3SigningOperation.uploadPart(partNumber: partNumber, uploadId: uploadId)
                     let preSignedURL = try await serviceProxy.preSignedURLBuilder.getPreSignedURL(