juplo.de Git - demos/kafka/chat/blob - src/main/java/de/juplo/kafka/chat/backend/api/ChatBackendController.java

   1 package de.juplo.kafka.chat.backend.api;
   2
   3 import de.juplo.kafka.chat.backend.domain.Chatroom;
   4 import de.juplo.kafka.chat.backend.domain.ChatroomFactory;
   5 import lombok.RequiredArgsConstructor;
   6 import org.springframework.http.MediaType;
   7 import org.springframework.web.bind.annotation.*;
   8 import reactor.core.publisher.Flux;
   9 import reactor.core.publisher.Mono;
  10
  11 import java.time.Clock;
  12 import java.time.LocalDateTime;
  13 import java.util.Collection;
  14 import java.util.HashMap;
  15 import java.util.Map;
  16 import java.util.UUID;
  17
  18
  19 @RestController
  20 @RequiredArgsConstructor
  21 public class ChatBackendController
  22 {
  23   private final Map<UUID, Chatroom> chatrooms = new HashMap<>();
  24   private final ChatroomFactory factory;
  25   private final Clock clock;
  26
  27
  28   @PostMapping("create")
  29   public Chatroom create(@RequestBody String name)
  30   {
  31     Chatroom chatroom = factory.createChatroom(UUID.randomUUID(), name);
  32     chatrooms.put(chatroom.getId(), chatroom);
  33     return chatroom;
  34   }
  35
  36   @GetMapping("list")
  37   public Collection<Chatroom> list()
  38   {
  39     return chatrooms.values();
  40   }
  41
  42   @GetMapping("get/{chatroomId}")
  43   public Chatroom get(@PathVariable UUID chatroomId)
  44   {
  45     return chatrooms.get(chatroomId);
  46   }
  47
  48   @PutMapping("put/{chatroomId}/{username}/{messageId}")
  49   public Mono<MessageTo> put(
  50       @PathVariable UUID chatroomId,
  51       @PathVariable String username,
  52       @PathVariable Long messageId,
  53       @RequestBody String text)
  54   {
  55     Chatroom chatroom = chatrooms.get(chatroomId);
  56     return
  57         chatroom
  58             .addMessage(
  59                 messageId,
  60                 LocalDateTime.now(clock),
  61                 username,
  62                 text)
  63             .switchIfEmpty(chatroom.getMessage(username, messageId))
  64             .map(message -> MessageTo.from(message));
  65   }
  66
  67   @GetMapping("get/{chatroomId}/{username}/{messageId}")
  68   public Mono<MessageTo> get(
  69       @PathVariable UUID chatroomId,
  70       @PathVariable String username,
  71       @PathVariable Long messageId)
  72   {
  73     return
  74         chatrooms
  75             .get(chatroomId)
  76             .getMessage(username, messageId)
  77             .map(message -> MessageTo.from(message));
  78   }
  79
  80   @GetMapping(
  81       path = "listen/{chatroomId}",
  82       produces = MediaType.TEXT_EVENT_STREAM_VALUE)
  83   public Flux<MessageTo> listen(@PathVariable UUID chatroomId)
  84   {
  85     return chatrooms
  86         .get(chatroomId)
  87         .listen()
  88         .log()
  89         .map(message -> MessageTo.from(message));
  90   }
  91 }