1 files changed, 41 insertions, 0 deletions
diff --git a/lib/gitlab/background_migration/backfill_projects_with_coverage.rb b/lib/gitlab/background_migration/backfill_projects_with_coverage.rb
new file mode 100644
index 00000000000..ca262c0bd59
--- /dev/null
+++ b/lib/gitlab/background_migration/backfill_projects_with_coverage.rb
@@ -0,0 +1,41 @@
+# frozen_string_literal: true
+
+module Gitlab
+  module BackgroundMigration
+    # Backfill project_ci_feature_usages for a range of projects with coverage
+    class BackfillProjectsWithCoverage
+      class ProjectCiFeatureUsage < ActiveRecord::Base # rubocop:disable Style/Documentation
+        self.table_name = 'project_ci_feature_usages'
+      end
+
+      COVERAGE_ENUM_VALUE = 1
+      INSERT_DELAY_SECONDS = 0.1
+
+      def perform(start_id, end_id, sub_batch_size)
+        report_results = ActiveRecord::Base.connection.execute <<~SQL
+          SELECT DISTINCT project_id, default_branch
+          FROM ci_daily_build_group_report_results
+          WHERE id BETWEEN #{start_id} AND #{end_id}
+        SQL
+
+        report_results.to_a.in_groups_of(sub_batch_size, false) do |batch|
+          ProjectCiFeatureUsage.insert_all(build_values(batch))
+
+          sleep INSERT_DELAY_SECONDS
+        end
+      end
+
+      private
+
+      def build_values(batch)
+        batch.map do |data|
+          {
+            project_id: data['project_id'],
+            feature: COVERAGE_ENUM_VALUE,
+            default_branch: data['default_branch']
+          }
+        end
+      end
+    end
+  end
+end