1 package de.juplo.kafka.chat.backend.persistence.inmemory;
3 import de.juplo.kafka.chat.backend.domain.ChatHome;
4 import de.juplo.kafka.chat.backend.domain.ChatRoom;
5 import de.juplo.kafka.chat.backend.domain.UnknownChatroomException;
6 import lombok.extern.slf4j.Slf4j;
7 import reactor.core.publisher.Flux;
8 import reactor.core.publisher.Mono;
14 public class SimpleChatHome implements ChatHome
16 private final int shard;
17 private final Map<UUID, ChatRoom> chatrooms;
21 public SimpleChatHome(
23 Flux<ChatRoom> chatroomFlux)
25 log.info("Created SimpleChatHome for shard {}", shard);
27 this.chatrooms = new HashMap<>();
31 if (shard > -1 && chatRoom.getShard() == shard)
38 "SimpleChatHome for shard {} ignores not owned chat-room {}",
45 .forEach(chatroom -> chatrooms.put(chatroom.getId(), chatroom));
51 public Mono<ChatRoom> getChatRoom(UUID id)
54 .justOrEmpty(chatrooms.get(id))
55 .switchIfEmpty(Mono.error(() -> new UnknownChatroomException(id)));
59 public Flux<ChatRoom> getChatRooms()
61 return Flux.fromIterable(chatrooms.values());