From 30f050435819d84b87d55a5b64db2ac7a155449a Mon Sep 17 00:00:00 2001
From: Brandon Clayton <bclayton@usgs.gov>
Date: Thu, 16 Jan 2025 10:32:49 -0700
Subject: [PATCH] test

---
 .../java/gov/usgs/earthquake/nshmp/www/PrimingResource.java  | 5 ++++-
 src/main/java/gov/usgs/earthquake/nshmp/www/ServletUtil.java | 4 +++-
 2 files changed, 7 insertions(+), 2 deletions(-)

diff --git a/src/main/java/gov/usgs/earthquake/nshmp/www/PrimingResource.java b/src/main/java/gov/usgs/earthquake/nshmp/www/PrimingResource.java
index 8a8ce704..8da139a7 100644
--- a/src/main/java/gov/usgs/earthquake/nshmp/www/PrimingResource.java
+++ b/src/main/java/gov/usgs/earthquake/nshmp/www/PrimingResource.java
@@ -1,6 +1,7 @@
 package gov.usgs.earthquake.nshmp.www;
 
 import java.nio.file.Path;
+import java.util.logging.Logger;
 
 import org.crac.Context;
 import org.crac.Resource;
@@ -27,13 +28,15 @@ public class PrimingResource implements OrderedResource {
 
   @Override
   public void beforeCheckpoint(Context<? extends Resource> context) throws Exception {
+    Logger.getAnonymousLogger().info("Before checkpoint");
     ServletUtil.model(ServletUtil.loadModel(modelPath));
     updateParameter();
   }
 
   @Override
   public void afterRestore(Context<? extends Resource> context) throws Exception {
-    System.out.println("Reset exexutors");
+    Logger.getAnonymousLogger().info("After restore: reset execurors");
+    // Reset executors on restore as Shutdown event does not fire on AWS lambda
     ServletUtil.resetExecutors();
   }
 
diff --git a/src/main/java/gov/usgs/earthquake/nshmp/www/ServletUtil.java b/src/main/java/gov/usgs/earthquake/nshmp/www/ServletUtil.java
index e007dea5..5f75c7c2 100644
--- a/src/main/java/gov/usgs/earthquake/nshmp/www/ServletUtil.java
+++ b/src/main/java/gov/usgs/earthquake/nshmp/www/ServletUtil.java
@@ -91,6 +91,9 @@ public class ServletUtil {
         .create();
   }
 
+  /**
+   * Reset executors on AWS Lambda SnapStart restore
+   */
   static void resetExecutors() {
     CALC_EXECUTOR.shutdown();
     TASK_EXECUTOR.shutdown();
@@ -115,7 +118,6 @@ public class ServletUtil {
 
   @EventListener
   void startup(StartupEvent event) {
-    System.out.println("on start up");
     if (!awsRuntime.isPresent()) {
       HAZARD_MODEL = loadModel(modelPath);
     }
-- 
GitLab