From 8c9587ad654014d3219be1b5b491adaff496858c Mon Sep 17 00:00:00 2001
From: realaravinth <realaravinth@batsense.net>
Date: Tue, 9 Aug 2022 16:30:04 +0530
Subject: [PATCH] feat: set custom runers and queue length

---
 docs/HACKING.md  | 2 +-
 src/data.rs      | 8 +++++++-
 src/tests/mod.rs | 2 ++
 3 files changed, 10 insertions(+), 2 deletions(-)

diff --git a/docs/HACKING.md b/docs/HACKING.md
index 30a379c6..c1c1c22c 100644
--- a/docs/HACKING.md
+++ b/docs/HACKING.md
@@ -107,7 +107,7 @@ I use mariadb also in Docker
 2. Create create database user:
 
 ```bash
-docker create
+docker create \
 	-p 3306:3306 \
 	--name some-mariadb \
 	--env MARIADB_USER=maria \
diff --git a/src/data.rs b/src/data.rs
index f3c405f3..4aaf5031 100644
--- a/src/data.rs
+++ b/src/data.rs
@@ -116,11 +116,17 @@ impl SystemGroup {
             .build()
             .unwrap();
 
+        let runners = if let Some(runners) = s.captcha.runners {
+            runners
+        } else {
+            num_cpus::get_physical()
+        };
         SystemBuilder::default()
             .pow(pow)
             .cache(c)
             .master(m)
-            .runners(num_cpus::get_physical())
+            .runners(runners)
+            .queue_length(s.captcha.queue_length)
             .build()
     }
 
diff --git a/src/tests/mod.rs b/src/tests/mod.rs
index a933c23b..9de70cf9 100644
--- a/src/tests/mod.rs
+++ b/src/tests/mod.rs
@@ -49,6 +49,7 @@ pub mod pg {
     pub async fn get_data() -> ArcData {
         let url = env::var("POSTGRES_DATABASE_URL").unwrap();
         let mut settings = get_settings();
+        settings.captcha.runners = Some(1);
         settings.database.url = url.clone();
         settings.database.database_type = DBType::Postgres;
         let data = Data::new(&settings).await;
@@ -67,6 +68,7 @@ pub mod maria {
     pub async fn get_data() -> ArcData {
         let url = env::var("MARIA_DATABASE_URL").unwrap();
         let mut settings = get_settings();
+        settings.captcha.runners = Some(1);
         settings.database.url = url.clone();
         settings.database.database_type = DBType::Maria;
         let data = Data::new(&settings).await;