X-Git-Url: https://juplo.de/gitweb/?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fde%2Fjuplo%2Fkafka%2Fchat%2Fbackend%2Fpersistence%2Fkafka%2FChatRoomChannel.java;h=8bbc82ec1befd788d1c9c65a620aeabdf8fc1082;hb=28a98bfdbed0cf56697bece5efbe6eb52f331611;hp=b08a50d2a6c78a3309e3bb347005af065ec9ff7e;hpb=716ffa6a7665496e614ce6a1671c8e49c562a4c2;p=demos%2Fkafka%2Fchat 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 b08a50d2..8bbc82ec 100644 --- 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 @@ -9,7 +9,6 @@ import org.apache.kafka.clients.consumer.ConsumerRecords; 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.RecordDeserializationException; import org.apache.kafka.common.errors.WakeupException; import reactor.core.publisher.Mono; @@ -23,10 +22,12 @@ import java.util.UUID; public class ChatRoomChannel implements Runnable { private final String topic; - private final Consumer consumer; - private final Producer producer; + private final Producer producer; + private final Consumer consumer; private final ShardingStrategy shardingStrategy; private final ChatMessageChannel chatMessageChannel; + private final Clock clock; + private final int bufferSize; private boolean running; @@ -42,66 +43,33 @@ public class ChatRoomChannel implements Runnable { try { - ConsumerRecords records = consumer.poll(Duration.ofMinutes(5)); + ConsumerRecords records = consumer.poll(Duration.ofMinutes(5)); log.info("Fetched {} messages", records.count()); - for (ConsumerRecord record : records) + for (ConsumerRecord record : records) { - UUID id = record.value().getId(); - String name = record.value().getName(); - chatRoomFactory.createChatRoom(id, name); + createChatRoom(record.value().toChatRoomInfo()); } } catch (WakeupException e) { - } - catch (RecordDeserializationException e) - { + log.info("Received WakeupException, exiting!"); + running = false; } } + + log.info("Exiting normally"); } - void createChatRoom() + + void createChatRoom(ChatRoomInfo chatRoomInfo) { - log.info("Creating ChatRoom with buffer-size {}", bufferSize); + 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); - int shard = shardingStrategy.selectShard(id); ChatRoom chatRoom = new ChatRoom(id, name, shard, clock, service, bufferSize); chatMessageChannel.putChatRoom(chatRoom); } - - Mono sendCreateChatRoomRequest( - UUID chatRoomId, - String name) - { - int shard = this.shardingStrategy.selectShard(chatRoomId); - ChatRoomTo chatRoomTo = ChatRoomTo.of(chatRoomId, name, shard); - return Mono.create(sink -> - { - ProducerRecord record = - new ProducerRecord<>( - topic, - shard, - chatRoomTo); - - producer.send(record, ((metadata, exception) -> - { - if (metadata != null) - { - log.info("Successfully send chreate-request for chat room: {}", chatRoomTo); - sink.success(chatRoomTo.toChatRoomInfo()); - } - else - { - // On send-failure - log.error( - "Could not send create-request for chat room (id={}, name={}): {}", - chatRoomId, - name, - exception); - sink.error(exception); - } - })); - }); - } }