check is flushed after resolve so we flush remaining logs

Author: vahidlazio

openfeature-provider-local/src/main/java/com/spotify/confidence/SwapWasmResolverApi.java

@@ -46,9 +46,9 @@ public void updateStateAndFlushLogs(byte[] state, String accountId) {
     // Get current instance before switching
     final WasmResolveApi oldInstance = wasmResolverApiRef.getAndSet(newInstance);
 
-    // Flush logs from the old instance (this allows it to be GC'd after method completion)
     if (oldInstance != null) {
       oldInstance.flushLogs();
+      oldInstance.isIsFlushed(true);
     }
     logResolveLock.unlock();
   }
@@ -75,7 +75,8 @@ public CompletableFuture<ResolveFlagsResponse> resolveWithSticky(
 
   private CompletableFuture<ResolveFlagsResponse> resolveWithStickyInternal(
       ResolveWithStickyRequest request) {
-    final var response = wasmResolverApiRef.get().resolveWithSticky(request);
+    final var instance = wasmResolverApiRef.get();
+    final var response = instance.resolveWithSticky(request);
 
     switch (response.getResolveResultCase()) {
       case SUCCESS -> {
@@ -84,6 +85,9 @@ private CompletableFuture<ResolveFlagsResponse> resolveWithStickyInternal(
         if (!success.getUpdatesList().isEmpty()) {
           storeUpdates(success.getUpdatesList());
         }
+        if (instance.isFlushed()) {
+          CompletableFuture.runAsync(instance::flushLogs);
+        }
         return CompletableFuture.completedFuture(success.getResponse());
       }
       case MISSING_MATERIALIZATIONS -> {
@@ -205,7 +209,11 @@ private ResolveWithStickyRequest handleMissingMaterializations(
   @Override
   public ResolveFlagsResponse resolve(ResolveFlagsRequest request) {
     logResolveLock.lock();
-    final var response = wasmResolverApiRef.get().resolve(request);
+    final var instance = wasmResolverApiRef.get();
+    final var response = instance.resolve(request);
+    if (instance.isFlushed()) {
+      CompletableFuture.runAsync(instance::flushLogs);
+    }
     logResolveLock.unlock();
     return response;
   }

openfeature-provider-local/src/main/java/com/spotify/confidence/WasmResolveApi.java

@@ -51,6 +51,7 @@ class WasmResolveApi {
 
   // Retry strategy
   private final RetryStrategy retryStrategy;
+  private boolean isFlushed;
 
   public WasmResolveApi(WasmFlagLogger flagLogger, RetryStrategy retryStrategy) {
     this.retryStrategy = retryStrategy;
@@ -219,6 +220,14 @@ private <T extends GeneratedMessage> ImportFunction createImportFunction(
         });
   }
 
+  public void isIsFlushed(boolean b) {
+    this.isFlushed = true;
+  }
+
+  public boolean isFlushed() {
+    return isFlushed;
+  }
+
   private interface ParserFn<T> {
 
     T apply(byte[] data) throws InvalidProtocolBufferException;