iv-org-mirror/invidious-copy-2022-03-16
1
0
Fork 0
mirror of https://gitea.invidious.io/iv-org/invidious-copy-2022-03-16.git synced 2024-08-15 00:53:18 +00:00
Code Issues Projects Releases Packages Wiki Activity

Add support for force_resolve to QUIC client

Browse source
This commit is contained in:
Omar Roth 2019-11-28 08:19:28 -06:00
parent 3fea1976c8
commit a017574f74
3 changed files with 14 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

12
src/invidious/helpers/utils.cr
View file

@ -31,8 +31,10 @@ struct QUICPool
begin
response = yield conn
rescue ex
conn.destroy_engine
conn.close
conn = QUIC::Client.new(url)
conn.family = (url.host == "www.youtube.com") ? CONFIG.force_resolve : Socket::Family::INET
conn.family = Socket::Family::INET if conn.family == Socket::Family::UNSPEC
conn.before_request { |r| add_yt_headers(r) } if url.host == "www.youtube.com"
response = yield conn
ensure
@ -45,9 +47,11 @@ struct QUICPool
private def build_pool
ConnectionPool(QUIC::Client).new(capacity: capacity, timeout: timeout) do
client = QUIC::Client.new(url)
client.before_request { |r| add_yt_headers(r) } if url.host == "www.youtube.com"
client
conn = QUIC::Client.new(url)
conn.family = (url.host == "www.youtube.com") ? CONFIG.force_resolve : Socket::Family::INET
conn.family = Socket::Family::INET if conn.family == Socket::Family::UNSPEC
conn.before_request { |r| add_yt_headers(r) } if url.host == "www.youtube.com"
conn
end
end
end