Add a shutdown hook for killing apps on Ctrl-C

cloudera · Jul 5, 2012 · a3eda2d · a3eda2d
1 parent 8564321
commit a3eda2d
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/java/client/src/main/java/com/cloudera/kitten/client/service/YarnClientServiceImpl.java b/java/client/src/main/java/com/cloudera/kitten/client/service/YarnClientServiceImpl.java
@@ -102,6 +102,16 @@ protected void startUp() {
     appContext.setPriority(ContainerLaunchContextFactory.createPriority(appMasterParams.getPriority()));
     submitApplication(appContext);
 
+    // Make sure we stop the application in the case that it isn't done already.
+    Runtime.getRuntime().addShutdownHook(new Thread() {
+      @Override
+      public void run() {
+        if (YarnClientServiceImpl.this.isRunning()) {
+          YarnClientServiceImpl.this.stop();
+        }
+      }
+    });
+
     stopwatch.start();
   }