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