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

add option to disable quic

Browse Source
This commit is contained in:
Andrew Zhao
2021-04-03 16:11:35 -04:00
parent edb37c95e1
commit 9a8f1a0b0a
3 changed files with 13 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
src/invidious/helpers/utils.cr
View File

@@ -16,15 +16,15 @@ def add_yt_headers(request)
end
end
struct QUICPool
struct YoutubeConnectionPool
property! url : URI
property! capacity : Int32
property! timeout : Float64
property pool : ConnectionPool(QUIC::Client)
property pool : ConnectionPool(QUIC::Client | HTTP::Client)
def initialize(url : URI, @capacity = 5, @timeout = 5.0)
def initialize(url : URI, @capacity = 5, @timeout = 5.0, use_quic = true)
@url = url
@pool = build_pool
@pool = build_pool(use_quic)
end
def client(region = nil, &block)
@@ -50,9 +50,13 @@ struct QUICPool
response
end
private def build_pool
ConnectionPool(QUIC::Client).new(capacity: capacity, timeout: timeout) do
conn = QUIC::Client.new(url)
private def build_pool(use_quic)
ConnectionPool(QUIC::Client | HTTP::Client).new(capacity: capacity, timeout: timeout) do
if use_quic
conn = QUIC::Client.new(url)
else
conn = HTTP::Client.new(url)
end
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"