mirrors/nadeko_invidious
1
0
Fork 0
mirror of https://git.nadeko.net/Fijxu/invidious.git synced 2025-07-09 15:28:25 +00:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

chore: make companion pool a class and store their companion instance

Browse Source
This commit is contained in:
Fijxu 2025-06-20 16:38:12 -04:00
parent 5eba7cf150
commit b71539449e
No known key found for this signature in database
GPG Key ID: 32C1DDF333EDA6A4
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
src/invidious/yt_backend/connection_pool.cr
View File

@ -46,8 +46,9 @@ struct YoutubeConnectionPool
end end
end end
struct CompanionConnectionPool class CompanionConnectionPool
property pool : DB::Pool(HTTP::Client) property pool : DB::Pool(HTTP::Client)
property companion : URI
def initialize(companion, capacity = 5, timeout = 5.0) def initialize(companion, capacity = 5, timeout = 5.0)
options = DB::Pool::Options.new( options = DB::Pool::Options.new(
@ -57,8 +58,10 @@ struct CompanionConnectionPool
checkout_timeout: timeout checkout_timeout: timeout
) )
@companion = companion.private_url
@pool = DB::Pool(HTTP::Client).new(options) do @pool = DB::Pool(HTTP::Client).new(options) do
next make_client(companion.private_url, use_http_proxy: false) next make_client(@companion, use_http_proxy: false)
end end
end end
@ -70,10 +73,7 @@ struct CompanionConnectionPool
rescue ex rescue ex
conn.close conn.close
scheme = "https" if conn.tls || "http" conn = make_client(@companion, use_http_proxy: false)
same_companion = "#{scheme}://#{conn.host}:#{conn.port}"
conn = make_client(URI.parse(same_companion), use_http_proxy: false)
response = yield conn response = yield conn
ensure ensure