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