fix: GREEN - `ChatRoomData` obeys to the added expectations.

[demos/kafka/chat] / src / main / java / de / juplo / kafka / chat / backend / implementation / StorageStrategy.java

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 647ac52..87208dc 100644 (file)
--- a/src/main/java/de/juplo/kafka/chat/backend/implementation/StorageStrategy.java
+++ b/src/main/java/de/juplo/kafka/chat/backend/implementation/StorageStrategy.java
@@ -24,12 +24,13 @@ public interface StorageStrategy
                 .getChatRoomData(chatRoomInfo.getId())
                 .flatMapMany(chatRoomData -> chatRoomData.getMessages())
             )
-            .then()
-            .doOnSuccess(empty -> log.info("Stored {}", chatRoomInfo))
+            .count()
+            .doOnSuccess(count -> log.info("Stored {} messages for {}", count, chatRoomInfo))
             .doOnError(throwable -> log.error("Could not store {}: {}", chatRoomInfo, throwable)))
-        .then()
-        .doOnSuccess(empty -> log.info("Stored {}", chatHomeService))
-        .doOnError(throwable -> log.error("Could not store {}: {}", chatHomeService, throwable));
+        .count()
+        .doOnSuccess(count -> log.info("Stored {} chat-rooms for {}", count, chatHomeService))
+        .doOnError(throwable -> log.error("Could not store {}: {}", chatHomeService, throwable))
+        .then();
   }
 
   Flux<ChatRoomInfo> writeChatRoomInfo(Flux<ChatRoomInfo> chatRoomInfoFlux);