67 行
2.2 KiB
Crystal
67 行
2.2 KiB
Crystal
class Invidious::Jobs::RefreshChannelsJob < Invidious::Jobs::BaseJob
|
|
private getter db : DB::Database
|
|
|
|
def initialize(@db)
|
|
end
|
|
|
|
def begin
|
|
max_fibers = CONFIG.channel_threads
|
|
lim_fibers = max_fibers
|
|
active_fibers = 0
|
|
active_channel = Channel(Bool).new
|
|
backoff = 2.minutes
|
|
|
|
loop do
|
|
LOGGER.debug("RefreshChannelsJob: Refreshing all channels")
|
|
PG_DB.query("SELECT id FROM channels ORDER BY updated") do |rs|
|
|
rs.each do
|
|
id = rs.read(String)
|
|
|
|
if active_fibers >= lim_fibers
|
|
LOGGER.trace("RefreshChannelsJob: Fiber limit reached, waiting...")
|
|
if active_channel.receive
|
|
LOGGER.trace("RefreshChannelsJob: Fiber limit ok, continuing")
|
|
active_fibers -= 1
|
|
end
|
|
end
|
|
|
|
LOGGER.debug("RefreshChannelsJob: #{id} : Spawning fiber")
|
|
active_fibers += 1
|
|
spawn do
|
|
begin
|
|
LOGGER.trace("RefreshChannelsJob: #{id} fiber : Fetching channel")
|
|
channel = fetch_channel(id, pull_all_videos: CONFIG.full_refresh)
|
|
|
|
lim_fibers = max_fibers
|
|
|
|
LOGGER.trace("RefreshChannelsJob: #{id} fiber : Updating DB")
|
|
Invidious::Database::Channels.update_author(id, channel.author)
|
|
rescue ex
|
|
LOGGER.error("RefreshChannelsJob: #{id} : #{ex.message}")
|
|
if ex.message == "Deleted or invalid channel"
|
|
Invidious::Database::Channels.update_mark_deleted(id)
|
|
else
|
|
lim_fibers = 1
|
|
LOGGER.error("RefreshChannelsJob: #{id} fiber : backing off for #{backoff}s")
|
|
sleep backoff
|
|
if backoff < 1.days
|
|
backoff += backoff
|
|
else
|
|
backoff = 1.days
|
|
end
|
|
end
|
|
ensure
|
|
LOGGER.debug("RefreshChannelsJob: #{id} fiber : Done")
|
|
active_channel.send(true)
|
|
end
|
|
end
|
|
end
|
|
end
|
|
|
|
LOGGER.debug("RefreshChannelsJob: Done, sleeping for #{CONFIG.channel_refresh_interval}")
|
|
sleep CONFIG.channel_refresh_interval
|
|
Fiber.yield
|
|
end
|
|
end
|
|
end
|