TMP

[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 b077a90..3be8f95 100644 (file)
--- a/src/main/java/de/juplo/kafka/ApplicationConfiguration.java
+++ b/src/main/java/de/juplo/kafka/ApplicationConfiguration.java
@@ -18,12 +18,20 @@ import java.util.concurrent.Executors;
 public class ApplicationConfiguration
 {
   @Bean
-  public WordcountRecordHandler wordcountRecordHandler(
+  public SumRecordHandler sumRecordHandler()
+  {
+    return new SumRecordHandler();
+  }
+
+  @Bean
+  public SumRebalanceListener sumRebalanceListener(
+      SumRecordHandler sumRecordHandler,
       PartitionStatisticsRepository repository,
       Consumer<String, String> consumer,
       ApplicationProperties properties)
   {
-    return new WordcountRecordHandler(
+    return new SumRebalanceListener(
+        sumRecordHandler,
         repository,
         properties.getClientId(),
         properties.getTopic(),
@@ -36,7 +44,8 @@ public class ApplicationConfiguration
   public EndlessConsumer<String, String> endlessConsumer(
       KafkaConsumer<String, String> kafkaConsumer,
       ExecutorService executor,
-      WordcountRecordHandler wordcountRecordHandler,
+      SumRebalanceListener sumRebalanceListener,
+      SumRecordHandler sumRecordHandler,
       ApplicationProperties properties)
   {
     return
@@ -45,7 +54,8 @@ public class ApplicationConfiguration
             properties.getClientId(),
             properties.getTopic(),
             kafkaConsumer,
-            wordcountRecordHandler);
+            sumRebalanceListener,
+            sumRecordHandler);
   }
 
   @Bean