Fix: correct limit checks for S3Container refactoring

Signed-off-by: bansvaru <[email protected]>
linuxpi · Sep 1, 2023 · a76be11 · a76be11
1 parent ee470bc
commit a76be11
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/plugins/repository-s3/src/main/java/org/opensearch/repositories/s3/S3BlobContainer.java b/plugins/repository-s3/src/main/java/org/opensearch/repositories/s3/S3BlobContainer.java
@@ -356,10 +356,11 @@ public List<BlobMetadata> listBlobsByPrefixInSortedOrder(String blobNamePrefix,
             }
             String prefix = blobNamePrefix == null ? keyPath : buildKey(blobNamePrefix);
             try (AmazonS3Reference clientReference = blobStore.clientReference()) {
-                return executeListing(clientReference, listObjectsRequest(prefix, limit), limit).stream()
+                List<BlobMetadata> blobs = executeListing(clientReference, listObjectsRequest(prefix, limit), limit).stream()
                     .flatMap(listing -> listing.contents().stream())
                     .map(s3Object -> new PlainBlobMetadata(s3Object.key().substring(keyPath.length()), s3Object.size()))
                     .collect(Collectors.toList());
+                return blobs.subList(0, Math.min(limit, blobs.size()));
             } catch (final Exception e) {
                 throw new IOException("Exception when listing blobs by prefix [" + prefix + "]", e);
             }

diff --git a/server/src/main/java/org/opensearch/common/blobstore/BlobContainer.java b/server/src/main/java/org/opensearch/common/blobstore/BlobContainer.java
@@ -227,6 +227,9 @@ default void listBlobsByPrefixInSortedOrder(
         BlobNameSortOrder blobNameSortOrder,
         ActionListener<List<BlobMetadata>> listener
     ) {
+        if (limit < 0) {
+            throw new IllegalArgumentException("limit should not be a negative value");
+        }
         try {
             listener.onResponse(listBlobsByPrefixInSortedOrder(blobNamePrefix, limit, blobNameSortOrder));
         } catch (Exception e) {