Add callback support to RotatingStoreVerticle for keyset key fail-fast feature

- Add Consumer<Boolean> refreshCallback field
- Add constructor overload to accept callback
- Call callback on refresh success/failure
- Maintains backward compatibility with existing constructor

Author: lizk886

src/main/java/com/uid2/shared/vertx/RotatingStoreVerticle.java

@@ -15,6 +15,7 @@
 
 import java.util.concurrent.atomic.AtomicInteger;
 import java.util.concurrent.atomic.AtomicLong;
+import java.util.function.Consumer;
 
 public class RotatingStoreVerticle extends AbstractVerticle {
     private static final Logger LOGGER = LoggerFactory.getLogger(RotatingStoreVerticle.class);
@@ -31,47 +32,54 @@ public class RotatingStoreVerticle extends AbstractVerticle {
     private final AtomicLong latestVersion = new AtomicLong(-1L);
     private final AtomicLong latestEntryCount = new AtomicLong(-1L);
     private final AtomicInteger storeRefreshIsFailing = new AtomicInteger(0);
+    private final Consumer<Boolean> refreshCallback;
 
     private final long refreshIntervalMs;
 
     public RotatingStoreVerticle(String storeName, long refreshIntervalMs, IMetadataVersionedStore versionedStore) {
+        this(storeName, refreshIntervalMs, versionedStore, null);
+    }
+
+    public RotatingStoreVerticle(String storeName, long refreshIntervalMs, IMetadataVersionedStore versionedStore,
+            Consumer<Boolean> refreshCallback) {
         this.healthComponent = HealthManager.instance.registerComponent(storeName + "-rotator");
         this.healthComponent.setHealthStatus(false, "not started");
 
         this.storeName = storeName;
         this.counterStoreRefreshed = Counter
-            .builder("uid2_config_store_refreshed_total")
-            .tag("store", storeName)
-            .description("counter for how many times " + storeName + " store is refreshed")
-            .register(Metrics.globalRegistry);
+                .builder("uid2_config_store_refreshed_total")
+                .tag("store", storeName)
+                .description("counter for how many times " + storeName + " store is refreshed")
+                .register(Metrics.globalRegistry);
         this.counterStoreRefreshTimeMs = Counter
-            .builder("uid2_config_store_refreshtime_ms_total")
-            .tag("store", storeName)
-            .description("counter for total time (ms) " + storeName + " store spend in refreshing")
-            .register(Metrics.globalRegistry);
+                .builder("uid2_config_store_refreshtime_ms_total")
+                .tag("store", storeName)
+                .description("counter for total time (ms) " + storeName + " store spend in refreshing")
+                .register(Metrics.globalRegistry);
         this.counterStoreRefreshFailures = Counter
-            .builder("uid2_config_store_refresh_failures_total")
-            .tag("store", storeName)
-            .description("counter for number of " + storeName + " store refresh failures")
-            .register(Metrics.globalRegistry);
+                .builder("uid2_config_store_refresh_failures_total")
+                .tag("store", storeName)
+                .description("counter for number of " + storeName + " store refresh failures")
+                .register(Metrics.globalRegistry);
         this.gaugeStoreVersion = Gauge
-            .builder("uid2_config_store_version", () -> this.latestVersion.get())
-            .tag("store", storeName)
-            .description("gauge for " + storeName + " store version")
-            .register(Metrics.globalRegistry);
+                .builder("uid2_config_store_version", () -> this.latestVersion.get())
+                .tag("store", storeName)
+                .description("gauge for " + storeName + " store version")
+                .register(Metrics.globalRegistry);
         this.gaugeStoreEntryCount = Gauge
-            .builder("uid2_config_store_entry_count", () -> this.latestEntryCount.get())
-            .tag("store", storeName)
-            .description("gauge for " + storeName + " store total entry count")
-            .register(Metrics.globalRegistry);
+                .builder("uid2_config_store_entry_count", () -> this.latestEntryCount.get())
+                .tag("store", storeName)
+                .description("gauge for " + storeName + " store total entry count")
+                .register(Metrics.globalRegistry);
         this.gaugeConsecutiveRefreshFailures = Gauge
-            .builder("uid2_config_store_consecutive_refresh_failures", () -> this.storeRefreshIsFailing.get())
-            .tag("store", storeName)
-            .description("gauge for number of consecutive " + storeName + " store refresh failures")
-            .register(Metrics.globalRegistry);
+                .builder("uid2_config_store_consecutive_refresh_failures", () -> this.storeRefreshIsFailing.get())
+                .tag("store", storeName)
+                .description("gauge for number of consecutive " + storeName + " store refresh failures")
+                .register(Metrics.globalRegistry);
         this.versionedStore = versionedStore;
         this.refreshIntervalMs = refreshIntervalMs;
         this.storeRefreshTimer = Metrics.timer("uid2_store_refresh_duration", "store_name", storeName);
+        this.refreshCallback = refreshCallback;
     }
 
     @Override
@@ -98,7 +106,8 @@ private void startRefresh(Promise<Void> promise) {
                 this.startBackgroundRefresh();
             } else {
                 this.healthComponent.setHealthStatus(false, ar.cause().getMessage());
-                LOGGER.error("Failed " + this.storeName + " loading. Trying again in " + refreshIntervalMs + "ms", ar.cause());
+                LOGGER.error("Failed " + this.storeName + " loading. Trying again in " + refreshIntervalMs + "ms",
+                        ar.cause());
                 vertx.setTimer(refreshIntervalMs, id -> this.startRefresh(promise));
             }
         });
@@ -109,23 +118,28 @@ private void startBackgroundRefresh() {
             final long start = System.nanoTime();
 
             vertx.executeBlocking(() -> {
-                    this.refresh();
-                    return null;
-                }).onComplete(asyncResult -> {
-                    final long end = System.nanoTime();
-                    final long elapsed = ((end - start) / 1000000);
-                    this.counterStoreRefreshTimeMs.increment(elapsed);
-                    if (asyncResult.failed()) {
-                        this.counterStoreRefreshFailures.increment();
-                        this.storeRefreshIsFailing.set(1);
-                        LOGGER.error("Failed to load " + this.storeName + ", " + elapsed + " ms", asyncResult.cause());
-                    } else {
-                        this.counterStoreRefreshed.increment();
-                        this.storeRefreshIsFailing.set(0);
-                        LOGGER.trace("Successfully refreshed " + this.storeName + ", " + elapsed + " ms");
+                this.refresh();
+                return null;
+            }).onComplete(asyncResult -> {
+                final long end = System.nanoTime();
+                final long elapsed = ((end - start) / 1000000);
+                this.counterStoreRefreshTimeMs.increment(elapsed);
+                if (asyncResult.failed()) {
+                    this.counterStoreRefreshFailures.increment();
+                    this.storeRefreshIsFailing.set(1);
+                    LOGGER.error("Failed to load " + this.storeName + ", " + elapsed + " ms", asyncResult.cause());
+                    if (this.refreshCallback != null) {
+                        this.refreshCallback.accept(false);
+                    }
+                } else {
+                    this.counterStoreRefreshed.increment();
+                    this.storeRefreshIsFailing.set(0);
+                    LOGGER.trace("Successfully refreshed " + this.storeName + ", " + elapsed + " ms");
+                    if (this.refreshCallback != null) {
+                        this.refreshCallback.accept(true);
                     }
                 }
-            );
+            });
         });
     }