Allow channel_threads to be configured and increase pool_size

2 files changed, 13 insertions, 4 deletions

diff --git a/src/helpers.cr b/src/helpers.cr
index 4fe8ae4f..4b5a77e0 100644
--- a/src/helpers.cr
+++ b/src/helpers.cr
@@ -15,9 +15,10 @@ end
 
 class Config
   YAML.mapping({
-    pool_size: Int32,
-    threads:   Int32,
-    db:        NamedTuple(
+    pool_size:       Int32,
+    threads:         Int32,
+    channel_threads: Int32,
+    db:              NamedTuple(
       user: String,
       password: String,
       host: String,
diff --git a/src/invidious.cr b/src/invidious.cr
index 5437c356..37ee19c6 100644
--- a/src/invidious.cr
+++ b/src/invidious.cr
@@ -27,7 +27,7 @@ CONFIG = Config.from_yaml(File.read("config/config.yml"))
 
 pool_size = CONFIG.pool_size
 threads = CONFIG.threads
-channel_threads = 10
+channel_threads = CONFIG.channel_threads
 
 Kemal.config.extra_options do |parser|
   parser.banner = "Usage: invidious [arguments]"
@@ -47,6 +47,14 @@ Kemal.config.extra_options do |parser|
       exit
     end
   end
+  parser.on("-c THREADS", "--channel-threads=THREADS", "Number of threads for refreshing channels (default: #{channel_threads})") do |number|
+    begin
+      channel_threads = number.to_i
+    rescue ex
+      puts "THREADS must be integer"
+      exit
+    end
+  end
 end
 
 Kemal::CLI.new