From 3a808a2fe11275ebee26ebe953528ca32fc1957e Mon Sep 17 00:00:00 2001
From: Cody Constine <cody.constine@thetradedesk.com>
Date: Fri, 6 Dec 2024 15:54:23 -0700
Subject: [PATCH] Added more work

---
 .../uid2/shared/store/RotatingEncryptedSaltProvider.java | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/src/main/java/com/uid2/shared/store/RotatingEncryptedSaltProvider.java b/src/main/java/com/uid2/shared/store/RotatingEncryptedSaltProvider.java
index 300bd8cf..74fd171c 100644
--- a/src/main/java/com/uid2/shared/store/RotatingEncryptedSaltProvider.java
+++ b/src/main/java/com/uid2/shared/store/RotatingEncryptedSaltProvider.java
@@ -5,13 +5,15 @@
 import com.uid2.shared.model.CloudEncryptionKey;
 import com.uid2.shared.store.reader.RotatingCloudEncryptionKeyProvider;
 
+import com.uid2.shared.store.reader.StoreReader;
 import io.vertx.core.json.JsonObject;
 import java.io.IOException;
 import java.io.InputStream;
 import java.nio.charset.StandardCharsets;
 import java.util.Base64;
+import java.util.Collection;
 
-public class RotatingEncryptedSaltProvider extends RotatingSaltProvider{
+public class RotatingEncryptedSaltProvider extends RotatingSaltProvider implements StoreReader<Collection<RotatingSaltProvider.SaltSnapshot>> {
     private final RotatingCloudEncryptionKeyProvider cloudEncryptionKeyProvider;
 
     public RotatingEncryptedSaltProvider(DownloadCloudStorage fileStreamProvider, String metadataPath, RotatingCloudEncryptionKeyProvider cloudEncryptionKeyProvider) {
@@ -38,4 +40,9 @@ protected String readInputStream(InputStream inputStream) throws IOException {
 
         return new String(decryptedBytes, StandardCharsets.UTF_8);
     }
+
+    @Override
+    public Collection<SaltSnapshot> getAll() {
+        return super.getSnapshots();
+    }
 }