Add isBucketValid method in ShardCalculator

santanusinha · May 2, 2024 · a2ed001 · a2ed001
1 parent a964f69
commit a2ed001
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/main/java/io/appform/dropwizard/sharding/utils/ShardCalculator.java b/src/main/java/io/appform/dropwizard/sharding/utils/ShardCalculator.java
@@ -48,8 +48,12 @@ public int shardId(T key) {
         return shardManager.shardForBucket(bucketId);
     }
 
+    public boolean isBucketValid(int bucketId) {
+        return shardManager.isMappedToValidShard(bucketId);
+    }
+
     public boolean isOnValidShard(T key) {
         int bucketId = extractor.bucketId(key);
-        return shardManager.isMappedToValidShard(bucketId);
+        return isBucketValid(bucketId);
     }
 }