apply shutdown hook to airbyte server to allow graceful termination (#…

…15934)
airbytehq · Aug 24, 2022 · 883d0ee · 883d0ee
1 parent 6f09860
commit 883d0ee
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/airbyte-server/src/main/java/io/airbyte/server/ServerApp.java b/airbyte-server/src/main/java/io/airbyte/server/ServerApp.java
@@ -133,6 +133,15 @@ public void start() throws Exception {
     final String banner = MoreResources.readResource("banner/banner.txt");
     LOGGER.info(banner + String.format("Version: %s\n", airbyteVersion.serialize()));
     server.join();
+
+    Runtime.getRuntime().addShutdownHook(new Thread(() -> {
+      try {
+        server.stop();
+      } catch (Exception ex) {
+        // silently fail at this stage because server is terminating.
+        LOGGER.warn("exception: " + ex);
+      }
+    }));
   }
 
   private static void assertDatabasesReady(final Configs configs,