From 583230d76b2a3a2014f3f070fc8d02149cd12a7a Mon Sep 17 00:00:00 2001
From: lprimak <lenny@flowlogix.com>
Date: Wed, 21 Aug 2024 22:09:24 -0500
Subject: [PATCH] bugfix: shut down executor threads due to PREPARE_SHUTDOWN
 event, instead of SERVER_SHUTDOWN, thus avoiding exceptions when events run
 during server shutdown

---
 .../PayaraExecutorService.java                | 19 ++++++++++++-------
 1 file changed, 12 insertions(+), 7 deletions(-)

diff --git a/nucleus/payara-modules/payara-executor-service/src/main/java/fish/payara/nucleus/executorservice/PayaraExecutorService.java b/nucleus/payara-modules/payara-executor-service/src/main/java/fish/payara/nucleus/executorservice/PayaraExecutorService.java
index 79c5bd48971..92c3be3069c 100644
--- a/nucleus/payara-modules/payara-executor-service/src/main/java/fish/payara/nucleus/executorservice/PayaraExecutorService.java
+++ b/nucleus/payara-modules/payara-executor-service/src/main/java/fish/payara/nucleus/executorservice/PayaraExecutorService.java
@@ -1,7 +1,7 @@
 /*
  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.
  *
- *    Copyright (c) [2017-2022] Payara Foundation and/or its affiliates. All rights reserved.
+ *    Copyright (c) [2017-2024] Payara Foundation and/or its affiliates. All rights reserved.
  *
  *     The contents of this file are subject to the terms of either the GNU
  *     General Public License Version 2 only ("GPL") or the Common Development
@@ -137,8 +137,10 @@ public void event(Event event) {
             if (null == threadPoolExecutor) {
                 initialiseThreadPools();
             }
-        } else if (event.is(EventTypes.SERVER_SHUTDOWN)) {
+        } else if (event.is(EventTypes.PREPARE_SHUTDOWN)) {
             terminateThreadPools();
+        } else if (event.is(EventTypes.SERVER_SHUTDOWN)) {
+            awaitTerminationOfThreadPools();
         }
     }
 
@@ -149,7 +151,13 @@ private void terminateThreadPools() {
         }
         threadPoolExecutor.shutdown();
         scheduledThreadPoolExecutor.shutdown();
+    }
 
+    private void awaitTerminationOfThreadPools() {
+        if (threadPoolExecutor == null) {
+            // we didn't initialize yet
+            return;
+        }
         // Wait until the schedulers actually terminate
         try {
           threadPoolExecutor.awaitTermination(5, TimeUnit.SECONDS);
@@ -189,9 +197,6 @@ private void initialiseThreadPools() {
                             Logger.getLogger(PayaraExecutorService.class.getName()).log(SEVERE, "Uncaught exception in Payara Scheduled Executor thread ", thrwbl));
                         return t;
             });
-            scheduledThreadPoolExecutor.setRemoveOnCancelPolicy(true);
-            
-            
         } else {
             int threadPoolExecutorQueueSize = Integer.valueOf(payaraExecutorServiceConfiguration.getThreadPoolExecutorQueueSize());
             
@@ -224,9 +229,9 @@ public void uncaughtException(Thread thread, Throwable thrwbl) {
                         });
                         return t;
             });
-            scheduledThreadPoolExecutor.setRemoveOnCancelPolicy(true);
         }
-        
+        scheduledThreadPoolExecutor.setRemoveOnCancelPolicy(true);
+        scheduledThreadPoolExecutor.setExecuteExistingDelayedTasksAfterShutdownPolicy(false);
     }
 
     public <T> Future<T> submit(Callable<T> task) {