Fixed SonarCloud warnings

javadev · Oct 30, 2024 · 6293ba9 · 6293ba9
1 parent ad341d9
commit 6293ba9
Showing 1 changed file with 9 additions and 7 deletions.
diff --git a/src/main/java/com/github/underscore/Underscore.java b/src/main/java/com/github/underscore/Underscore.java
@@ -2098,13 +2098,14 @@ public static <T> java.util.concurrent.ScheduledFuture<T> delay(
             final Supplier<T> function, final int delayMilliseconds) {
         final java.util.concurrent.ScheduledExecutorService scheduler =
                 java.util.concurrent.Executors.newSingleThreadScheduledExecutor();
-        final java.util.concurrent.ScheduledFuture<T> future =
-                scheduler.schedule(
-                        function::get,
-                        delayMilliseconds,
-                        java.util.concurrent.TimeUnit.MILLISECONDS);
-        scheduler.shutdown();
-        return future;
+        try {
+            return scheduler.schedule(
+                    function::get,
+                    delayMilliseconds,
+                    java.util.concurrent.TimeUnit.MILLISECONDS);
+        } finally {
+            scheduler.shutdown();
+        }
     }
 
     public static <T> java.util.concurrent.ScheduledFuture<T> defer(final Supplier<T> function) {
@@ -3624,6 +3625,7 @@ public static <T> java.util.concurrent.ScheduledFuture setInterval(
             final Supplier<T> function, final int delayMilliseconds) {
         final java.util.concurrent.ScheduledExecutorService scheduler =
                 java.util.concurrent.Executors.newSingleThreadScheduledExecutor();
+        Runtime.getRuntime().addShutdownHook(new Thread(scheduler::shutdown));
         return scheduler.scheduleAtFixedRate(
                 function::get,
                 delayMilliseconds,