package de.juplo.kafka.chat.backend.api;
import de.juplo.kafka.chat.backend.domain.ChatHome;
-import de.juplo.kafka.chat.backend.domain.Chatroom;
+import de.juplo.kafka.chat.backend.domain.ChatRoom;
+import de.juplo.kafka.chat.backend.persistence.StorageStrategy;
import lombok.RequiredArgsConstructor;
-import org.springframework.http.MediaType;
+import org.springframework.http.codec.ServerSentEvent;
import org.springframework.web.bind.annotation.*;
import reactor.core.publisher.Flux;
import reactor.core.publisher.Mono;
-import java.time.Clock;
-import java.time.LocalDateTime;
-import java.util.Collection;
-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}")
+ public Flux<MessageTo> list(@PathVariable UUID chatroomId)
+ {
+ return chatHome
+ .getChatRoom(chatroomId)
+ .flatMapMany(chatroom -> chatroom
+ .getMessages()
+ .map(MessageTo::from));
}
@GetMapping("get/{chatroomId}")
- public Optional<ChatroomTo> get(@PathVariable UUID 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}")
{
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 chatroom,
+ ChatRoom chatroom,
String username,
Long messageId,
String text)
chatroom
.addMessage(
messageId,
- LocalDateTime.now(clock),
username,
text)
- .switchIfEmpty(chatroom.getMessage(username, messageId))
.map(message -> MessageTo.from(message));
}
{
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(
- Chatroom chatroom,
+ ChatRoom chatroom,
String username,
Long messageId)
{
.map(message -> MessageTo.from(message));
}
- @GetMapping(
- path = "listen/{chatroomId}",
- produces = MediaType.TEXT_EVENT_STREAM_VALUE)
- public Flux<MessageTo> listen(@PathVariable UUID chatroomId)
+ @GetMapping(path = "listen/{chatroomId}")
+ 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<MessageTo> listen(Chatroom chatroom)
+ private Flux<ServerSentEvent<MessageTo>> listen(ChatRoom chatroom)
{
return chatroom
.listen()
.log()
- .map(message -> MessageTo.from(message));
+ .map(message -> MessageTo.from(message))
+ .map(messageTo ->
+ ServerSentEvent
+ .builder(messageTo)
+ .id(messageTo.getSerial().toString())
+ .event("message")
+ .build());
+ }
+
+ @PostMapping("/store")
+ public void store()
+ {
+ storageStrategy.writeChatrooms(chatHome.getChatRooms());
}
}