mirror of
https://gitea.invidious.io/iv-org/invidious.git
synced 2024-08-15 00:53:41 +00:00
Copy proxy_file in chunks
This commit is contained in:
parent
3ac8de0a64
commit
06bf0c2622
1 changed files with 22 additions and 10 deletions
|
@ -631,19 +631,31 @@ def cache_annotation(db, id, annotations)
|
||||||
end
|
end
|
||||||
|
|
||||||
def proxy_file(response, env)
|
def proxy_file(response, env)
|
||||||
if response.headers["Content-Length"]? && response.headers["Content-Length"] == "0"
|
if !response.body_io?
|
||||||
return
|
return
|
||||||
end
|
end
|
||||||
|
|
||||||
|
begin
|
||||||
if response.headers.includes_word?("Content-Encoding", "gzip")
|
if response.headers.includes_word?("Content-Encoding", "gzip")
|
||||||
Gzip::Writer.open(env.response) do |deflate|
|
Gzip::Writer.open(env.response) do |deflate|
|
||||||
IO.copy(response.body_io, deflate)
|
copy_in_chunks(response.body_io, deflate)
|
||||||
end
|
end
|
||||||
elsif response.headers.includes_word?("Content-Encoding", "deflate")
|
elsif response.headers.includes_word?("Content-Encoding", "deflate")
|
||||||
Flate::Writer.open(env.response) do |deflate|
|
Flate::Writer.open(env.response) do |deflate|
|
||||||
IO.copy(response.body_io, deflate)
|
copy_in_chunks(response.body_io, deflate)
|
||||||
end
|
end
|
||||||
else
|
else
|
||||||
IO.copy(response.body_io, env.response)
|
copy_in_chunks(response.body_io, env.response)
|
||||||
|
end
|
||||||
|
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
|
||||||
end
|
end
|
||||||
|
|
Loading…
Reference in a new issue