refactor: Fixed return-types of the controller

[demos/kafka/chat] / src / main / java / de / juplo / kafka / chat / backend / domain / ChatHome.java

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 a75a9a1..d04bd73 100644 (file)
--- a/src/main/java/de/juplo/kafka/chat/backend/domain/ChatHome.java
+++ b/src/main/java/de/juplo/kafka/chat/backend/domain/ChatHome.java
@@ -3,12 +3,13 @@ package de.juplo.kafka.chat.backend.domain;
 import lombok.RequiredArgsConstructor;
 
 import java.util.*;
+import java.util.stream.Stream;
 
 
 @RequiredArgsConstructor
 public class ChatHome
 {
-  private final Map<UUID, Chatroom> chatrooms = new HashMap<>();
+  private final Map<UUID, Chatroom> chatrooms;
   private final ChatroomFactory factory;
 
 
@@ -19,13 +20,13 @@ public class ChatHome
     return chatroom;
   }
 
-  public Chatroom getChatroom(UUID id)
+  public Optional<Chatroom> getChatroom(UUID id)
   {
-    return chatrooms.get(id);
+    return Optional.ofNullable(chatrooms.get(id));
   }
 
-  public Collection<Chatroom> list()
+  public Stream<Chatroom> list()
   {
-    return chatrooms.values();
+    return chatrooms.values().stream();
   }
 }