Git-Mirrors/invidious
1
0
Fork 0
mirror of https://github.com/iv-org/invidious.git synced 2025-04-20 23:46:26 -04:00
Code Issues Packages Projects Releases Wiki Activity

Disable auto reconnect for companion pool clients

Browse Source
This commit is contained in:
syeopite 2025-04-10 01:02:59 -07:00
parent 178628529e
commit 22b76bf20b
No known key found for this signature in database
GPG Key ID: A73C186DA3955A1A
2 changed files with 19 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/invidious.cr
View File

@ -115,7 +115,7 @@ COMPANION_POOL = Invidious::ConnectionPool::Pool.new(
reinitialize_proxy: false
) do
companion = CONFIG.invidious_companion.sample
next make_client(companion.private_url, use_http_proxy: false)
next make_client(companion.private_url, use_http_proxy: false, allow_auto_reconnect: false)
end
# CLI

20
src/invidious/connection/client.cr
View File

@ -12,6 +12,15 @@ module Invidious
end
class HTTPClient < HTTP::Client
def initialize(uri : URI, tls : TLSContext = nil, allow_auto_reconnect : Bool = true)
tls = HTTP::Client.tls_flag(uri, tls)
host = HTTP::Client.validate_host(uri)
super(host, uri.port, tls)
@reconnect = allow_auto_reconnect
end
def initialize(uri : URI, tls : TLSContext = nil, force_resolve : Socket::Family = Socket::Family::UNSPEC)
tls = HTTP::Client.tls_flag(uri, tls)
@ -72,14 +81,21 @@ def add_yt_headers(request)
end
end
def make_client(url : URI, region = nil, force_resolve : Bool = false, force_youtube_headers : Bool = true, use_http_proxy : Bool = true)
def make_client(
url : URI,
region = nil,
force_resolve : Bool = false,
force_youtube_headers : Bool = true,
use_http_proxy : Bool = true,
allow_auto_reconnect : Bool = true,
)
if CONFIG.http_proxy && use_http_proxy
client = Invidious::HTTPClient.new(url)
client.proxy = make_configured_http_proxy_client() if CONFIG.http_proxy && use_http_proxy
elsif force_resolve
client = Invidious::HTTPClient.new(url, force_resolve: CONFIG.force_resolve)
else
client = Invidious::HTTPClient.new(url)
client = Invidious::HTTPClient.new(url, allow_auto_reconnect: allow_auto_reconnect)
end
client.before_request { |r| add_yt_headers(r) } if url.host.try &.ends_with?("youtube.com") || force_youtube_headers