From 5159ec0f6e2a9f6f1b2fabdd3066b0ea3ce9c17d Mon Sep 17 00:00:00 2001
From: Omar Roth <omarroth@hotmail.com>
Date: Wed, 7 Feb 2018 22:04:47 -0600
Subject: [PATCH] Pull rank_videos into seperate thread

---
 src/invidious.cr    | 42 ++++++++++++++++++++++++++----------------
 src/views/index.ecr |  2 +-
 2 files changed, 27 insertions(+), 17 deletions(-)

diff --git a/src/invidious.cr b/src/invidious.cr
index 9eecf106..5497ad3a 100644
--- a/src/invidious.cr
+++ b/src/invidious.cr
@@ -109,26 +109,36 @@ threads.times do
   end
 end
 
+top_videos = [] of Video
+
+spawn do
+  loop do
+    top = rank_videos(PG_DB, 120)
+    client = get_client(pool)
+
+    args = [] of String
+    1..(top.size - 1).times { |i| args << "($#{i + 1}), " }
+    args << "($#{top.size}) "
+    args = args.join("")
+
+    PG_DB.query("SELECT * FROM videos d INNER JOIN (VALUES #{args}) v(id) USING (id)", top) do |rs|
+      rs.each do
+        video = rs.read(Video)
+        top_videos << video
+      end
+    end
+
+    pool << client
+
+    sleep 5.minutes
+  end
+end
+
 macro templated(filename)
     render "src/views/#{{{filename}}}.ecr", "src/views/layout.ecr"
   end
 
 get "/" do |env|
-  top = rank_videos(PG_DB, 120)
-
-  args = [] of String
-  1..(top.size - 1).times { |i| args << "($#{i + 1}), " }
-  args << "($#{top.size}) "
-  args = args.join("")
-
-  videos = [] of Video
-  PG_DB.query("SELECT * FROM videos d INNER JOIN (VALUES #{args}) v(id) USING (id)", top) do |rs|
-    rs.each do
-      video = rs.read(Video)
-      videos << video
-    end
-  end
-
   templated "index"
 end
 
@@ -259,7 +269,7 @@ get "/:path" do |env|
 end
 
 error 500 do |env|
-  "Error 500"
+  templated "index"
 end
 
 public_folder "assets"
diff --git a/src/views/index.ecr b/src/views/index.ecr
index 7a25c739..3b061fa1 100644
--- a/src/views/index.ecr
+++ b/src/views/index.ecr
@@ -2,7 +2,7 @@
 <title>Invidious</title>
 <% end %>
 
-<% videos.each_slice(4) do |slice| %>
+<% top_videos.each_slice(4) do |slice| %>
 <div class="pure-g">
 <% slice.each do |video| %>
     <% player_response = JSON.parse(video.info["player_response"]) %>