Add shutdown listener and block queue before exit

genepi · Oct 1, 2024 · 3a69297 · 3a69297
1 parent 80ccc17
commit 3a69297
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/src/main/java/cloudgene/mapred/server/Application.java b/src/main/java/cloudgene/mapred/server/Application.java
@@ -6,9 +6,12 @@
 import java.util.List;
 import java.util.Map;
 
+import cloudgene.mapred.jobs.AbstractJob;
 import cloudgene.mapred.jobs.engine.handler.IJobErrorHandler;
 import cloudgene.mapred.jobs.engine.handler.JobErrorHandlerFactory;
 import cloudgene.mapred.util.Configuration;
+import io.micronaut.runtime.event.ApplicationShutdownEvent;
+import io.micronaut.runtime.event.annotation.EventListener;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
@@ -118,6 +121,14 @@ public Application() throws Exception {
 
 	}
 
+	@EventListener
+	public void stop(final ApplicationShutdownEvent event) throws SQLException {
+		System.out.println("Shutting down Cloudgene...");
+		log.info("Shutting down Cloudgene...");
+		engine.block();
+		database.disconnect();
+	}
+
 	public WorkflowEngine getWorkflowEngine() {
 		return engine;
 	}