Optimise audit task

Author: lucko

src/main/java/me/lucko/bytebin/content/storage/AuditTask.java

@@ -61,7 +61,7 @@ private void run0() throws Exception {
             String backendId = backend.getBackendId();
 
             LOGGER.info("[AUDIT] Listing content for backend {}", backendId);
-            List<String> keys = backend.list().map(Content::getKey).toList();
+            List<String> keys = backend.listKeys().toList();
             LOGGER.info("[AUDIT] Found {} entries for backend {}", keys.size(), backendId);
 
             List<String> keysToDelete = keys.stream()

src/main/java/me/lucko/bytebin/content/storage/LocalDiskBackend.java

@@ -97,6 +97,11 @@ public void save(Content c) throws IOException {
         }
     }
 
+    @Override
+    public Stream<String> listKeys() throws Exception {
+        return Files.list(this.contentPath).map(path -> path.getFileName().toString());
+    }
+
     @Override
     public Stream<Content> list() throws IOException {
         return Files.list(this.contentPath)

src/main/java/me/lucko/bytebin/content/storage/S3Backend.java

@@ -41,6 +41,7 @@
 import software.amazon.awssdk.services.s3.model.ListObjectsV2Request;
 import software.amazon.awssdk.services.s3.model.NoSuchKeyException;
 import software.amazon.awssdk.services.s3.model.PutObjectRequest;
+import software.amazon.awssdk.services.s3.model.S3Object;
 import software.amazon.awssdk.services.s3.paginators.ListObjectsV2Iterable;
 
 import java.util.Date;
@@ -111,6 +112,15 @@ public void delete(String key) throws Exception {
         );
     }
 
+    @Override
+    public Stream<String> listKeys() throws Exception {
+        ListObjectsV2Iterable iter = this.client.listObjectsV2Paginator(ListObjectsV2Request.builder()
+                .bucket(this.bucketName)
+                .build()
+        );
+        return iter.stream().flatMap(resp -> resp.contents().stream().map(S3Object::key));
+    }
+
     @Override
     public Stream<Content> list() throws Exception {
         ListObjectsV2Iterable iter = this.client.listObjectsV2Paginator(ListObjectsV2Request.builder()

src/main/java/me/lucko/bytebin/content/storage/StorageBackend.java

@@ -66,6 +66,14 @@ public interface StorageBackend {
      */
     void delete(String key) throws Exception;
 
+    /**
+     * Lists the keys for all content stored in the backend.
+     *
+     * @return a list of keys
+     * @throws Exception catch all
+     */
+    Stream<String> listKeys() throws Exception;
+
     /**
      * Lists metadata about all the content stored in the backend. (doesn't load the actual data).
      * Used primarily if the index needs to be rebuilt.