1 files changed, 44 insertions, 0 deletions
diff --git a/db/migrate/20230109232316_create_wiki_repository_states.rb b/db/migrate/20230109232316_create_wiki_repository_states.rb
new file mode 100644
index 00000000000..61afc18c0f6
--- /dev/null
+++ b/db/migrate/20230109232316_create_wiki_repository_states.rb
@@ -0,0 +1,44 @@
+# frozen_string_literal: true
+
+class CreateWikiRepositoryStates < Gitlab::Database::Migration[2.0]
+  VERIFICATION_STATE_INDEX_NAME = "index_wiki_repository_states_on_verification_state"
+  PENDING_VERIFICATION_INDEX_NAME = "index_wiki_repository_states_pending_verification"
+  FAILED_VERIFICATION_INDEX_NAME = "index_wiki_repository_states_failed_verification"
+  NEEDS_VERIFICATION_INDEX_NAME = "index_wiki_repository_states_needs_verification"
+
+  enable_lock_retries!
+
+  def up
+    create_table :wiki_repository_states do |t|
+      t.datetime_with_timezone :verification_started_at
+      t.datetime_with_timezone :verification_retry_at
+      t.datetime_with_timezone :verified_at
+      t.references :project_wiki_repository, null: false, index: { unique: true }, foreign_key: { on_delete: :cascade }
+      t.integer :verification_state, default: 0, limit: 2, null: false
+      t.integer :verification_retry_count, limit: 2
+      t.binary :verification_checksum, using: 'verification_checksum::bytea'
+      t.text :verification_failure, limit: 255
+
+      t.index :verification_state,
+        name: VERIFICATION_STATE_INDEX_NAME
+
+      t.index :verified_at,
+        where: "(verification_state = 0)",
+        order: { verified_at: 'ASC NULLS FIRST' },
+        name: PENDING_VERIFICATION_INDEX_NAME
+
+      t.index :verification_retry_at,
+        where: "(verification_state = 3)",
+        order: { verification_retry_at: 'ASC NULLS FIRST' },
+        name: FAILED_VERIFICATION_INDEX_NAME
+
+      t.index :verification_state,
+        where: "(verification_state = 0 OR verification_state = 3)",
+        name: NEEDS_VERIFICATION_INDEX_NAME
+    end
+  end
+
+  def down
+    drop_table :wiki_repository_states
+  end
+end