1 package de.juplo.kafka.chat.backend.implementation;
3 import de.juplo.kafka.chat.backend.domain.ChatHomeService;
4 import de.juplo.kafka.chat.backend.domain.ChatRoomInfo;
5 import de.juplo.kafka.chat.backend.domain.Message;
6 import org.slf4j.Logger;
7 import org.slf4j.LoggerFactory;
8 import reactor.core.publisher.Flux;
10 import java.util.UUID;
11 import java.util.concurrent.Phaser;
12 import java.util.concurrent.atomic.AtomicBoolean;
13 import java.util.concurrent.atomic.AtomicInteger;
14 import java.util.function.BiConsumer;
15 import java.util.function.Consumer;
18 public interface StorageStrategy
20 Logger log = LoggerFactory.getLogger(StorageStrategy.class.getCanonicalName());
22 AtomicBoolean running = new AtomicBoolean(true);
25 default void write(ChatHomeService chatHomeService)
27 if (!running.getAndSet(false))
29 log.info("{} is not running, skip write...", chatHomeService);
33 Phaser writtenChatRooms = new Phaser(1);
34 AtomicInteger numErrors = new AtomicInteger();
39 .doOnNext(chatRoomInfo -> writtenChatRooms.register())
40 .doOnNext(chatRoomInfo -> writeChatRoomData(
43 .getChatRoomData(chatRoomInfo.getId())
44 .flatMapMany(chatRoomData -> chatRoomData.getMessages()),
47 logSuccess(chatRoomId);
48 writtenChatRooms.arriveAndDeregister();
50 (chatRoomId, throwable) ->
52 logFailure(chatRoomId, throwable);
53 numErrors.incrementAndGet();
54 writtenChatRooms.arriveAndDeregister();
57 writtenChatRooms.arriveAndAwaitAdvance();
58 if (numErrors.get() > 0)
60 throw new RuntimeException("Could not write all chat-rooms for " + chatHomeService);
63 log.info("All chat-rooms were written successfully for {}", chatHomeService);
66 void writeChatRoomInfo(Flux<ChatRoomInfo> chatRoomInfoFlux);
67 Flux<ChatRoomInfo> readChatRoomInfo();
68 default void writeChatRoomData(
70 Flux<Message> messageFlux,
71 SuccessCallback successCallback,
72 FailureCallback failureCallback)
77 .doOnComplete(() -> successCallback.accept(chatRoomId))
78 .doOnError(throwable -> failureCallback.accept(chatRoomId, throwable)));
80 void writeChatRoomData(UUID chatRoomId, Flux<Message> messageFlux);
81 Flux<Message> readChatRoomData(UUID chatRoomId);
83 interface SuccessCallback extends Consumer<UUID> {}
84 interface FailureCallback extends BiConsumer<UUID, Throwable> {}
86 default void logSuccess(UUID chatRoomId)
88 log.info("Successfully stored chat-room {}", chatRoomId);
91 default void logFailure(UUID chatRoomId, Throwable throwable)
93 log.error("Could not store chat-room {}: {}", chatRoomId, throwable);