From 604f8c824ea52233c6c6dfa72f786eabceb5a14c Mon Sep 17 00:00:00 2001
From: Himanshu Gupta <g.himanshu@gmail.com>
Date: Sat, 26 Aug 2017 12:50:43 -0500
Subject: [PATCH] kafkaIndexTask unannounce service in final block

---
 .../main/java/io/druid/indexing/kafka/KafkaIndexTask.java   | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/extensions-core/kafka-indexing-service/src/main/java/io/druid/indexing/kafka/KafkaIndexTask.java b/extensions-core/kafka-indexing-service/src/main/java/io/druid/indexing/kafka/KafkaIndexTask.java
index a496a0bb16b7..f81515ba18ea 100644
--- a/extensions-core/kafka-indexing-service/src/main/java/io/druid/indexing/kafka/KafkaIndexTask.java
+++ b/extensions-core/kafka-indexing-service/src/main/java/io/druid/indexing/kafka/KafkaIndexTask.java
@@ -613,10 +613,10 @@ public String apply(DataSegment input)
       if (chatHandlerProvider.isPresent()) {
         chatHandlerProvider.get().unregister(getId());
       }
-    }
 
-    toolbox.getDruidNodeAnnouncer().unannounce(discoveryDruidNode);
-    toolbox.getDataSegmentServerAnnouncer().unannounce();
+      toolbox.getDruidNodeAnnouncer().unannounce(discoveryDruidNode);
+      toolbox.getDataSegmentServerAnnouncer().unannounce();
+    }
 
     return success();
   }