about summary refs log tree commit diff
path: root/db/migrate/20200717015413_backfill_status_nest_level.rb
diff options
context:
space:
mode:
Diffstat (limited to 'db/migrate/20200717015413_backfill_status_nest_level.rb')
-rw-r--r--db/migrate/20200717015413_backfill_status_nest_level.rb24
1 files changed, 24 insertions, 0 deletions
diff --git a/db/migrate/20200717015413_backfill_status_nest_level.rb b/db/migrate/20200717015413_backfill_status_nest_level.rb
new file mode 100644
index 000000000..1f37ef847
--- /dev/null
+++ b/db/migrate/20200717015413_backfill_status_nest_level.rb
@@ -0,0 +1,24 @@
+class BackfillStatusNestLevel < ActiveRecord::Migration[5.2]
+  disable_ddl_transaction!
+
+  def up
+    Rails.logger.info("Populating nest levels for orphaned replies...")
+    Status.select(:id, :account_id).where(reply: true, in_reply_to_id: nil).reorder(nil).in_batches.update_all(nest_level: 1)
+
+    count = 1.0
+    total = Conversation.count
+
+    Conversation.reorder('conversations.id DESC').find_each do |conversation|
+      Rails.logger.info("(#{(count/total*100).to_i}%) Populating nest levels for threads...")
+      conversation.statuses.where(reply: true).reorder('statuses.id ASC').find_each do |status|
+        level = [status.thread&.account_id == status.account_id ? status.thread&.nest_level.to_i : status.thread&.nest_level.to_i + 1, 127].min
+        status.update(nest_level: level) if level != status.nest_level
+      end
+      count += 1
+    end
+  end
+
+  def down
+    true
+  end
+end