Handle companion stream failures after commit

This commit is contained in:
sergiobuilds 2026-08-12 02:23:19 +09:00
parent 471c1170e6
commit d817ab1431

View File

@ -13,6 +13,7 @@ module Invidious::Routes::Companion
end
end
rescue ex
LOGGER.warn("Companion GET upstream request failed: #{ex.class}")
env.response.status_code = 502
return env.response.print("502 Bad Gateway")
end
@ -32,6 +33,7 @@ module Invidious::Routes::Companion
end
end
rescue ex
LOGGER.warn("Companion POST upstream request failed: #{ex.class}")
env.response.status_code = 502
return env.response.print("502 Bad Gateway")
end
@ -50,6 +52,7 @@ module Invidious::Routes::Companion
end
end
rescue ex
LOGGER.warn("Companion OPTIONS upstream request failed: #{ex.class}")
env.response.status_code = 502
return env.response.print("502 Bad Gateway")
end
@ -61,6 +64,12 @@ module Invidious::Routes::Companion
env.response.headers[key] = value
end
begin
return IO.copy response.body_io, env.response
rescue ex
LOGGER.warn("Companion response stream failed: #{ex.class}")
env.response.close unless env.response.closed?
return
end
end
end