From 188cddefe90c550227ea8bbd842a7c6ed3615eab Mon Sep 17 00:00:00 2001
From: Eugen Rochko <eugen@zeonfederated.com>
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