import de.juplo.kafka.chat.backend.domain.ChatRoom;
import de.juplo.kafka.chat.backend.domain.ChatHomeService;
-import de.juplo.kafka.chat.backend.domain.ShardingStrategy;
import lombok.extern.slf4j.Slf4j;
import reactor.core.publisher.Flux;
import reactor.core.publisher.Mono;
@Slf4j
-public class InMemoryChatHomeService implements ChatHomeService<InMemoryChatRoomService>
+public class InMemoryChatHomeService implements ChatHomeService
{
- private final ShardingStrategy shardingStrategy;
private final Map<UUID, ChatRoom>[] chatrooms;
public InMemoryChatHomeService(
- ShardingStrategy shardingStrategy,
int numShards,
int[] ownedShards,
Flux<ChatRoom> chatroomFlux)
{
log.debug("Creating InMemoryChatHomeService");
- this.shardingStrategy = shardingStrategy;
this.chatrooms = new Map[numShards];
Set<Integer> owned = Arrays
.stream(ownedShards)
chatroomFlux
.filter(chatRoom ->
{
- int shard = shardingStrategy.selectShard(chatRoom.getId());
- if (owned.contains(shard))
+ if (owned.contains(chatRoom.getShard()))
{
return true;
}
}
})
.toStream()
- .forEach(chatRoom ->
- {
- getChatRoomMapFor(chatRoom).put(chatRoom.getId(), chatRoom);
- });
+ .forEach(chatroom -> chatrooms[chatroom.getShard()].put(chatroom.getId(), chatroom));
}
- @Override
- public Mono<ChatRoom> putChatRoom(ChatRoom chatRoom)
+ public void putChatRoom(ChatRoom chatRoom)
{
- getChatRoomMapFor(chatRoom).put(chatRoom.getId(), chatRoom);
- return Mono.just(chatRoom);
+ chatrooms[chatRoom.getShard()].put(chatRoom.getId(), chatRoom);
}
@Override
{
return Flux.fromStream(chatrooms[shard].values().stream());
}
-
-
- private Map<UUID, ChatRoom> getChatRoomMapFor(ChatRoom chatRoom)
- {
- int shard = shardingStrategy.selectShard(chatRoom.getId());
- return chatrooms[shard];
- }
}