1 package de.juplo.kafka.chat.backend.implementation.kafka;
3 import de.juplo.kafka.chat.backend.ChatBackendProperties;
4 import de.juplo.kafka.chat.backend.domain.ShardingPublisherStrategy;
5 import lombok.extern.slf4j.Slf4j;
6 import org.apache.kafka.clients.producer.ProducerRecord;
7 import org.apache.kafka.common.TopicPartition;
8 import org.springframework.boot.context.properties.EnableConfigurationProperties;
9 import org.springframework.boot.test.context.TestConfiguration;
10 import org.springframework.context.annotation.Bean;
11 import org.springframework.context.annotation.Import;
12 import org.springframework.kafka.core.KafkaTemplate;
13 import org.springframework.kafka.support.SendResult;
14 import reactor.core.publisher.Mono;
16 import java.time.Clock;
17 import java.util.List;
21 public class KafkaTestUtils
24 @EnableConfigurationProperties(ChatBackendProperties.class)
25 @Import(KafkaServicesConfiguration.class)
26 public static class KafkaTestConfiguration
29 public ShardingPublisherStrategy shardingPublisherStrategy()
31 return shard -> Mono.just("MOCKED!");
35 public WorkAssignor dataChannelWorkAssignor(
36 ChatBackendProperties properties,
37 DataChannel dataChannel)
41 List<TopicPartition> assignedPartitions =
42 List.of(new TopicPartition(properties.getKafka().getDataChannelTopic(), 2));
43 consumer.assign(assignedPartitions);
44 dataChannel.onPartitionsAssigned(assignedPartitions);
51 return Clock.systemDefaultZone();
56 public static void sendAndLoadStoredData(
57 KafkaTemplate<String, String> messageTemplate,
60 ConsumerTaskRunner consumerTaskRunner)
62 send(messageTemplate, infoTopic, "5c73531c-6fc4-426c-adcb-afc5c140a0f7","{ \"id\": \"5c73531c-6fc4-426c-adcb-afc5c140a0f7\", \"shard\": 2, \"name\": \"FOO\" }", "event_chatroom_created");
63 send(messageTemplate, dataTopic, "5c73531c-6fc4-426c-adcb-afc5c140a0f7","{ \"id\" : 1, \"user\" : \"peter\", \"text\" : \"Hallo, ich heiße Peter!\" }", "event_chatmessage_received");
64 send(messageTemplate, dataTopic, "5c73531c-6fc4-426c-adcb-afc5c140a0f7","{ \"id\" : 1, \"user\" : \"ute\", \"text\" : \"Ich bin Ute...\" }", "event_chatmessage_received");
65 send(messageTemplate, dataTopic, "5c73531c-6fc4-426c-adcb-afc5c140a0f7","{ \"id\" : 2, \"user\" : \"peter\", \"text\" : \"Willst du mit mir gehen?\" }", "event_chatmessage_received");
66 send(messageTemplate, dataTopic, "5c73531c-6fc4-426c-adcb-afc5c140a0f7","{ \"id\" : 1, \"user\" : \"klaus\", \"text\" : \"Ja? Nein? Vielleicht??\" }", "event_chatmessage_received");
68 consumerTaskRunner.executeConsumerTasks();
71 private static void send(
72 KafkaTemplate<String, String> kafkaTemplate,
78 ProducerRecord<String, String> record = new ProducerRecord<>(topic, key, value);
79 record.headers().add("__TypeId__", typeId.getBytes());
80 SendResult<String, String> result = kafkaTemplate.send(record).join();
85 new TopicPartition(result.getRecordMetadata().topic(), result.getRecordMetadata().partition()));
88 public static void joinConsumerTasks(ConsumerTaskRunner consumerTaskRunner) throws InterruptedException
90 consumerTaskRunner.joinConsumerTasks();