Refaktorisierung für Tests - Record-Handler als Bean konfigurierbar

[demos/kafka/training] / src / main / java / de / juplo / kafka / ApplicationConfiguration.java

diff --git a/src/main/java/de/juplo/kafka/ApplicationConfiguration.java b/src/main/java/de/juplo/kafka/ApplicationConfiguration.java
index 6e04453..f228d85 100644 (file)
--- a/src/main/java/de/juplo/kafka/ApplicationConfiguration.java
+++ b/src/main/java/de/juplo/kafka/ApplicationConfiguration.java
@@ -1,5 +1,6 @@
 package de.juplo.kafka;
 
+import org.apache.kafka.clients.consumer.ConsumerRecord;
 import org.apache.kafka.clients.consumer.KafkaConsumer;
 import org.apache.kafka.common.serialization.LongDeserializer;
 import org.apache.kafka.common.serialization.StringDeserializer;
@@ -8,28 +9,44 @@ import org.springframework.context.annotation.Bean;
 import org.springframework.context.annotation.Configuration;
 
 import java.util.Properties;
+import java.util.concurrent.ExecutorService;
 import java.util.concurrent.Executors;
+import java.util.function.Consumer;
 
 
 @Configuration
 @EnableConfigurationProperties(ApplicationProperties.class)
 public class ApplicationConfiguration
 {
+  @Bean
+  public Consumer<ConsumerRecord<String, String>> consumer()
+  {
+    return (record) ->
+    {
+      // Handle record
+    };
+  }
+
   @Bean
   public EndlessConsumer endlessConsumer(
       KafkaConsumer<String, String> kafkaConsumer,
+      ExecutorService executor,
+      Consumer<ConsumerRecord<String, String>> handler,
       ApplicationProperties properties)
   {
-    EndlessConsumer consumer =
+    return
         new EndlessConsumer(
-            Executors.newFixedThreadPool(1),
+            executor,
             properties.getClientId(),
             properties.getTopic(),
-            kafkaConsumer);
-
-    consumer.start();
+            kafkaConsumer,
+            handler);
+  }
 
-    return consumer;
+  @Bean
+  public ExecutorService executor()
+  {
+    return Executors.newSingleThreadExecutor();
   }
 
   @Bean(destroyMethod = "close")