db/post_migrate/20230105172120_sync_new_amount_used_with_amount_used_on_ci_namespace_monthly_usages_table.rb


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19

# frozen_string_literal: true

class SyncNewAmountUsedWithAmountUsedOnCiNamespaceMonthlyUsagesTable < Gitlab::Database::Migration[2.1]
  disable_ddl_transaction!

  restrict_gitlab_migration gitlab_schema: :gitlab_ci

  def up
    # This migration will only run on rollback, there is no need for the positive case
  end

  def down
    project_usages = define_batchable_model('ci_namespace_monthly_usages')

    project_usages.each_batch(of: 500) do |batch|
      batch.where('amount_used > 0').update_all('new_amount_used = amount_used')
    end
  end
end