Fixed the chunk ordering of s3 multi-part uploads

Signed-off-by: Navneet Verma <[email protected]>
navneet1v · Jan 1, 2025 · 9cbc493 · 9cbc493
1 parent b4bac5e
commit 9cbc493
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/main/java/org/opensearch/knn/remote/index/s3/S3Client.java b/src/main/java/org/opensearch/knn/remote/index/s3/S3Client.java
@@ -40,6 +40,7 @@
 import java.io.InputStream;
 import java.io.OutputStream;
 import java.util.ArrayList;
+import java.util.Comparator;
 import java.util.List;
 import java.util.concurrent.BlockingQueue;
 import java.util.concurrent.CompletableFuture;
@@ -269,7 +270,7 @@ public long uploadWithProgress(final InputStream inputStream, final String key)
                 // totalBytesUploaded += partData.length;
                 // progressCallback.onProgress(totalBytesUploaded);
             }
-
+            completedParts.sort(Comparator.comparingInt(CompletedPart::partNumber));
             // Complete the multipart upload
             CompletedMultipartUpload completedMultipartUpload = CompletedMultipartUpload.builder().parts(completedParts).build();