X-Git-Url: http://juplo.de/gitweb/?a=blobdiff_plain;ds=sidebyside;f=src%2Fmain%2Fjava%2Fde%2Fjuplo%2Fkafka%2Fchat%2Fbackend%2Fimplementation%2FStorageStrategy.java;h=868ae11b280a9942c434ddfb431d55f693adfc17;hb=e3979aa4f5ad0c5a4f55dff6391e29f7b14d9b53;hp=ba8bc23a9d7fc58565f8787e64add015f0cd91b2;hpb=9fbb1a24b8c62619f8e51c5575b70b66fcd99ff8;p=demos%2Fkafka%2Fchat

diff --git a/src/main/java/de/juplo/kafka/chat/backend/implementation/StorageStrategy.java b/src/main/java/de/juplo/kafka/chat/backend/implementation/StorageStrategy.java
index ba8bc23a..868ae11b 100644
--- a/src/main/java/de/juplo/kafka/chat/backend/implementation/StorageStrategy.java
+++ b/src/main/java/de/juplo/kafka/chat/backend/implementation/StorageStrategy.java
@@ -8,6 +8,8 @@ import org.slf4j.LoggerFactory;
 import reactor.core.publisher.Flux;
 
 import java.util.UUID;
+import java.util.concurrent.Phaser;
+import java.util.concurrent.atomic.AtomicInteger;
 import java.util.function.BiConsumer;
 import java.util.function.Consumer;
 
@@ -18,35 +20,66 @@ public interface StorageStrategy
 
   default void write(ChatHomeService chatHomeService)
   {
+    Phaser writtenChatRooms = new Phaser(1);
+    AtomicInteger numErrors = new AtomicInteger();
+
     writeChatRoomInfo(
         chatHomeService
             .getChatRoomInfo()
-            .doOnNext(chatRoomInfo ->
-                writeChatRoomData(
-                    chatRoomInfo.getId(),
-                    chatHomeService
-                        .getChatRoomData(chatRoomInfo.getId())
-                        .flatMapMany(chatRoomData -> chatRoomData.getMessages()))));
-  }
+            .doOnNext(chatRoomInfo -> writtenChatRooms.register())
+            .doOnNext(chatRoomInfo -> writeChatRoomData(
+                chatRoomInfo.getId(),
+                chatHomeService
+                    .getChatRoomData(chatRoomInfo.getId())
+                    .flatMapMany(chatRoomData -> chatRoomData.getMessages()),
+                (chatRoomId) ->
+                {
+                  logSuccess(chatRoomId);
+                  writtenChatRooms.arriveAndDeregister();
+                },
+                (chatRoomId, throwable) ->
+                {
+                  logFailure(chatRoomId, throwable);
+                  numErrors.incrementAndGet();
+                  writtenChatRooms.arriveAndDeregister();
+                })));
 
-  default void writeChatRoomData(UUID chatRoomId, Flux<Message> messageFlux)
-  {
-    writeChatRoomData(
-        chatRoomId,
-        messageFlux,
-        (id) -> log.info("Successfully stored chat-room {}", id),
-        (id, throwable) -> log.error("Could not store chat-room {}: {}", id, throwable));
+    writtenChatRooms.arriveAndAwaitAdvance();
+    if (numErrors.get() > 0)
+    {
+      throw new RuntimeException("Could not write all chat-rooms for " + chatHomeService);
+    }
+
+    log.info("All chat-rooms were written successfully for {}", chatHomeService);
   }
 
   void writeChatRoomInfo(Flux<ChatRoomInfo> chatRoomInfoFlux);
   Flux<ChatRoomInfo> readChatRoomInfo();
-  void writeChatRoomData(
+  default void writeChatRoomData(
       UUID chatRoomId,
       Flux<Message> messageFlux,
-      SuccessCallback callback,
-      FailureCallback failureCallback);
+      SuccessCallback successCallback,
+      FailureCallback failureCallback)
+  {
+    writeChatRoomData(
+        chatRoomId,
+        messageFlux
+            .doOnComplete(() -> successCallback.accept(chatRoomId))
+            .doOnError(throwable -> failureCallback.accept(chatRoomId, throwable)));
+  }
+  void writeChatRoomData(UUID chatRoomId, Flux<Message> messageFlux);
   Flux<Message> readChatRoomData(UUID chatRoomId);
 
   interface SuccessCallback extends Consumer<UUID> {}
   interface FailureCallback extends BiConsumer<UUID, Throwable> {}
+
+  default void logSuccess(UUID chatRoomId)
+  {
+    log.info("Successfully stored chat-room {}", chatRoomId);
+  }
+
+  default void logFailure(UUID chatRoomId, Throwable throwable)
+  {
+    log.error("Could not store chat-room {}: {}", chatRoomId, throwable);
+  }
 }