X-Git-Url: https://juplo.de/gitweb/?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fde%2Fjuplo%2Fkafka%2Fchat%2Fbackend%2Fdomain%2FChatroom.java;h=966a28eeb13a331196bda881647bb2c170f07c2a;hb=6c96b0b685b7fece1761f9471d5f0689171956ed;hp=62c97327c4409dcccf283cf198d3ad2fde70c093;hpb=f0113ff34d7a4147ce54116680b611061244f39a;p=demos%2Fkafka%2Fchat

diff --git a/src/main/java/de/juplo/kafka/chat/backend/domain/Chatroom.java b/src/main/java/de/juplo/kafka/chat/backend/domain/Chatroom.java
index 62c97327..966a28ee 100644
--- a/src/main/java/de/juplo/kafka/chat/backend/domain/Chatroom.java
+++ b/src/main/java/de/juplo/kafka/chat/backend/domain/Chatroom.java
@@ -1,8 +1,6 @@
 package de.juplo.kafka.chat.backend.domain;
 
 import lombok.Getter;
-import lombok.RequiredArgsConstructor;
-import lombok.Value;
 import lombok.extern.slf4j.Slf4j;
 import reactor.core.publisher.Flux;
 import reactor.core.publisher.Mono;
@@ -10,10 +8,8 @@ import reactor.core.publisher.Sinks;
 
 import java.time.LocalDateTime;
 import java.util.*;
-import java.util.stream.Stream;
 
 
-@RequiredArgsConstructor
 @Slf4j
 public class Chatroom
 {
@@ -21,50 +17,37 @@ public class Chatroom
   private final UUID id;
   @Getter
   private final String name;
-  private final LinkedHashMap<MessageKey, Message> messages = new LinkedHashMap<>();
-  private final Sinks.Many<Message> sink = Sinks.many().multicast().onBackpressureBuffer();
+  private final PersistenceStrategy persistence;
+  private final Sinks.Many<Message> sink;
 
-  synchronized public Mono<Message> addMessage(
-      Long id,
-      LocalDateTime timestamp,
-      String user,
-      String text)
+  public Chatroom(
+      UUID id,
+      String name,
+      PersistenceStrategy persistence,
+      int bufferSize)
   {
-    return persistMessage(id, timestamp, user, text)
-        .doOnNext(message -> sink.tryEmitNext(message).orThrow());
+    this.id = id;
+    this.name = name;
+    this.persistence = persistence;
+    this.sink = Sinks.many().multicast().onBackpressureBuffer(bufferSize);
   }
 
-  private Mono<Message> persistMessage(
+
+  synchronized public Mono<Message> addMessage(
       Long id,
       LocalDateTime timestamp,
       String user,
       String text)
   {
-    Message message = new Message(id, (long)messages.size(), timestamp, user, text);
-
-    MessageKey key = new MessageKey(user, id);
-    Message existing = messages.get(key);
-    if (existing != null)
-    {
-      log.info("Message with key {} already exists; {}", key, existing);
-      if (!message.equals(existing))
-        throw new MessageMutationException(message, existing);
-      return Mono.empty();
-    }
-
-    messages.put(key, message);
-    return Mono
-        .fromSupplier(() -> message)
-        .log();
+    return persistence
+        .persistMessage(Message.MessageKey.of(user, id), timestamp, text)
+        .doOnNext(message -> sink.tryEmitNext(message).orThrow());
   }
 
+
   public Mono<Message> getMessage(String username, Long messageId)
   {
-    return Mono.fromSupplier(() ->
-    {
-      MessageKey key = MessageKey.of(username, messageId);
-      return messages.get(key);
-    });
+    return persistence.getMessage(Message.MessageKey.of(username, messageId));
   }
 
   public Flux<Message> listen()
@@ -72,19 +55,13 @@ public class Chatroom
     return sink.asFlux();
   }
 
-  public Stream<Message> getMessages(long firstMessage)
+  public Flux<Message> getMessages()
   {
-    return messages
-        .values()
-        .stream()
-        .filter(message -> message.getSerialNumber() >= firstMessage);
+    return getMessages(0, Long.MAX_VALUE);
   }
 
-
-  @Value(staticConstructor = "of")
-  static class MessageKey
+  public Flux<Message> getMessages(long first, long last)
   {
-    String username;
-    Long messageId;
+    return persistence.getMessages(first, last);
   }
 }