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

refactor: move backend status handling to own function

Browse Source
This commit is contained in:
Fijxu
2026-01-01 21:47:43 -03:00
parent 6027bb04ec
commit d4e88dd3a7
1 changed files with 14 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -45,32 +45,19 @@ module BackendInfo
body = response.body
status_json = CompanionData.from_json(body)
if status_json.blocked
@@check_mutex.synchronize do
updated_status[index] = Status::Blocked.to_i
updated_ends.push(index)
end
updated_ends, updated_status = self.set_status(index, updated_ends, updated_status, Status::Blocked, true)
else
@@check_mutex.synchronize do
updated_status[index] = Status::Working.to_i
updated_ends.push(index)
end
updated_ends, updated_status = self.set_status(index, updated_ends, updated_status, Status::Working, true)
end
else
@@check_mutex.synchronize do
updated_status[index] = Status::Working.to_i
updated_ends.push(index)
end
updated_ends, updated_status = self.set_status(index, updated_ends, updated_status, Status::Working, true)
end
generate_csp([companion.public_url, companion.i2p_public_url], index)
self.generate_csp([companion.public_url, companion.i2p_public_url], index)
else
@@check_mutex.synchronize do
updated_status[index] = Status::Dead.to_i
end
_, updated_status = self.set_status(index, updated_ends, updated_status, Status::Dead, false)
end
rescue
@@check_mutex.synchronize do
updated_status[index] = Status::Dead.to_i
end
_, updated_status = self.set_status(index, updated_ends, updated_status, Status::Dead, false)
ensure
LOGGER.trace("Invidious companion: Done Index: \"#{index}\"")
channels.send(nil)
@@ -94,6 +81,14 @@ module BackendInfo
end
end
private def set_status(index, updated_ends, updated_status, status, push = false)
@@check_mutex.synchronize do
updated_status[index] = status.to_i
updated_ends.push(index) if push
end
return {updated_ends, updated_status}
end
def get_status
# Shouldn't need to lock since we never edit this array, only change the pointer.
return @@status