add shutdown hook to execute shutdown sequence on jvm shutdown (#103)

flosell · Apr 3, 2016 · f5f5029 · f5f5029
1 parent 856ac16
commit f5f5029
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/src/clj/lambdacd/core.clj b/src/clj/lambdacd/core.clj
@@ -10,8 +10,10 @@
 
 (defn- add-shutdown-sequence! [ctx]
   (doto (Runtime/getRuntime)
-    (.addShutdownHook (fn []
-                        ((:shutdown-sequence (:config ctx)) ctx)))))
+    (.addShutdownHook (Thread. (fn []
+                                 (log/info "Shutting down LambdaCD...")
+                                 ((:shutdown-sequence (:config ctx)) ctx)))))
+  ctx)
 
 (def default-shutdown-sequence
   (fn [ctx]
@@ -35,7 +37,8 @@
                                     (event-bus/initialize-event-bus)
                                     (running-builds-tracking/initialize-running-builds-tracking)
                                     (assoc :pipeline-state-component pipeline-state-component)
-                                    (initialize-pipeline-state-updater))]
+                                    (initialize-pipeline-state-updater)
+                                    (add-shutdown-sequence!))]
      {:context                context
       :pipeline-def           pipeline-def})))