midou/invidious
1
0
Fork 1
mirror of https://github.com/iv-org/invidious.git synced 2024-11-26 16:52:12 +05:30
Code Issues Packages Projects Releases Wiki Activity

Ensure IP family is always used when force_resolve

Browse Source
This commit is contained in:
syeopite 2024-08-24 19:37:04 -07:00
parent bd48af825c
commit 7521902e88
No known key found for this signature in database
GPG Key ID: A73C186DA3955A1A
1 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -31,7 +31,6 @@ struct YoutubeConnectionPool
rescue ex rescue ex
conn.close conn.close
conn = make_client(url) conn = make_client(url)
conn.family = Socket::Family::INET if conn.family == Socket::Family::UNSPEC
response = yield conn response = yield conn
ensure ensure
pool.release(conn) pool.release(conn)
@ -42,9 +41,7 @@ struct YoutubeConnectionPool
private def build_pool private def build_pool
DB::Pool(HTTP::Client).new(initial_pool_size: 0, max_pool_size: capacity, max_idle_pool_size: capacity, checkout_timeout: timeout) do DB::Pool(HTTP::Client).new(initial_pool_size: 0, max_pool_size: capacity, max_idle_pool_size: capacity, checkout_timeout: timeout) do
conn = make_client(url, force_resolve: true) next make_client(url, force_resolve: true)
conn.family = Socket::Family::INET if conn.family == Socket::Family::UNSPEC
conn
end end
end end
end end
@ -55,6 +52,7 @@ def make_client(url : URI, region = nil, force_resolve : Bool = false, force_you
# Force the usage of a specific configured IP Family # Force the usage of a specific configured IP Family
if force_resolve if force_resolve
client.family = CONFIG.force_resolve client.family = CONFIG.force_resolve
client.family = Socket::Family::INET if client.family == Socket::Family::UNSPEC
end end
client.before_request { |r| add_yt_headers(r) } if url.host == "www.youtube.com" || force_youtube_headers client.before_request { |r| add_yt_headers(r) } if url.host == "www.youtube.com" || force_youtube_headers