Release v2.20261103.0

config/migrate-scripts/migrate-db-8bc91ce.sh

@@ -1,6 +0,0 @@
-CREATE INDEX channel_videos_ucid_published_idx
-  ON public.channel_videos
-  USING btree
-  (ucid COLLATE pg_catalog."default", published);
-
-DROP INDEX channel_videos_ucid_idx;

config/sql/channel_videos.sql

@@ -19,12 +19,12 @@ CREATE TABLE IF NOT EXISTS public.channel_videos
 
 GRANT ALL ON TABLE public.channel_videos TO current_user;
 
--- Index: public.channel_videos_ucid_published_idx
+-- Index: public.channel_videos_ucid_idx
 
--- DROP INDEX public.channel_videos_ucid_published_idx;
+-- DROP INDEX public.channel_videos_ucid_idx;
 
-CREATE INDEX IF NOT EXISTS channel_videos_ucid_published_idx
+CREATE INDEX IF NOT EXISTS channel_videos_ucid_idx
   ON public.channel_videos
   USING btree
-  (ucid COLLATE pg_catalog."default", published);
+  (ucid COLLATE pg_catalog."default");
 

shard.yml

@@ -1,5 +1,5 @@
 name: invidious
-version: 2.20261103.0-dev
+version: 2.20261103.0
 
 authors:
   - Invidious team <contact@invidious.io>

src/invidious.cr

@@ -130,6 +130,14 @@ Kemal.config.extra_options do |parser|
       exit
     end
   end
+  parser.on("-f THREADS", "--feed-threads=THREADS", "Number of threads for refreshing feeds (default: #{CONFIG.feed_threads})") do |number|
+    begin
+      CONFIG.feed_threads = number.to_i
+    rescue
+      puts "THREADS must be integer"
+      exit
+    end
+  end
   parser.on("-o OUTPUT", "--output=OUTPUT", "Redirect output (default: #{CONFIG.output})") do |output|
     CONFIG.output = output
   end
@@ -181,6 +189,10 @@ if CONFIG.channel_threads > 0
   Invidious::Jobs.register Invidious::Jobs::RefreshChannelsJob.new(PG_DB)
 end
 
+if CONFIG.feed_threads > 0
+  Invidious::Jobs.register Invidious::Jobs::RefreshFeedsJob.new(PG_DB)
+end
+
 if CONFIG.statistics_enabled
   Invidious::Jobs.register Invidious::Jobs::StatisticsRefreshJob.new(PG_DB, SOFTWARE)
 end

src/invidious/config.cr

@@ -101,6 +101,8 @@ class Config
   # Time interval between two executions of the job that crawls channel videos (subscriptions update).
   @[YAML::Field(converter: Preferences::TimeSpanConverter)]
   property channel_refresh_interval : Time::Span = 30.minutes
+  # Number of threads to use for updating feeds
+  property feed_threads : Int32 = 1
   # Log file path or STDOUT
   property output : String = "STDOUT"
   # Default log level, valid YAML values are ints and strings, see src/invidious/helpers/logger.cr

src/invidious/jobs/refresh_feeds_job.cr

