import lombok.extern.slf4j.Slf4j;
import reactor.core.publisher.Flux;
+import reactor.core.publisher.Mono;
import java.util.*;
-import java.util.stream.Stream;
@Slf4j
public class ChatHome
{
- private final Map<UUID, Chatroom> chatrooms;
- private final ChatroomFactory factory;
+ private final Map<UUID, ChatRoom> chatrooms;
+ private final ChatHomeService service;
- public ChatHome(ChatroomFactory factory, Flux<Chatroom> chatroomFlux)
+ public ChatHome(ChatHomeService service, Flux<ChatRoom> chatroomFlux)
{
- log.debug("Creating ChatHome with factory: {}", factory);
- this.factory = factory;
+ log.debug("Creating ChatHome with factory: {}", service);
+ this.service = service;
this.chatrooms = new HashMap<>();
chatroomFlux.subscribe(chatroom -> chatrooms.put(chatroom.getId(), chatroom));
}
- public Chatroom createChatroom(String name)
+ public Mono<ChatRoom> createChatroom(String name)
{
- Chatroom chatroom = factory.createChatroom(UUID.randomUUID(), name);
+ ChatRoom chatroom = service.createChatroom(UUID.randomUUID(), name);
chatrooms.put(chatroom.getId(), chatroom);
- return chatroom;
+ return Mono.justOrEmpty(chatroom);
}
- public Optional<Chatroom> getChatroom(UUID id)
+ public Mono<ChatRoom> getChatroom(UUID id)
{
- return Optional.ofNullable(chatrooms.get(id));
+ return Mono
+ .justOrEmpty(chatrooms.get(id))
+ .or(Mono.error(() -> new UnknownChatroomException(id)));
}
- public Stream<Chatroom> list()
+ public Flux<ChatRoom> list()
{
- return chatrooms.values().stream();
+ return Flux.fromStream(chatrooms.values().stream());
}
}