1 files changed, 18 insertions, 0 deletions
diff --git a/db/post_migrate/20211022112202_add_cluster_id_location_index_to_vulnerability_occurrences.rb b/db/post_migrate/20211022112202_add_cluster_id_location_index_to_vulnerability_occurrences.rb
new file mode 100644
index 00000000000..9c40e81f527
--- /dev/null
+++ b/db/post_migrate/20211022112202_add_cluster_id_location_index_to_vulnerability_occurrences.rb
@@ -0,0 +1,18 @@
+# frozen_string_literal: true
+
+class AddClusterIdLocationIndexToVulnerabilityOccurrences < Gitlab::Database::Migration[1.0]
+  disable_ddl_transaction!
+
+  INDEX_NAME = 'index_vulnerability_occurrences_on_location_cluster_id'
+
+  def up
+    add_concurrent_index :vulnerability_occurrences, "(location -> 'cluster_id')",
+                         using: 'GIN',
+                         where: 'report_type = 7',
+                         name: INDEX_NAME
+  end
+
+  def down
+    remove_concurrent_index_by_name :vulnerability_occurrences, INDEX_NAME
+  end
+end