X-Git-Url: https://juplo.de/gitweb/?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fde%2Fjuplo%2Fkafka%2Fchat%2Fbackend%2Fdomain%2FChatHome.java;h=2fc0e356091a3791cc37c31fe50f55738042fdc2;hb=9d61871563f4be4850cebca4fad0545d504522c3;hp=bb4d89c5b4297ef86450a7a9592bfe95e3b983a3;hpb=0e6fd57b277df31ef319f47c63bc182694e46082;p=demos%2Fkafka%2Fchat

diff --git a/src/main/java/de/juplo/kafka/chat/backend/domain/ChatHome.java b/src/main/java/de/juplo/kafka/chat/backend/domain/ChatHome.java
index bb4d89c5..2fc0e356 100644
--- a/src/main/java/de/juplo/kafka/chat/backend/domain/ChatHome.java
+++ b/src/main/java/de/juplo/kafka/chat/backend/domain/ChatHome.java
@@ -1,5 +1,6 @@
 package de.juplo.kafka.chat.backend.domain;
 
+import lombok.RequiredArgsConstructor;
 import lombok.extern.slf4j.Slf4j;
 import reactor.core.publisher.Flux;
 import reactor.core.publisher.Mono;
@@ -8,36 +9,26 @@ import java.util.*;
 
 
 @Slf4j
+@RequiredArgsConstructor
 public class ChatHome
 {
-  private final Map<UUID, ChatRoom> chatrooms;
   private final ChatHomeService service;
+  private final int shard;
 
-  public ChatHome(ChatHomeService service, Flux<ChatRoom> chatroomFlux)
+  public Mono<ChatRoom> putChatRoom(ChatRoom chatRoom)
   {
-    log.debug("Creating ChatHome with factory: {}", service);
-    this.service = service;
-    this.chatrooms = new HashMap<>();
-    chatroomFlux.subscribe(chatroom -> chatrooms.put(chatroom.getId(), chatroom));
+    return service.putChatRoom(chatRoom);
   }
 
-  public Mono<ChatRoom> createChatroom(String name)
+  public Mono<ChatRoom> getChatRoom(UUID id)
   {
-    ChatRoom chatroom = service.createChatroom(UUID.randomUUID(), name);
-    chatrooms.put(chatroom.getId(), chatroom);
-    return Mono.just(chatroom);
+    return service
+        .getChatRoom(shard, id)
+        .switchIfEmpty(Mono.error(() -> new UnknownChatroomException(id)));
   }
 
-  public Mono<ChatRoom> getChatroom(UUID id)
+  public Flux<ChatRoom> getChatRooms()
   {
-    ChatRoom chatroom = chatrooms.get(id);
-    return chatroom == null
-        ? Mono.error(() -> new UnknownChatroomException(id))
-        : Mono.just(chatroom);
-  }
-
-  public Flux<ChatRoom> list()
-  {
-    return Flux.fromStream(chatrooms.values().stream());
+    return service.getChatRooms(shard);
   }
 }