import reactor.core.publisher.Flux;
import reactor.core.publisher.Mono;
-import java.time.Clock;
-import java.time.LocalDateTime;
-import java.util.Optional;
import java.util.UUID;
-import java.util.stream.Stream;
@RestController
public class ChatBackendController
{
private final ChatHome chatHome;
- private final Clock clock;
private final StorageStrategy storageStrategy;
@PostMapping("create")
- public ChatroomTo create(@RequestBody String name)
+ public Mono<ChatRoomTo> create(@RequestBody String name)
{
- return ChatroomTo.from(chatHome.createChatroom(name));
+ return chatHome.createChatroom(name).map(ChatRoomTo::from);
}
@GetMapping("list")
- public Stream<ChatroomTo> list()
+ public Flux<ChatRoomTo> list()
{
- return chatHome.list().map(chatroom -> ChatroomTo.from(chatroom));
+ return chatHome.getChatRooms().map(chatroom -> ChatRoomTo.from(chatroom));
}
- @GetMapping("list/{chatroomId}")
+ @GetMapping("{chatroomId}/list")
public Flux<MessageTo> list(@PathVariable UUID chatroomId)
{
return chatHome
- .getChatroom(chatroomId)
- .map(chatroom -> chatroom
+ .getChatRoom(chatroomId)
+ .flatMapMany(chatroom -> chatroom
.getMessages()
- .map(MessageTo::from))
- .get();
+ .map(MessageTo::from));
}
- @GetMapping("get/{chatroomId}")
- public Optional<ChatroomTo> get(@PathVariable UUID chatroomId)
+ @GetMapping("{chatroomId}")
+ public Mono<ChatRoomTo> get(@PathVariable UUID chatroomId)
{
- return chatHome.getChatroom(chatroomId).map(chatroom -> ChatroomTo.from(chatroom));
+ return chatHome.getChatRoom(chatroomId).map(chatroom -> ChatRoomTo.from(chatroom));
}
- @PutMapping("put/{chatroomId}/{username}/{messageId}")
+ @PutMapping("{chatroomId}/{username}/{messageId}")
public Mono<MessageTo> put(
@PathVariable UUID chatroomId,
@PathVariable String username,
{
return
chatHome
- .getChatroom(chatroomId)
- .map(chatroom -> put(chatroom, username, messageId, text))
- .orElseThrow(() -> new UnknownChatroomException(chatroomId));
+ .getChatRoom(chatroomId)
+ .flatMap(chatroom -> put(chatroom, username, messageId, text));
}
public Mono<MessageTo> put(
chatroom
.addMessage(
messageId,
- LocalDateTime.now(clock),
username,
text)
- .switchIfEmpty(chatroom.getMessage(username, messageId))
.map(message -> MessageTo.from(message));
}
- @GetMapping("get/{chatroomId}/{username}/{messageId}")
+ @GetMapping("{chatroomId}/{username}/{messageId}")
public Mono<MessageTo> get(
@PathVariable UUID chatroomId,
@PathVariable String username,
{
return
chatHome
- .getChatroom(chatroomId)
- .map(chatroom -> get(chatroom, username, messageId))
- .orElseThrow(() -> new UnknownChatroomException(chatroomId));
+ .getChatRoom(chatroomId)
+ .flatMap(chatroom -> get(chatroom, username, messageId));
}
private Mono<MessageTo> get(
.map(message -> MessageTo.from(message));
}
- @GetMapping(path = "listen/{chatroomId}")
+ @GetMapping(path = "{chatroomId}/listen")
public Flux<ServerSentEvent<MessageTo>> listen(@PathVariable UUID chatroomId)
{
return chatHome
- .getChatroom(chatroomId)
- .map(chatroom -> listen(chatroom))
- .orElseThrow(() -> new UnknownChatroomException(chatroomId));
+ .getChatRoom(chatroomId)
+ .flatMapMany(chatroom -> listen(chatroom));
}
private Flux<ServerSentEvent<MessageTo>> listen(ChatRoom chatroom)
@PostMapping("/store")
public void store()
{
- storageStrategy.writeChatrooms(Flux.fromStream(chatHome.list()));
+ storageStrategy.write(chatHome.getChatRooms());
}
}