summaryrefslogtreecommitdiff
path: root/db/migrate/20211101165656_create_upload_states.rb
blob: 64873f4c9a25ba220bae075c461a053f0c435614 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
# frozen_string_literal: true

class CreateUploadStates < Gitlab::Database::Migration[1.0]
  VERIFICATION_STATE_INDEX_NAME = "index_upload_states_on_verification_state"
  PENDING_VERIFICATION_INDEX_NAME = "index_upload_states_pending_verification"
  FAILED_VERIFICATION_INDEX_NAME = "index_upload_states_failed_verification"
  NEEDS_VERIFICATION_INDEX_NAME = "index_upload_states_needs_verification"

  disable_ddl_transaction!

  def up
    create_table :upload_states, id: false do |t|
      t.datetime_with_timezone :verification_started_at
      t.datetime_with_timezone :verification_retry_at
      t.datetime_with_timezone :verified_at
      t.references :upload, primary_key: true, null: false, 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 :upload_states
  end
end