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

Copy proxy_file in chunks

Browse source
This commit is contained in:
Omar Roth 2019-05-20 12:06:44 -05:00
parent 3ac8de0a64
commit 06bf0c2622
No known key found for this signature in database
GPG key ID: B8254FB7EC3D37F2
1 changed files with 22 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

32
src/invidious/helpers/helpers.cr
View file

@ -631,19 +631,31 @@ def cache_annotation(db, id, annotations)
end
def proxy_file(response, env)
if response.headers["Content-Length"]? && response.headers["Content-Length"] == "0"
if !response.body_io?
return
end
if response.headers.includes_word?("Content-Encoding", "gzip")
Gzip::Writer.open(env.response) do |deflate|
IO.copy(response.body_io, deflate)
begin
if response.headers.includes_word?("Content-Encoding", "gzip")
Gzip::Writer.open(env.response) do |deflate|
copy_in_chunks(response.body_io, deflate)
end
elsif response.headers.includes_word?("Content-Encoding", "deflate")
Flate::Writer.open(env.response) do |deflate|
copy_in_chunks(response.body_io, deflate)
end
else
copy_in_chunks(response.body_io, env.response)
end
elsif response.headers.includes_word?("Content-Encoding", "deflate")
Flate::Writer.open(env.response) do |deflate|
IO.copy(response.body_io, deflate)
end
else
IO.copy(response.body_io, env.response)
rescue ex
end
end
# https://stackoverflow.com/a/44802810 <3
def copy_in_chunks(input, output, chunk_size = 4096)
size = 1
while size > 0
size = IO.copy(input, output, chunk_size)
Fiber.yield
end
end