package de.juplo.kafka.chat.backend.persistence.inmemory;
import de.juplo.kafka.chat.backend.domain.ChatRoom;
-import de.juplo.kafka.chat.backend.domain.ChatHomeService;
import lombok.extern.slf4j.Slf4j;
import reactor.core.publisher.Flux;
import reactor.core.publisher.Mono;
import java.util.*;
+import java.util.stream.IntStream;
@Slf4j
-public class InMemoryChatHomeService implements ChatHomeService
+public class InMemoryChatHomeService
{
private final Map<UUID, ChatRoom>[] chatrooms;
.forEach(chatroom -> chatrooms[chatroom.getShard()].put(chatroom.getId(), chatroom));
}
- @Override
- public Mono<ChatRoom> putChatRoom(ChatRoom chatRoom)
+ public void putChatRoom(ChatRoom chatRoom)
{
chatrooms[chatRoom.getShard()].put(chatRoom.getId(), chatRoom);
- return Mono.just(chatRoom);
}
- @Override
public Mono<ChatRoom> getChatRoom(int shard, UUID id)
{
return Mono.justOrEmpty(chatrooms[shard].get(id));
}
- @Override
+ public int[] getOwnedShards()
+ {
+ return IntStream
+ .range(0, chatrooms.length)
+ .filter(i -> chatrooms[i] != null)
+ .toArray();
+ }
+
public Flux<ChatRoom> getChatRooms(int shard)
{
return Flux.fromStream(chatrooms[shard].values().stream());