1 files changed, 30 insertions, 0 deletions
diff --git a/db/post_migrate/20200709101408_schedule_populate_project_snippet_statistics.rb b/db/post_migrate/20200709101408_schedule_populate_project_snippet_statistics.rb
new file mode 100644
index 00000000000..28527e67f4a
--- /dev/null
+++ b/db/post_migrate/20200709101408_schedule_populate_project_snippet_statistics.rb
@@ -0,0 +1,30 @@
+# frozen_string_literal: true
+
+class SchedulePopulateProjectSnippetStatistics < ActiveRecord::Migration[6.0]
+  include Gitlab::Database::MigrationHelpers
+
+  DOWNTIME = false
+  DELAY_INTERVAL = 2.minutes.to_i
+  BATCH_SIZE = 500
+  MIGRATION = 'PopulateProjectSnippetStatistics'
+
+  disable_ddl_transaction!
+
+  def up
+    snippets = exec_query <<~SQL
+      SELECT snippets.id
+      FROM snippets
+      INNER JOIN projects ON projects.id = snippets.project_id
+      WHERE snippets.type = 'ProjectSnippet'
+      ORDER BY projects.namespace_id ASC, snippets.project_id ASC, snippets.id ASC
+    SQL
+
+    snippets.rows.flatten.in_groups_of(BATCH_SIZE, false).each_with_index do |snippet_ids, index|
+      migrate_in(index * DELAY_INTERVAL, MIGRATION, [snippet_ids])
+    end
+  end
+
+  def down
+    # no-op
+  end
+end