Add latest changes from gitlab-org/gitlab@15-3-stable-eev15.3.0-rc42

1 files changed, 30 insertions, 0 deletions

diff --git a/lib/gitlab/background_migration/backfill_vulnerability_reads_cluster_agent.rb b/lib/gitlab/background_migration/backfill_vulnerability_reads_cluster_agent.rb
new file mode 100644
index 00000000000..728b60f7a0e
--- /dev/null
+++ b/lib/gitlab/background_migration/backfill_vulnerability_reads_cluster_agent.rb
@@ -0,0 +1,30 @@
+# frozen_string_literal: true
+
+module Gitlab
+  module BackgroundMigration
+    # Backfills the `vulnerability_reads.casted_cluster_agent_id` column
+    class BackfillVulnerabilityReadsClusterAgent < Gitlab::BackgroundMigration::BatchedMigrationJob
+      CLUSTER_AGENTS_JOIN = <<~SQL
+        INNER JOIN cluster_agents
+        ON CAST(vulnerability_reads.cluster_agent_id AS bigint) = cluster_agents.id AND
+        vulnerability_reads.project_id = cluster_agents.project_id
+      SQL
+
+      RELATION = ->(relation) do
+        relation
+          .where(report_type: 7)
+      end
+
+      def perform
+        each_sub_batch(
+          operation_name: :update_all,
+          batching_scope: RELATION
+        ) do |sub_batch|
+          sub_batch
+            .joins(CLUSTER_AGENTS_JOIN)
+            .update_all('casted_cluster_agent_id = CAST(vulnerability_reads.cluster_agent_id AS bigint)')
+        end
+      end
+    end
+  end
+end