Update MinioStorageEngine.java

qwlabs · Jan 17, 2024 · 4cbf1e2 · 4cbf1e2
1 parent f02522a
commit 4cbf1e2
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/storage-minio/src/main/java/com/qwlabs/storage/minio/MinioStorageEngine.java b/storage-minio/src/main/java/com/qwlabs/storage/minio/MinioStorageEngine.java
@@ -2,6 +2,7 @@
 
 
 import com.google.common.collect.Lists;
+import com.qwlabs.cdi.dispatch.Dispatchable;
 import com.qwlabs.storage.messages.StorageMessages;
 import com.qwlabs.storage.models.CompleteUploadCommand;
 import com.qwlabs.storage.models.GetDownloadUrlCommand;
@@ -15,11 +16,14 @@
 import com.qwlabs.storage.services.StorageEngine;
 import io.minio.messages.ListPartsResult;
 import io.minio.messages.Part;
+import org.jetbrains.annotations.Nullable;
 
 import java.io.InputStream;
 import java.util.List;
+import java.util.Objects;
 
-public class MinioStorageEngine implements StorageEngine {
+public class MinioStorageEngine implements StorageEngine, Dispatchable<String> {
+    private static final String PROVIDER = "minio";
     protected final CustomMinioClient minioClient;
 
     public MinioStorageEngine(CustomMinioClient minioClient) {
@@ -92,4 +96,9 @@ private void setupBucket(String bucket) {
             minioClient.makeBucket(bucket);
         }
     }
+
+    @Override
+    public boolean dispatchable(@Nullable String context) {
+        return Objects.equals(PROVIDER, context);
+    }
 }