chore: adding finally to shutdown hook

littlehorse-enterprises · Jan 31, 2024 · 0569af8 · 0569af8
1 parent 138c4da
commit 0569af8
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/canary/src/main/java/io/littlehorse/canary/Main.java b/canary/src/main/java/io/littlehorse/canary/Main.java
@@ -71,8 +71,13 @@ private static void initiateKafkaTopicBootstrap(CanaryConfig config, CountDownLa
     private static void addShutdownHook(Bootstrap bootstrap, CountDownLatch latch) {
         Runtime.getRuntime().addShutdownHook(new Thread(() -> {
             log.trace("{} shutdown process started", bootstrap.getClass().getSimpleName());
-            bootstrap.shutdown();
-            latch.countDown();
+            try {
+                bootstrap.shutdown();
+            } catch (Exception e) {
+                log.error("Error in ShutdownHook '{}'", bootstrap.getClass().getSimpleName(), e);
+            } finally {
+                latch.countDown();
+            }
         }));
     }
 }