NG

[demos/kafka/chat] / src / main / java / de / juplo / kafka / chat / backend / persistence / kafka / ChatRoomChannel.java

diff --git a/src/main/java/de/juplo/kafka/chat/backend/persistence/kafka/ChatRoomChannel.java b/src/main/java/de/juplo/kafka/chat/backend/persistence/kafka/ChatRoomChannel.java
index 8bbc82e..1c6ae91 100644 (file)
--- a/src/main/java/de/juplo/kafka/chat/backend/persistence/kafka/ChatRoomChannel.java
+++ b/src/main/java/de/juplo/kafka/chat/backend/persistence/kafka/ChatRoomChannel.java
@@ -10,11 +10,15 @@ import org.apache.kafka.clients.producer.Producer;
 import org.apache.kafka.clients.producer.ProducerRecord;
 import org.apache.kafka.common.TopicPartition;
 import org.apache.kafka.common.errors.WakeupException;
+import reactor.core.publisher.Flux;
 import reactor.core.publisher.Mono;
 
 import java.time.*;
+import java.util.HashMap;
 import java.util.List;
+import java.util.Map;
 import java.util.UUID;
+import java.util.stream.IntStream;
 
 
 @RequiredArgsConstructor
@@ -22,12 +26,8 @@ import java.util.UUID;
 public class ChatRoomChannel implements Runnable
 {
   private final String topic;
-  private final Producer<Integer, CreateChatRoomRequestTo> producer;
-  private final Consumer<Integer, CreateChatRoomRequestTo> consumer;
-  private final ShardingStrategy shardingStrategy;
-  private final ChatMessageChannel chatMessageChannel;
-  private final Clock clock;
-  private final int bufferSize;
+  private final Consumer<String, AbstractTo> consumer;
+  private final Map<UUID, ChatRoomInfo> chatrooms = new HashMap<>();
 
   private boolean running;
 
@@ -43,12 +43,24 @@ public class ChatRoomChannel implements Runnable
     {
       try
       {
-        ConsumerRecords<Integer, CreateChatRoomRequestTo> records = consumer.poll(Duration.ofMinutes(5));
+        ConsumerRecords<String, AbstractTo> records = consumer.poll(Duration.ofMinutes(5));
         log.info("Fetched {} messages", records.count());
 
-        for (ConsumerRecord<Integer, CreateChatRoomRequestTo> record : records)
+        for (ConsumerRecord<String, AbstractTo> record : records)
         {
-          createChatRoom(record.value().toChatRoomInfo());
+          switch (record.value().getType())
+          {
+            case CHATROOM_INFO:
+              createChatRoom((ChatRoomInfoTo) record.value());
+              break;
+
+            default:
+              log.debug(
+                  "Ignoring message for key {} with offset {}: {}",
+                  record.key(),
+                  record.offset(),
+                  record.value());
+          }
         }
       }
       catch (WakeupException e)
@@ -62,14 +74,14 @@ public class ChatRoomChannel implements Runnable
   }
 
 
-  void createChatRoom(ChatRoomInfo chatRoomInfo)
+  void createChatRoom(ChatRoomInfoTo chatRoomInfoTo)
+  {
+    ChatRoomInfo chatRoomInfo = chatRoomInfoTo.toChatRoomInfo();
+    chatrooms.put(chatRoomInfo.getId(), chatRoomInfo);
+  }
+
+  Flux<ChatRoomInfo> getChatRooms()
   {
-    UUID id = chatRoomInfo.getId();
-    String name = chatRoomInfo.getName();
-    int shard = chatRoomInfo.getShard();
-    log.info("Creating ChatRoom {} with buffer-size {}", id, bufferSize);
-    KafkaChatRoomService service = new KafkaChatRoomService(chatMessageChannel, id);
-    ChatRoom chatRoom = new ChatRoom(id, name, shard, clock, service, bufferSize);
-    chatMessageChannel.putChatRoom(chatRoom);
+    return Flux.fromIterable(chatrooms.values());
   }
 }