1 files changed, 26 insertions, 0 deletions
diff --git a/db/migrate/20210420173030_add_verification_indexes_to_terraform_state_versions.rb b/db/migrate/20210420173030_add_verification_indexes_to_terraform_state_versions.rb
new file mode 100644
index 00000000000..2c0d0bee39d
--- /dev/null
+++ b/db/migrate/20210420173030_add_verification_indexes_to_terraform_state_versions.rb
@@ -0,0 +1,26 @@
+# frozen_string_literal: true
+
+class AddVerificationIndexesToTerraformStateVersions < ActiveRecord::Migration[6.0]
+  include Gitlab::Database::MigrationHelpers
+
+  VERIFICATION_STATE_INDEX_NAME = "index_terraform_state_versions_on_verification_state"
+  PENDING_VERIFICATION_INDEX_NAME = "index_terraform_state_versions_pending_verification"
+  FAILED_VERIFICATION_INDEX_NAME = "index_terraform_state_versions_failed_verification"
+  NEEDS_VERIFICATION_INDEX_NAME = "index_terraform_state_versions_needs_verification"
+
+  disable_ddl_transaction!
+
+  def up
+    add_concurrent_index :terraform_state_versions, :verification_state, name: VERIFICATION_STATE_INDEX_NAME
+    add_concurrent_index :terraform_state_versions, :verified_at, where: "(verification_state = 0)", order: { verified_at: 'ASC NULLS FIRST' }, name: PENDING_VERIFICATION_INDEX_NAME
+    add_concurrent_index :terraform_state_versions, :verification_retry_at, where: "(verification_state = 3)", order: { verification_retry_at: 'ASC NULLS FIRST' }, name: FAILED_VERIFICATION_INDEX_NAME
+    add_concurrent_index :terraform_state_versions, :verification_state, where: "(verification_state = 0 OR verification_state = 3)", name: NEEDS_VERIFICATION_INDEX_NAME
+  end
+
+  def down
+    remove_concurrent_index_by_name :terraform_state_versions, VERIFICATION_STATE_INDEX_NAME
+    remove_concurrent_index_by_name :terraform_state_versions, PENDING_VERIFICATION_INDEX_NAME
+    remove_concurrent_index_by_name :terraform_state_versions, FAILED_VERIFICATION_INDEX_NAME
+    remove_concurrent_index_by_name :terraform_state_versions, NEEDS_VERIFICATION_INDEX_NAME
+  end
+end