X-Git-Url: http://juplo.de/gitweb/?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fde%2Fjuplo%2Fkafka%2Fchat%2Fbackend%2Fpersistence%2Fkafka%2FChatRoomActiveMessageHandlingStrategy.java;fp=src%2Fmain%2Fjava%2Fde%2Fjuplo%2Fkafka%2Fchat%2Fbackend%2Fpersistence%2Fkafka%2FChatRoomActiveMessageHandlingStrategy.java;h=69b6fe9ba2cd92b111a133fdd62c30e8b774a802;hb=45323044eba7653812c0baafa91be3855d003e4a;hp=8eac9900b16874f48e9325b5f05a48f1ad5405df;hpb=7d8b0b3c36669b5e18c1f43134e2e1e55a4e6f65;p=demos%2Fkafka%2Fchat diff --git a/src/main/java/de/juplo/kafka/chat/backend/persistence/kafka/ChatRoomActiveMessageHandlingStrategy.java b/src/main/java/de/juplo/kafka/chat/backend/persistence/kafka/ChatRoomActiveMessageHandlingStrategy.java index 8eac9900..69b6fe9b 100644 --- a/src/main/java/de/juplo/kafka/chat/backend/persistence/kafka/ChatRoomActiveMessageHandlingStrategy.java +++ b/src/main/java/de/juplo/kafka/chat/backend/persistence/kafka/ChatRoomActiveMessageHandlingStrategy.java @@ -1,19 +1,70 @@ package de.juplo.kafka.chat.backend.persistence.kafka; import de.juplo.kafka.chat.backend.domain.Message; +import de.juplo.kafka.chat.backend.domain.MessageMutationException; import lombok.RequiredArgsConstructor; +import lombok.extern.log4j.Log4j; +import org.apache.kafka.clients.producer.Producer; +import org.apache.kafka.clients.producer.ProducerRecord; import org.apache.kafka.common.TopicPartition; +import reactor.core.publisher.Mono; + +import java.time.LocalDateTime; +import java.time.ZoneOffset; +import java.util.UUID; @RequiredArgsConstructor +@Log4j class ChatRoomActiveMessageHandlingStrategy implements MessageHandlingStrategy { + private final Producer producer; private final TopicPartition tp; + private final UUID chatRoomId; + private final ZoneOffset zoneOffset; + private final KafkaChatRoomService chatRoomService; + + + @Override + public Mono persistMessage( + Message.MessageKey key, + LocalDateTime timestamp, + String text) + { + return Mono.create(sink -> + { + ProducerRecord record = + new ProducerRecord<>( + tp.topic(), + tp.partition(), + timestamp.toEpochSecond(zoneOffset), + chatRoomId.toString(), + MessageTo.of(key.getUsername(), key.getMessageId(), text)); + + producer.send(record, ((metadata, exception) -> + { + if (metadata != null) + { + // On successful send + { + // Emit new message + message = new Message(key, metadata.offset(), timestamp, text); + messages.put(message.getKey(), message); + } + + sink.success(); + } + else + { + // On send-failure + sink.error(exception); + } + })); + }); + } @Override public MessageHandlingStrategy handleMessage(Message message) { - chatrooms[tp.partition()].put() - return this; } }