1 files changed, 62 insertions, 0 deletions
diff --git a/lib/gitlab/background_migration/populate_has_vulnerabilities.rb b/lib/gitlab/background_migration/populate_has_vulnerabilities.rb
new file mode 100644
index 00000000000..78140b768fc
--- /dev/null
+++ b/lib/gitlab/background_migration/populate_has_vulnerabilities.rb
@@ -0,0 +1,62 @@
+# frozen_string_literal: true
+
+module Gitlab
+  module BackgroundMigration
+    # This class populates missing dismissal information for
+    # vulnerability entries.
+    class PopulateHasVulnerabilities
+      class ProjectSetting < ActiveRecord::Base # rubocop:disable Style/Documentation
+        self.table_name = 'project_settings'
+
+        UPSERT_SQL = <<~SQL
+          WITH upsert_data (project_id, has_vulnerabilities, created_at, updated_at) AS (
+            SELECT projects.id, true, current_timestamp, current_timestamp FROM projects WHERE projects.id IN (%{project_ids})
+          )
+          INSERT INTO project_settings
+          (project_id, has_vulnerabilities, created_at, updated_at)
+          (SELECT * FROM upsert_data)
+          ON CONFLICT (project_id)
+          DO UPDATE SET
+            has_vulnerabilities = true,
+            updated_at = EXCLUDED.updated_at
+        SQL
+
+        def self.upsert_for(project_ids)
+          connection.execute(UPSERT_SQL % { project_ids: project_ids.join(', ') })
+        end
+      end
+
+      class Vulnerability < ActiveRecord::Base # rubocop:disable Style/Documentation
+        include EachBatch
+
+        self.table_name = 'vulnerabilities'
+      end
+
+      def perform(*project_ids)
+        ProjectSetting.upsert_for(project_ids)
+      rescue StandardError => e
+        log_error(e, project_ids)
+      ensure
+        log_info(project_ids)
+      end
+
+      private
+
+      def log_error(error, project_ids)
+        ::Gitlab::BackgroundMigration::Logger.error(
+          migrator: self.class.name,
+          message: error.message,
+          project_ids: project_ids
+        )
+      end
+
+      def log_info(project_ids)
+        ::Gitlab::BackgroundMigration::Logger.info(
+          migrator: self.class.name,
+          message: 'Projects has been processed to populate `has_vulnerabilities` information',
+          count: project_ids.length
+        )
+      end
+    end
+  end
+end