From 026f5839e2359d565229c05b768f8a7ed694bf4b Mon Sep 17 00:00:00 2001
From: Keuin <keuinx@gmail.com>
Date: Mon, 28 Mar 2022 21:40:33 +0800
Subject: Make `sqlite_thread_pool_size` configurable.

---
 src/config.rs   | 6 ++++++
 src/database.rs | 6 ++----
 src/main.rs     | 2 +-
 3 files changed, 9 insertions(+), 5 deletions(-)

(limited to 'src')

diff --git a/src/config.rs b/src/config.rs
index 1300e9a..2f4bbd8 100644
--- a/src/config.rs
+++ b/src/config.rs
@@ -19,6 +19,8 @@ pub struct Config {
     pub log_level: String,
     #[serde(default = "Config::default_max_body_size")]
     pub max_body_size: u64,
+    #[serde(default = "Config::default_sqlite_thread_pool_size")]
+    pub sqlite_thread_pool_size: u32,
 }
 
 impl Config {
@@ -41,6 +43,10 @@ impl Config {
     fn default_db_file() -> String {
         String::from("kimikuri.db")
     }
+
+    fn default_sqlite_thread_pool_size() -> u32 {
+        16
+    }
 }
 
 
diff --git a/src/database.rs b/src/database.rs
index a73aafd..07830a6 100644
--- a/src/database.rs
+++ b/src/database.rs
@@ -9,15 +9,13 @@ use crate::user::User;
 
 pub type DbPool = sqlx::sqlite::SqlitePool;
 
-const SQLITE_THREAD_POOL_SIZE: u32 = 16;
-
-pub async fn open(file_path: &str) -> Result<DbPool, sqlx::Error> {
+pub async fn open(file_path: &str, sqlite_thread_pool_size: u32) -> Result<DbPool, sqlx::Error> {
     let opt =
         SqliteConnectOptions::from_str(format!("sqlite://{}", file_path).as_str())?
             .create_if_missing(true);
     debug!("Opening database pool...");
     let pool = SqlitePoolOptions::new()
-        .max_connections(SQLITE_THREAD_POOL_SIZE)
+        .max_connections(sqlite_thread_pool_size)
         .connect_with(opt).await?;
 
     // create table
diff --git a/src/main.rs b/src/main.rs
index 8ada3a0..e5cb1bb 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -76,7 +76,7 @@ async fn main() {
 
     let db = config.db_file.as_str();
     info!(db, "Opening database...");
-    let db: DbPool = database::open(db)
+    let db: DbPool = database::open(db, config.sqlite_thread_pool_size)
         .await.expect(&*format!("cannot open database {}", db));
 
     info!("Spawning bot coroutine...");
-- 
cgit v1.2.3