mirrors/nadeko_invidious
1
0
Fork 0
mirror of https://git.nadeko.net/Fijxu/invidious.git synced 2026-01-30 18:51:17 +00:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

feat: do all the backend balancing on the invidious side

Browse Source 
This will make invidious easier to maintain and escalate without the need of an overcomplicated reverse proxy configuration and multiple invidious instances with each one with a different configuration (in this case, invidious companion)
This commit is contained in:
Fijxu
2025-03-30 20:08:15 -03:00
parent ddf6802d76
commit d47aa3dd6a
13 changed files with 341 additions and 265 deletions
Download Patch File Download Diff File Expand all files Collapse all files

38
src/invidious/helpers/backend_info.cr
View File

@@ -1,45 +1,49 @@
module BackendInfo
extend self
@@exvpp_url : String = ""
@@status : Int32 = 0
@@exvpp_url : Array(String) = Array.new(CONFIG.invidious_companion.size, "")
@@status : Array(Int32) = Array.new(CONFIG.invidious_companion.size, 0)
def check_backends
check_companion()
end
def check_companion
begin
response = HTTP::Client.get "#{CONFIG.invidious_companion.sample.private_url}/healthz"
if response.status_code == 200
check_videoplayback_proxy()
else
@@status = 0
CONFIG.invidious_companion.each_with_index do |companion, index|
spawn do
begin
response = HTTP::Client.get "#{companion.private_url}/healthz"
if response.status_code == 200
check_videoplayback_proxy(companion, index)
else
@@status[index] = 0
end
rescue
@@status[index] = 0
end
end
rescue
@@status = 0
end
end
def check_videoplayback_proxy
private def check_videoplayback_proxy(companion : Config::CompanionConfig, index : Int32)
begin
info = HTTP::Client.get "#{CONFIG.invidious_companion.sample.private_url}/info"
info = HTTP::Client.get "#{companion.private_url}/info"
exvpp_url = JSON.parse(info.body)["external_videoplayback_proxy"]?.try &.to_s
exvpp_url = "" if exvpp_url.nil?
@@exvpp_url = exvpp_url
@@exvpp_url[index] = exvpp_url
if exvpp_url.empty?
@@status = 2
@@status[index] = 2
return
else
begin
exvpp_health = HTTP::Client.get "#{exvpp_url}/health"
if exvpp_health.status_code == 200
@@status = 2
@@status[index] = 2
return
else
@@status = 1
@@status[index] = 1
end
rescue
@@status = 1
@@status[index] = 1
end
end
rescue