@@ -0,0 +1,75 @@
+class Invidious::Jobs::RefreshFeedsJob < Invidious::Jobs::BaseJob
+  private getter db : DB::Database
+
+  def initialize(@db)
+  end
+
+  def begin
+    max_fibers = CONFIG.feed_threads
+    active_fibers = 0
+    active_channel = ::Channel(Bool).new
+
+    loop do
+      db.query("SELECT email FROM users WHERE feed_needs_update = true OR feed_needs_update IS NULL") do |rs|
+        rs.each do
+          email = rs.read(String)
+          view_name = "subscriptions_#{sha256(email)}"
+
+          if active_fibers >= max_fibers
+            if active_channel.receive
+              active_fibers -= 1
+            end
+          end
+
+          active_fibers += 1
+          spawn do
+            begin
+              # Drop outdated views
+              column_array = Invidious::Database.get_column_array(db, view_name)
+              ChannelVideo.type_array.each_with_index do |name, i|
+                if name != column_array[i]?
+                  LOGGER.info("RefreshFeedsJob: DROP MATERIALIZED VIEW #{view_name}")
+                  db.exec("DROP MATERIALIZED VIEW #{view_name}")
+                  raise "view does not exist"
+                end
+              end
+
+              if !db.query_one("SELECT pg_get_viewdef('#{view_name}')", as: String).includes? "WHERE ((cv.ucid = ANY (u.subscriptions))"
+                LOGGER.info("RefreshFeedsJob: Materialized view #{view_name} is out-of-date, recreating...")
+                db.exec("DROP MATERIALIZED VIEW #{view_name}")
+              end
+
+              db.exec("REFRESH MATERIALIZED VIEW #{view_name}")
+              db.exec("UPDATE users SET feed_needs_update = false WHERE email = $1", email)
+            rescue ex
+              # Rename old views
+              begin
+                legacy_view_name = "subscriptions_#{sha256(email)[0..7]}"
+
+                db.exec("SELECT * FROM #{legacy_view_name} LIMIT 0")
+                LOGGER.info("RefreshFeedsJob: RENAME MATERIALIZED VIEW #{legacy_view_name}")
+                db.exec("ALTER MATERIALIZED VIEW #{legacy_view_name} RENAME TO #{view_name}")
+              rescue ex
+                begin
+                  # While iterating through, we may have an email stored from a deleted account
+                  if db.query_one?("SELECT true FROM users WHERE email = $1", email, as: Bool)
+                    LOGGER.info("RefreshFeedsJob: CREATE #{view_name}")
+                    db.exec("CREATE MATERIALIZED VIEW #{view_name} AS #{Invidious::User::Users::MATERIALIZED_VIEW_SQL.call(email)}")
+                    db.exec("UPDATE users SET feed_needs_update = false WHERE email = $1", email)
+                  end
+                rescue ex
+                  LOGGER.error("RefreshFeedJobs: REFRESH #{email} : #{ex.message}")
+                end
+              end
+            end
+
+            active_channel.send(true)
+          end
+        end
+      end
+
+      sleep 5.seconds
+      Fiber.yield
+    end
+  end
+end

src/invidious/routes/account.cr

@@ -123,8 +123,10 @@ module Invidious::Routes::Account
       return Errors.error_template(400, ex)
     end
 
+    view_name = "subscriptions_#{sha256(user.email)}"
     Invidious::Database::Users.delete(user)
     Invidious::Database::SessionIDs.delete(email: user.email)
+    PG_DB.exec("DROP MATERIALIZED VIEW #{view_name}")
 
     env.request.cookies.each do |cookie|
       cookie.expires = Time.utc(1990, 1, 1)

src/invidious/routes/login.cr

@@ -125,6 +125,9 @@ module Invidious::Routes::Login
         Invidious::Database::Users.insert(user)
         Invidious::Database::SessionIDs.insert(sid, email)
 
+        view_name = "subscriptions_#{sha256(user.email)}"
+        PG_DB.exec("CREATE MATERIALIZED VIEW #{view_name} AS #{Invidious::User::Users::MATERIALIZED_VIEW_SQL.call(user.email)}")
+
         if alt = CONFIG.alternative_domains.index(host)
           env.response.cookies["SID"] = Invidious::User::Cookies.sid(CONFIG.alternative_domains[alt], sid)
         else

src/invidious/search/processors.cr

@@ -36,19 +36,19 @@ module Invidious::Search::Processors
 
   # Search inside of user subscriptions
   def subscriptions(query : Query, user : Invidious::User) : Array(ChannelVideo)
