refactor: Added success- and failure-callbacks to `StorageStrategy`

[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 cbf55a4..ba8bc23 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
@@ -1,15 +1,21 @@
-package de.juplo.kafka.chat.backend.persistence;
+package de.juplo.kafka.chat.backend.implementation;
 
 import de.juplo.kafka.chat.backend.domain.ChatHomeService;
 import de.juplo.kafka.chat.backend.domain.ChatRoomInfo;
 import de.juplo.kafka.chat.backend.domain.Message;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
 import reactor.core.publisher.Flux;
 
 import java.util.UUID;
+import java.util.function.BiConsumer;
+import java.util.function.Consumer;
 
 
 public interface StorageStrategy
 {
+  Logger log = LoggerFactory.getLogger(StorageStrategy.class.getCanonicalName());
+
   default void write(ChatHomeService chatHomeService)
   {
     writeChatRoomInfo(
@@ -23,8 +29,24 @@ public interface StorageStrategy
                         .flatMapMany(chatRoomData -> chatRoomData.getMessages()))));
   }
 
+  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));
+  }
+
   void writeChatRoomInfo(Flux<ChatRoomInfo> chatRoomInfoFlux);
   Flux<ChatRoomInfo> readChatRoomInfo();
-  void writeChatRoomData(UUID chatRoomId, Flux<Message> messageFlux);
+  void writeChatRoomData(
+      UUID chatRoomId,
+      Flux<Message> messageFlux,
+      SuccessCallback callback,
+      FailureCallback failureCallback);
   Flux<Message> readChatRoomData(UUID chatRoomId);
+
+  interface SuccessCallback extends Consumer<UUID> {}
+  interface FailureCallback extends BiConsumer<UUID, Throwable> {}
 }