import de.juplo.kafka.chat.backend.domain.ChatHome;
import de.juplo.kafka.chat.backend.domain.ChatRoom;
+import de.juplo.kafka.chat.backend.domain.ChatRoomFactory;
import de.juplo.kafka.chat.backend.persistence.StorageStrategy;
import lombok.RequiredArgsConstructor;
import org.springframework.http.codec.ServerSentEvent;
@RequiredArgsConstructor
public class ChatBackendController
{
- private final ChatHome chatHome;
+ private final ChatHome[] chatHomes;
+ private final ShardingStrategy selectionStrategy;
+ private final ChatRoomFactory factory;
private final StorageStrategy storageStrategy;
@PostMapping("create")
public Mono<ChatRoomTo> create(@RequestBody String name)
{
- return chatHome.createChatroom(name).map(ChatRoomTo::from);
+ UUID chatRoomId = UUID.randomUUID();
+ return factory
+ .createChatRoom(chatRoomId, name)
+ .flatMap(chatRoom -> chatHomes[chatRoom.getShard()].putChatRoom(chatRoom))
+ .map(ChatRoomTo::from);
}
@GetMapping("list")
public Flux<ChatRoomTo> list()
{
- return chatHome.getChatRooms().map(chatroom -> ChatRoomTo.from(chatroom));
+ return Flux
+ .fromArray(chatHomes)
+ .flatMap(chatHome -> chatHome.getChatRooms())
+ .map(chatroom -> ChatRoomTo.from(chatroom));
}
@GetMapping("{chatroomId}/list")
public Flux<MessageTo> list(@PathVariable UUID chatroomId)
{
- return chatHome
+ return chatHomes[selectShard(chatroomId)]
.getChatRoom(chatroomId)
.flatMapMany(chatroom -> chatroom
.getMessages()
@GetMapping("{chatroomId}")
public Mono<ChatRoomTo> get(@PathVariable UUID chatroomId)
{
- return chatHome.getChatRoom(chatroomId).map(chatroom -> ChatRoomTo.from(chatroom));
+ return chatHomes[selectShard(chatroomId)]
+ .getChatRoom(chatroomId)
+ .map(chatroom -> ChatRoomTo.from(chatroom));
}
@PutMapping("{chatroomId}/{username}/{messageId}")
@RequestBody String text)
{
return
- chatHome
+ chatHomes[selectShard(chatroomId)]
.getChatRoom(chatroomId)
.flatMap(chatroom -> put(chatroom, username, messageId, text));
}
@PathVariable Long messageId)
{
return
- chatHome
+ chatHomes[selectShard(chatroomId)]
.getChatRoom(chatroomId)
.flatMap(chatroom -> get(chatroom, username, messageId));
}
@GetMapping(path = "{chatroomId}/listen")
public Flux<ServerSentEvent<MessageTo>> listen(@PathVariable UUID chatroomId)
{
- return chatHome
+ return chatHomes[selectShard(chatroomId)]
.getChatRoom(chatroomId)
.flatMapMany(chatroom -> listen(chatroom));
}
@PostMapping("/store")
public void store()
{
- storageStrategy.write(chatHome.getChatRooms());
+ for (int shard = 0; shard < chatHomes.length; shard++)
+ storageStrategy.write(chatHomes[shard].getChatRooms());
+ }
+
+ private int selectShard(UUID chatroomId)
+ {
+ return selectionStrategy.selectShard(chatroomId);
}
}