From 76e624140d83a6dcad3f58043f50b0ba4c4b64d5 Mon Sep 17 00:00:00 2001
From: Gauthier Roebroeck <gauthier.roebroeck@gmail.com>
Date: Thu, 28 Jul 2022 09:47:22 +0800
Subject: [PATCH] fix: add configuration to set the database pool size

---
 .../komga/infrastructure/configuration/KomgaProperties.kt      | 3 +++
 .../infrastructure/datasource/DataSourcesConfiguration.kt      | 1 +
 2 files changed, 4 insertions(+)

diff --git a/komga/src/main/kotlin/org/gotson/komga/infrastructure/configuration/KomgaProperties.kt b/komga/src/main/kotlin/org/gotson/komga/infrastructure/configuration/KomgaProperties.kt
index bc9763767..4afa71703 100644
--- a/komga/src/main/kotlin/org/gotson/komga/infrastructure/configuration/KomgaProperties.kt
+++ b/komga/src/main/kotlin/org/gotson/komga/infrastructure/configuration/KomgaProperties.kt
@@ -72,6 +72,9 @@ class KomgaProperties {
     @get:Positive
     var batchChunkSize: Int = 1000
 
+    @get:Positive
+    var poolSize: Int? = null
+
     @get:Positive
     var maxPoolSize: Int = 1
 
diff --git a/komga/src/main/kotlin/org/gotson/komga/infrastructure/datasource/DataSourcesConfiguration.kt b/komga/src/main/kotlin/org/gotson/komga/infrastructure/datasource/DataSourcesConfiguration.kt
index b37d20fe1..b2e0f8e96 100644
--- a/komga/src/main/kotlin/org/gotson/komga/infrastructure/datasource/DataSourcesConfiguration.kt
+++ b/komga/src/main/kotlin/org/gotson/komga/infrastructure/datasource/DataSourcesConfiguration.kt
@@ -37,6 +37,7 @@ class DataSourcesConfiguration(
 
     val poolSize =
       if (komgaProperties.database.file.contains(":memory:")) 1
+      else if (komgaProperties.database.poolSize != null) komgaProperties.database.poolSize!!
       else Runtime.getRuntime().availableProcessors().coerceAtMost(komgaProperties.database.maxPoolSize)
 
     return HikariDataSource(