summaryrefslogtreecommitdiff
path: root/db/post_migrate/20220606054503_add_tmp_index_job_artifacts_id_and_expire_at.rb
blob: 28346eb1a9762e375cac51f46c5943d01a2bc1ae (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
# frozen_string_literal: true

class AddTmpIndexJobArtifactsIdAndExpireAt < Gitlab::Database::Migration[2.0]
  disable_ddl_transaction!

  INDEX_NAME = 'tmp_index_ci_job_artifacts_on_id_expire_at_file_type_trace'

  EXPIRE_AT_ON_22_MIDNIGHT_IN_TIMEZONE_OR_TRACE = <<~SQL
    (EXTRACT(day FROM timezone('UTC', expire_at)) IN (21, 22, 23)
    AND EXTRACT(minute FROM timezone('UTC', expire_at)) IN (0, 30, 45)
    AND EXTRACT(second FROM timezone('UTC', expire_at)) = 0)
    OR file_type = 3
  SQL

  def up
    return if Gitlab.com?
    return if index_exists_by_name?(:ci_job_artifacts, INDEX_NAME)

    add_concurrent_index :ci_job_artifacts, :id,
                         where: EXPIRE_AT_ON_22_MIDNIGHT_IN_TIMEZONE_OR_TRACE, name: INDEX_NAME
  end

  def down
    return if Gitlab.com?
    return unless index_exists_by_name?(:ci_job_artifacts, INDEX_NAME)

    remove_concurrent_index_by_name :ci_job_artifacts, INDEX_NAME
  end
end