Refactor client creation

2 files changed, 10 insertions, 12 deletions

diff --git a/src/helpers.cr b/src/helpers.cr
index 23cab087..27c49c6e 100644
--- a/src/helpers.cr
+++ b/src/helpers.cr
@@ -209,3 +209,10 @@ def rank_videos(db, n)
   # Return top
   return top[1..n]
 end
+
+def make_client(url, context)
+  client = HTTP::Client.new(URL, CONTEXT)
+  client.read_timeout = 10.seconds
+  client.connect_timeout = 10.seconds
+  return client
+end
diff --git a/src/invidious.cr b/src/invidious.cr
index 592028e3..8573d2a5 100644
--- a/src/invidious.cr
+++ b/src/invidious.cr
@@ -48,10 +48,7 @@ CONTEXT.add_options(
   OpenSSL::SSL::Options::NO_SSL_V3
 )
 pool = Deque.new((threads * 1.2 + 1).to_i) do
-  client = HTTP::Client.new(URL, CONTEXT)
-  client.read_timeout = 10.seconds
-  client.connect_timeout = 10.seconds
-  client
+  make_client(URL, CONTEXT)
 end
 
 # Refresh pool by crawling YT
@@ -74,10 +71,7 @@ threads.times do
       end
 
       if rand(300) < 1
-        client = HTTP::Client.new(URL, CONTEXT)
-        client.read_timeout = 10.seconds
-        client.connect_timeout = 10.seconds
-        pool << client
+        pool << make_client(URL, CONTEXT)
       end
 
       time = Time.now
@@ -87,10 +81,7 @@ threads.times do
         video = get_video(id, client, PG_DB)
       rescue ex
         io << id << " : " << ex << "\n"
-        client = HTTP::Client.new(URL, CONTEXT)
-        client.read_timeout = 10.seconds
-        client.connect_timeout = 10.seconds
-        pool << client
+        pool << make_client(URL, CONTEXT)
         next
       ensure
         ids.delete(id)