Wait for file writer complete (opensource4you#570)

garyparrot · Aug 10, 2022 · ae0147a · ae0147a
1 parent 0d62821
commit ae0147a
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/app/src/main/java/org/astraea/app/performance/Performance.java b/app/src/main/java/org/astraea/app/performance/Performance.java
@@ -167,7 +167,8 @@ public static String execute(final Argument param) throws InterruptedException,
                             .sum(),
                     tracker));
 
-    fileWriter.ifPresent(CompletableFuture::runAsync);
+    var fileWriterFuture =
+        fileWriter.map(CompletableFuture::runAsync).orElse(CompletableFuture.completedFuture(null));
 
     CompletableFuture.runAsync(
         () -> {
@@ -191,6 +192,7 @@ public static String execute(final Argument param) throws InterruptedException,
 
     consumerThreads.forEach(AbstractThread::waitForDone);
     tracker.waitForDone();
+    fileWriterFuture.join();
     return param.topic;
   }