X-Git-Url: http://juplo.de/gitweb/?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fde%2Fjuplo%2Fkafka%2FApplicationHealthIndicator.java;h=0466df4cc2b77ad408e9d414735b4b2b8fe657e8;hb=refs%2Fheads%2Fsumup-adder--springified;hp=df4e653cfbb311bcdb8ee5bf8ec9facb0a75be94;hpb=eb0ead4c178df50cdaf2197cf4e63a1fe709e852;p=demos%2Fkafka%2Ftraining

diff --git a/src/main/java/de/juplo/kafka/ApplicationHealthIndicator.java b/src/main/java/de/juplo/kafka/ApplicationHealthIndicator.java
index df4e653..0466df4 100644
--- a/src/main/java/de/juplo/kafka/ApplicationHealthIndicator.java
+++ b/src/main/java/de/juplo/kafka/ApplicationHealthIndicator.java
@@ -3,30 +3,21 @@ package de.juplo.kafka;
 import lombok.RequiredArgsConstructor;
 import org.springframework.boot.actuate.health.Health;
 import org.springframework.boot.actuate.health.HealthIndicator;
-import org.springframework.stereotype.Component;
+import org.springframework.kafka.config.KafkaListenerEndpointRegistry;
 
 
-@Component
 @RequiredArgsConstructor
 public class ApplicationHealthIndicator implements HealthIndicator
 {
-  private final EndlessConsumer<String, String> consumer;
+  private final String id;
+  private final KafkaListenerEndpointRegistry registry;
 
 
   @Override
   public Health health()
   {
-    try
-    {
-      return consumer
-          .exitStatus()
-          .map(Health::down)
-          .orElse(Health.outOfService())
-          .build();
-    }
-    catch (IllegalStateException e)
-    {
-      return Health.up().build();
-    }
+    return registry.getListenerContainer(id).isRunning()
+        ? Health.up().build()
+        : Health.down().build();
   }
 }