feat: add shutdown hook (#365)

netease-im · Dec 19, 2024 · 9ff36e7 · 9ff36e7
1 parent 2460656
commit 9ff36e7
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 0 deletions.
diff --git a/...com/netease/nim/camellia/delayqueue/server/springboot/CamelliaDelayQueueServerStatus.java b/...com/netease/nim/camellia/delayqueue/server/springboot/CamelliaDelayQueueServerStatus.java
@@ -39,6 +39,10 @@ public void run() {
         }
     }
 
+    static {
+        Runtime.getRuntime().addShutdownHook(new Thread(CamelliaDelayQueueServerStatus::invokeOfflineCallback));
+    }
+
     private static Status status = Status.ONLINE;
     private static long lastUseTime = System.currentTimeMillis();
 

diff --git a/...-key-server/src/main/java/com/netease/nim/camellia/hot/key/server/netty/ServerStatus.java b/...-key-server/src/main/java/com/netease/nim/camellia/hot/key/server/netty/ServerStatus.java
@@ -26,6 +26,10 @@ public enum Status {
         ;
     }
 
+    static {
+        Runtime.getRuntime().addShutdownHook(new Thread(ServerStatus::invokeOfflineCallback));
+    }
+
     public static Status getStatus() {
         return status;
     }

diff --git a/...dis-proxy-core/src/main/java/com/netease/nim/camellia/redis/proxy/netty/ServerStatus.java b/...dis-proxy-core/src/main/java/com/netease/nim/camellia/redis/proxy/netty/ServerStatus.java
@@ -26,6 +26,10 @@ public enum Status {
         ;
     }
 
+    static {
+        Runtime.getRuntime().addShutdownHook(new Thread(ServerStatus::invokeOfflineCallback));
+    }
+
     public static Status getStatus() {
         return status;
     }