MOSIP-35611 Fixed startup issue

Signed-off-by: kameshsr <[email protected]>
mosip · Nov 29, 2024 · 6eb2a75 · 6eb2a75
1 parent 4ae6f33
commit 6eb2a75
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/resident/resident-service/src/main/java/io/mosip/resident/config/Config.java b/resident/resident-service/src/main/java/io/mosip/resident/config/Config.java
@@ -71,7 +71,15 @@ public class Config {
 
 	@Value("${" + ResidentConstants.RESIDENT_REST_TEMPLATE_METRICS_INTERCEPTOR_FILTER_ENABLED + ":false}")
 	private boolean isResidentMetricsInterceptorFilterEnabled;
-
+
+	@Value("${task.scheduler.pool-size:10}")
+	private int poolSize;
+
+	@Value("${task.scheduler.await-termination-seconds:30}")
+	private int awaitTerminationSeconds;
+
+	@Value("${task.scheduler.wait-for-tasks-to-complete-on-shutdown:true}")
+	private boolean waitForTasksToCompleteOnShutdown;
 
 	@Autowired(required = false)
 	private RestTemplateLoggingInterceptor restTemplateLoggingInterceptor;
@@ -177,6 +185,9 @@ public ThreadPoolTaskScheduler threadPoolTaskScheduler() {
 		ThreadPoolTaskScheduler threadPoolTaskScheduler = new ThreadPoolTaskScheduler();
 		threadPoolTaskScheduler.setPoolSize(5);
 		threadPoolTaskScheduler.setThreadNamePrefix("ThreadPoolTaskScheduler");
+		threadPoolTaskScheduler.setPoolSize(poolSize);
+		threadPoolTaskScheduler.setWaitForTasksToCompleteOnShutdown(waitForTasksToCompleteOnShutdown);
+		threadPoolTaskScheduler.setAwaitTerminationSeconds(awaitTerminationSeconds);
 		return threadPoolTaskScheduler;
 	}