PrivateCoffee/invidious
4
1
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 2

Add back in refreshing of channels every 2 days

Browse source
This commit is contained in:
matthewmcgarvey 2022-01-24 23:14:13 -06:00
parent a82d21ff78
commit e92b3779ad
2 changed files with 14 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
src/invidious/channels/channels.cr
View file

@ -152,12 +152,14 @@ def get_batch_channels(channels)
return final
end
def get_channel(id)
def get_channel(id) : InvidiousChannel
channel = Invidious::Database::Channels.select(id)
return channel if channel
if channel.nil? || (Time.utc - channel.updated) > 2.days
channel = fetch_channel(id, pull_all_videos: false)
Invidious::Database::Channels.insert(channel)
Invidious::Database::Channels.insert(channel, update_on_conflict: true)
end
return channel
end

9
src/invidious/database/channels.cr
View file

@ -10,7 +10,7 @@ module Invidious::Database::Channels
# Insert / delete
# -------------------
def insert(channel : InvidiousChannel)
def insert(channel : InvidiousChannel, update_on_conflict : Bool = false)
channel_array = channel.to_a
request = <<-SQL
@ -18,6 +18,13 @@ module Invidious::Database::Channels
VALUES (#{arg_array(channel_array)})
SQL
if update_on_conflict
request += <<-SQL
ON CONFLICT (id) DO UPDATE
SET author = $2, updated = $3
SQL
end
PG_DB.exec(request, args: channel_array)
end