db/post_migrate/20221107184758_sync_new_amount_used_for_ci_namespace_monthly_usages.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 SyncNewAmountUsedForCiNamespaceMonthlyUsages < Gitlab::Database::Migration[2.0]
  restrict_gitlab_migration gitlab_schema: :gitlab_ci

  def up
    namespace_usages = define_batchable_model('ci_namespace_monthly_usages')

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

  def down
    # Non reversible migration.
    # This data migration keeps `new_amount_used` in sync with the old `amount_used`.
    # In case of failure or interruption the migration can be retried.
  end
end