From d5fd5018cd1b73ef74a928dc46d1b382ab892200 Mon Sep 17 00:00:00 2001
From: Robert von Burg <eitch@eitchnet.ch>
Date: Tue, 3 Mar 2020 15:59:09 +0100
Subject: [PATCH] [Fix] ExecutorPool should use different pools for
 scheduledExecutors and ExecutorService

---
 .../java/li/strolch/utils/ExecutorPool.java   | 34 +++++++++++++------
 1 file changed, 23 insertions(+), 11 deletions(-)

diff --git a/li.strolch.utils/src/main/java/li/strolch/utils/ExecutorPool.java b/li.strolch.utils/src/main/java/li/strolch/utils/ExecutorPool.java
index a7e3fa405..e1588b2f1 100644
--- a/li.strolch.utils/src/main/java/li/strolch/utils/ExecutorPool.java
+++ b/li.strolch.utils/src/main/java/li/strolch/utils/ExecutorPool.java
@@ -18,9 +18,11 @@ public class ExecutorPool {
 	private static final Logger logger = LoggerFactory.getLogger(ExecutorPool.class);
 
 	private Map<String, ExecutorService> executors;
+	private Map<String, ScheduledExecutorService> scheduledExecutors;
 
 	public ExecutorPool() {
 		this.executors = Collections.synchronizedMap(new HashMap<>());
+		this.scheduledExecutors = Collections.synchronizedMap(new HashMap<>());
 	}
 
 	public ExecutorService getExecutor(String poolName) {
@@ -35,7 +37,7 @@ public class ExecutorPool {
 
 	public ScheduledExecutorService getScheduledExecutor(String poolName) {
 		DBC.PRE.assertNotEmpty("poolName must be set!", poolName);
-		return (ScheduledExecutorService) this.executors
+		return this.scheduledExecutors
 				.computeIfAbsent(poolName, p -> newScheduledThreadPool(4, new NamedThreadPoolFactory(p)));
 	}
 
@@ -45,16 +47,26 @@ public class ExecutorPool {
 			logger.info("Shutting down executor pool " + poolName);
 			ExecutorService executor = this.executors.get(poolName);
 
-			try {
-				executor.shutdown();
-				executor.awaitTermination(5, TimeUnit.SECONDS);
-			} catch (InterruptedException e) {
-				logger.error("Was interrupted while shutting down tasks");
-			} finally {
-				if (!executor.isTerminated()) {
-					logger.error("Tasks not stopped after " + 5 + "s. Shutting down now.");
-					executor.shutdownNow();
-				}
+			shutdownExecutor(executor);
+		}
+		for (String poolName : this.scheduledExecutors.keySet()) {
+			logger.info("Shutting down scheduled executor pool " + poolName);
+			ExecutorService executor = this.scheduledExecutors.get(poolName);
+
+			shutdownExecutor(executor);
+		}
+	}
+
+	private void shutdownExecutor(ExecutorService executor) {
+		try {
+			executor.shutdown();
+			executor.awaitTermination(5, TimeUnit.SECONDS);
+		} catch (InterruptedException e) {
+			logger.error("Was interrupted while shutting down tasks");
+		} finally {
+			if (!executor.isTerminated()) {
+				logger.error("Tasks not stopped after " + 5 + "s. Shutting down now.");
+				executor.shutdownNow();
 			}
 		}
 	}