diff options
| -rw-r--r-- | src/invidious.cr | 18 |
1 files changed, 13 insertions, 5 deletions
diff --git a/src/invidious.cr b/src/invidious.cr index ca3a40da..9ed41865 100644 --- a/src/invidious.cr +++ b/src/invidious.cr @@ -7,10 +7,10 @@ require "time" PG_DB = DB.open "postgres://kemal:kemal@localhost:5432/invidious" CONTEXT = OpenSSL::SSL::Context::Client.insecure -POOL = [] of HTTP::Client -10.times do +POOL = [] of HTTP::Client +10.times do POOL << HTTP::Client.new("www.youtube.com", 443, CONTEXT) -end +end macro templated(filename) render "src/views/#{{{filename}}}.ecr", "src/views/layout.ecr" @@ -71,6 +71,10 @@ end def fetch_video(id) # Grab connection from pool + while POOL.empty? + sleep rand(0..10).milliseconds + end + client = POOL.pop # client = HTTP::Client.new("www.youtube.com", 443, CONTEXT) @@ -86,9 +90,9 @@ def fetch_video(id) # Return connection to pool POOL << client - + video = Video.new(id, info, html, Time.now) - + return video end @@ -196,6 +200,10 @@ end get "/search" do |env| query = env.params.query["q"] + while POOL.empty? + sleep rand(0..10).milliseconds + end + client = POOL.pop html = client.get("https://www.youtube.com/results?q=#{URI.escape(query)}&page=1").body |