X-Git-Url: https://juplo.de/gitweb/?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fde%2Fjuplo%2Fkafka%2FSimpleConsumer.java;h=8e557c2b3071070c3bf29862ef75e6ce767f1647;hb=e0d3872018d67445881d1da2d7fbfed3909d0649;hp=0e376864d1958d477117a5b05be13e0515a2de75;hpb=5b9f4cd21a87b03cb1c432e9965fb0082ab05dd3;p=demos%2Fkafka%2Ftraining

diff --git a/src/main/java/de/juplo/kafka/SimpleConsumer.java b/src/main/java/de/juplo/kafka/SimpleConsumer.java
index 0e37686..8e557c2 100644
--- a/src/main/java/de/juplo/kafka/SimpleConsumer.java
+++ b/src/main/java/de/juplo/kafka/SimpleConsumer.java
@@ -9,11 +9,12 @@ import org.apache.kafka.common.errors.WakeupException;
 
 import java.time.Duration;
 import java.util.Arrays;
+import java.util.concurrent.Callable;
 
 
 @Slf4j
 @RequiredArgsConstructor
-public class SimpleConsumer implements Runnable
+public class SimpleConsumer implements Callable<Integer>
 {
   private final String id;
   private final String topic;
@@ -23,7 +24,7 @@ public class SimpleConsumer implements Runnable
 
 
   @Override
-  public void run()
+  public Integer call()
   {
     try
     {
@@ -38,27 +39,25 @@ public class SimpleConsumer implements Runnable
         log.info("{} - Received {} messages", id, records.count());
         for (ConsumerRecord<String, String> record : records)
         {
-          consumed++;
-          log.info(
-              "{} - {}: {}/{} - {}={}",
-              id,
-              record.offset(),
-              record.topic(),
-              record.partition(),
-              record.key(),
-              record.value()
-          );
+          handleRecord(
+            record.topic(),
+            record.partition(),
+            record.offset(),
+            record.key(),
+            record.value());
         }
       }
     }
     catch(WakeupException e)
     {
       log.info("{} - Consumer was signaled to finish its work", id);
+      return 0;
     }
     catch(Exception e)
     {
       log.error("{} - Unexpected error: {}, unsubscribing!", id, e.toString());
       consumer.unsubscribe();
+      return 1;
     }
     finally
     {
@@ -67,4 +66,15 @@ public class SimpleConsumer implements Runnable
       log.info("{}: Consumed {} messages in total, exiting!", id, consumed);
     }
   }
+
+  private void handleRecord(
+    String topic,
+    Integer partition,
+    Long offset,
+    String key,
+    Message value)
+  {
+    consumed++;
+    log.info("{} - {}: {}/{} - {}={}", id, offset, topic, partition, key, value);
+  }
 }