From 188cddefe90c550227ea8bbd842a7c6ed3615eab Mon Sep 17 00:00:00 2001 From: Eugen Rochko Date: Fri, 31 Mar 2017 13:01:00 +0200 Subject: [PATCH] Add maintenance task for updating counter caches --- lib/tasks/mastodon.rake | 19 +++++++++++++++++++ 1 file changed, 19 insertions(+) diff --git a/lib/tasks/mastodon.rake b/lib/tasks/mastodon.rake index bb10410b50..79dcb722ae 100644 --- a/lib/tasks/mastodon.rake +++ b/lib/tasks/mastodon.rake @@ -62,4 +62,23 @@ namespace :mastodon do end end end + + namespace :maintenance do + desc 'Update counter caches' + task update_counter_caches: :environment do + Rails.logger.debug 'Updating counter caches for accounts...' + + Account.unscoped.select('id').find_in_batches do |batch| + Account.where(id: batch.map(&:id)).update_all('statuses_count = (select count(*) from statuses where account_id = accounts.id), followers_count = (select count(*) from follows where target_account_id = accounts.id), following_count = (select count(*) from follows where account_id = accounts.id)') + end + + Rails.logger.debug 'Updating counter caches for statuses...' + + Status.unscoped.select('id').find_in_batches do |batch| + Status.where(id: batch.map(&:id)).update_all('favourites_count = (select count(*) from favourites where favourites.status_id = statuses.id), reblogs_count = (select count(*) from statuses as reblogs where reblogs.reblog_of_id = statuses.id)') + end + + Rails.logger.debug 'Done!' + end + end end