make async config configurable (#9)

qbicsoftware · Jun 10, 2024 · a2293d2 · a2293d2
1 parent 3a70248
commit a2293d2
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 3 deletions.
diff --git a/rest-api/src/main/java/life/qbic/data_download/rest/config/AsyncDownloadConfig.java b/rest-api/src/main/java/life/qbic/data_download/rest/config/AsyncDownloadConfig.java
@@ -19,13 +19,20 @@
 @EnableAsync
 public class AsyncDownloadConfig implements AsyncConfigurer {
 
+  @Value("${spring.async.threadpool.core-size}")
+  private int corePoolSize;
+  @Value("${spring.async.threadpool.max-size}")
+  private int maxPoolSize;
+  @Value("${spring.async.threadpool.queue-capacity}")
+  private int queueCapacity;
+
   @Override
   @Bean("taskExecutor")
   public AsyncTaskExecutor getAsyncExecutor() {
     ThreadPoolTaskExecutor threadPoolTaskExecutor = new ThreadPoolTaskExecutor();
-    threadPoolTaskExecutor.setCorePoolSize(2);
-    threadPoolTaskExecutor.setMaxPoolSize(5);
-    threadPoolTaskExecutor.setQueueCapacity(200);
+    threadPoolTaskExecutor.setCorePoolSize(corePoolSize);
+    threadPoolTaskExecutor.setMaxPoolSize(maxPoolSize);
+    threadPoolTaskExecutor.setQueueCapacity(queueCapacity);
     threadPoolTaskExecutor.setThreadNamePrefix("download - ");
     threadPoolTaskExecutor.initialize();
     return threadPoolTaskExecutor;

diff --git a/rest-api/src/main/resources/application.properties b/rest-api/src/main/resources/application.properties
@@ -59,3 +59,7 @@ server.tomcat.remoteip.protocol-header=x-forwarded-proto
 spring.mvc.async.request-timeout=2d
 # the timeout of a session
 spring.session.timeout=2d
+
+spring.async.threadpool.core-size=${ASYNC_CORE_SIZE:2}
+spring.async.threadpool.max-size=${ASYNC_MAX_SIZE:5}
+spring.async.threadpool.queue-capacity=${ASYNC_QUEUE_SIZE:2}