WIP:TaskExecutor

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

diff --git a/src/main/java/de/juplo/kafka/SimpleConsumer.java b/src/main/java/de/juplo/kafka/SimpleConsumer.java
index 040e24b..64d5176 100644 (file)
--- a/src/main/java/de/juplo/kafka/SimpleConsumer.java
+++ b/src/main/java/de/juplo/kafka/SimpleConsumer.java
@@ -6,41 +6,44 @@ import org.apache.kafka.clients.consumer.Consumer;
 import org.apache.kafka.clients.consumer.ConsumerRecord;
 import org.apache.kafka.clients.consumer.ConsumerRecords;
 import org.apache.kafka.common.errors.WakeupException;
+import org.springframework.core.task.TaskExecutor;
 
 import java.time.Duration;
 import java.util.Arrays;
-import java.util.concurrent.ExecutorService;
 
 
 @Slf4j
 @RequiredArgsConstructor
 public class SimpleConsumer implements Runnable
 {
-  private final ExecutorService executor;
+  private final TaskExecutor taskExecutor;
   private final String id;
   private final String topic;
-  private final Consumer<String, Message> consumer;
+  private final Consumer<String, String> consumer;
 
-  private volatile boolean running = false;
   private long consumed = 0;
 
 
+  public void start()
+  {
+    taskExecutor.execute(this);
+  }
+
   @Override
   public void run()
   {
     try
     {
       log.info("{} - Subscribing to topic test", id);
-      consumer.subscribe(Arrays.asList("test"));
-      running = true;
+      consumer.subscribe(Arrays.asList(topic));
 
       while (true)
       {
-        ConsumerRecords<String, Message> records =
+        ConsumerRecords<String, String> records =
             consumer.poll(Duration.ofSeconds(1));
 
         log.info("{} - Received {} messages", id, records.count());
-        for (ConsumerRecord<String, Message> record : records)
+        for (ConsumerRecord<String, String> record : records)
         {
           consumed++;
           log.info(
@@ -66,7 +69,6 @@ public class SimpleConsumer implements Runnable
     }
     finally
     {
-      running = false;
       log.info("{} - Closing the KafkaConsumer", id);
       consumer.close();
       log.info("{}: Consumed {} messages in total, exiting!", id, consumed);