WIP:sleep

[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 dac22a2..4459a79 100644 (file)
--- a/src/main/java/de/juplo/kafka/SimpleConsumer.java
+++ b/src/main/java/de/juplo/kafka/SimpleConsumer.java
@@ -20,6 +20,7 @@ public class SimpleConsumer
   private final String topic;
   private final Consumer<String, String> consumer;
 
+  private volatile boolean running = false;
   private long consumed = 0;
 
 
@@ -30,6 +31,7 @@ public class SimpleConsumer
     {
       log.info("{} - Subscribing to topic test", id);
       consumer.subscribe(Arrays.asList(topic));
+      running = true;
 
       while (true)
       {
@@ -63,9 +65,15 @@ public class SimpleConsumer
     }
     finally
     {
+      running = false;
       log.info("{} - Closing the KafkaConsumer", id);
       consumer.close();
       log.info("{}: Consumed {} messages in total, exiting!", id, consumed);
     }
   }
+
+  public boolean isRunning()
+  {
+    return running;
+  }
 }