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 override to mute IO::Error in HTTP::Server::Response::Output

Browse source
This commit is contained in:
Samantaz Fox 2021-12-01 18:32:28 +01:00 committed by GitHub
parent e5557b515e
commit 1955241d58
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 30 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -48,11 +48,39 @@ class HTTP::Client
end
end
# Mute the ClientError exception raised when a connection is flushed.
# This happends when the connection is unexpectedly closed by the client.
# Mute the ClientError exception raised when a
# connection is unexpectedly closed by the client.
#
class HTTP::Server::Response
class Output
# Copy-paste of https://github.com/crystal-lang/crystal/blob/1.2.2/src/http/server/response.cr#L205-L228
# but without the "raise" statement in `rescue ex`
private def unbuffered_write(slice : Bytes)
return if slice.empty?
unless response.wrote_headers?
if response.version != "HTTP/1.0" && !response.headers.has_key?("Content-Length")
response.headers["Transfer-Encoding"] = "chunked"
@chunked = true
end
end
ensure_headers_written
if @chunked
slice.size.to_s(@io, 16)
@io << "\r\n"
@io.write(slice)
@io << "\r\n"
else
@io.write(slice)
end
rescue ex : IO::Error
unbuffered_close
end
# Copy-paste of https://github.com/crystal-lang/crystal/blob/1.2.2/src/http/server/response.cr#L274-L280
# but without the "raise" statement in `rescue ex`
private def unbuffered_flush
@io.flush
rescue ex : IO::Error