midou/invidious
1
0
Fork 1
mirror of https://github.com/iv-org/invidious.git synced 2024-11-27 01:01:59 +05:30
Code Issues Packages Projects Releases Wiki Activity

Add channel refresh

Browse Source
This commit is contained in:
Omar Roth 2018-03-25 22:18:29 -05:00
parent 252f1c0af3
commit edfe1eefe8
2 changed files with 24 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
src/helpers.cr
View File

@ -489,11 +489,12 @@ def get_channel(id, client, db)
if db.query_one?("SELECT EXISTS (SELECT true FROM channels WHERE id = $1)", id, as: Bool)
channel = db.query_one("SELECT * FROM channels WHERE id = $1", id, as: InvidiousChannel)
if Time.now - channel.updated > 1.hours
db.exec("DELETE FROM channels * WHERE id = $1", id)
if Time.now - channel.updated > 1.minutes
channel = fetch_channel(id, client)
args = arg_array(channel.to_a)
db.exec("INSERT INTO channels VALUES (#{args})", channel.to_a)
channel_array = channel.to_a[1..-1]
args = arg_array(channel_array)
db.exec("UPDATE channels SET (rss,updated,author) = (#{args}) WHERE id = '#{channel.id}'", channel_array)
end
else
channel = fetch_channel(id, client)

19
src/invidious.cr
View File

@ -27,6 +27,7 @@ CONFIG = Config.from_yaml(File.read("config/config.yml"))
pool_size = CONFIG.pool_size
threads = CONFIG.threads
channel_threads = 10
Kemal.config.extra_options do |parser|
parser.banner = "Usage: invidious [arguments]"
@ -122,6 +123,24 @@ threads.times do
end
end
channel_threads.times do |i|
spawn do
loop do
query = "SELECT id FROM channels ORDER BY updated \
LIMIT (SELECT count(*)/#{channel_threads} FROM channels) \
OFFSET (SELECT count(*)*#{i}/#{channel_threads} FROM channels)"
PG_DB.query(query) do |rs|
rs.each do
client = get_client(youtube_pool)
id = rs.read(String)
channel = get_channel(id, client, PG_DB)
youtube_pool << client
end
end
end
end
end
top_videos = [] of Video
spawn do