+    view_name = "subscriptions_#{sha256(user.email)}"
+
     return PG_DB.query_all(<<-SQL,
       SELECT id,title,published,updated,ucid,author,length_seconds
       FROM (
-        SELECT cv.*,
+        SELECT *,
-        to_tsvector(cv.title) ||
+        to_tsvector(#{view_name}.title) ||
-        to_tsvector(cv.author) AS document
+        to_tsvector(#{view_name}.author)
-        FROM channel_videos cv
+        as document
-        JOIN users ON cv.ucid = any(users.subscriptions)
+        FROM #{view_name}
-        WHERE users.email = $1 AND published > now() - interval '1 month'
+      ) v_search WHERE v_search.document @@ plainto_tsquery($1) LIMIT 20 OFFSET $2;
-        ORDER BY published
-      ) v_search WHERE v_search.document @@ plainto_tsquery($2) LIMIT 20 OFFSET $3;
       SQL
-      user.email, query.text, (query.page - 1) * 20,
+      query.text, (query.page - 1) * 20,
       as: ChannelVideo
     )
   end

src/invidious/user/users.cr

@@ -2,6 +2,8 @@ require "crypto/bcrypt/password"
 
 module Invidious::User::Users
   extend self
+  # Materialized views may not be defined using bound parameters (`$1` as used elsewhere)
+  MATERIALIZED_VIEW_SQL = ->(email : String) { "SELECT cv.* FROM channel_videos cv WHERE EXISTS (SELECT subscriptions FROM users u WHERE cv.ucid = ANY (u.subscriptions) AND u.email = E'#{email.gsub({'\'' => "\\'", '\\' => "\\\\"})}') ORDER BY published DESC" }
 
   def create_user(sid, email, password)
     password = Crypto::Bcrypt::Password.create(password, cost: 10)
@@ -27,6 +29,7 @@ module Invidious::User::Users
     offset = (page - 1) * limit
 
     notifications = Invidious::Database::Users.select_notifications(user)
+    view_name = "subscriptions_#{sha256(user.email)}"
 
     if user.preferences.notifications_only && !notifications.empty?
       # Only show notifications
@@ -51,19 +54,11 @@ module Invidious::User::Users
           # Show latest video from a channel that a user hasn't watched
           # "unseen_only" isn't really correct here, more accurate would be "unwatched_only"
 
-          videos = PG_DB.query_all("SELECT DISTINCT ON (cv.ucid) cv.* " \
+          videos = PG_DB.query_all("SELECT DISTINCT ON (ucid) * FROM #{view_name} WHERE NOT id = ANY ($1::text[]) ORDER BY ucid, published DESC", user.watched, as: ChannelVideo)
-                                   "FROM channel_videos cv " \
-                                   "JOIN users ON cv.ucid = any(users.subscriptions) " \
-                                   "WHERE users.email = $1 AND NOT cv.id = any(users.watched) AND published > now() - interval '1 month' " \
-                                   "ORDER BY ucid, published DESC", user.email, as: ChannelVideo)
         else
           # Show latest video from each channel
 
-          videos = PG_DB.query_all("SELECT DISTINCT ON (cv.ucid) cv.* " \
+          videos = PG_DB.query_all("SELECT DISTINCT ON (ucid) * FROM #{view_name} ORDER BY ucid, published DESC", as: ChannelVideo)
-                                   "FROM channel_videos cv " \
-                                   "JOIN users ON cv.ucid = any(users.subscriptions) " \
-                                   "WHERE users.email = $1 AND published > now() - interval '1 month' " \
-                                   "ORDER BY ucid, published DESC", user.email, as: ChannelVideo)
         end
 
         videos.sort_by!(&.published).reverse!
@@ -71,19 +66,11 @@ module Invidious::User::Users
         if user.preferences.unseen_only
           # Only show unwatched
 
-          videos = PG_DB.query_all("SELECT cv.* " \
+          videos = PG_DB.query_all("SELECT * FROM #{view_name} WHERE NOT id = ANY ($1::text[]) ORDER BY published DESC LIMIT $2 OFFSET $3", user.watched, limit, offset, as: ChannelVideo)
-                                   "FROM channel_videos cv " \
-                                   "JOIN users ON cv.ucid = any(users.subscriptions) " \
-                                   "WHERE users.email = $1 AND NOT cv.id = any(users.watched) AND published > now() - interval '1 month' " \
-                                   "ORDER BY published DESC LIMIT $2 OFFSET $3", user.email, limit, offset, as: ChannelVideo)
         else
           # Sort subscriptions as normal
 
-          videos = PG_DB.query_all("SELECT cv.* " \
+          videos = PG_DB.query_all("SELECT * FROM #{view_name} ORDER BY published DESC LIMIT $1 OFFSET $2", limit, offset, as: ChannelVideo)
-                                   "FROM channel_videos cv " \
-                                   "JOIN users ON cv.ucid = any(users.subscriptions) " \
-                                   "WHERE users.email = $1 AND published > now() - interval '1 month' " \
-                                   "ORDER BY published DESC LIMIT $2 OFFSET $3", user.email, limit, offset, as: ChannelVideo)
         end
       end