gitlab-org--gitlab-foss/lib/gitlab/database/background_migration_job.rb

40 lines
1.3 KiB
Ruby

# frozen_string_literal: true
module Gitlab
module Database
class BackgroundMigrationJob < ActiveRecord::Base # rubocop:disable Rails/ApplicationRecord
include EachBatch
self.table_name = :background_migration_jobs
scope :for_migration_class, -> (class_name) { where(class_name: normalize_class_name(class_name)) }
scope :for_migration_execution, -> (class_name, arguments) do
for_migration_class(class_name).where('arguments = ?', arguments.to_json) # rubocop:disable Rails/WhereEquals
end
scope :for_partitioning_migration, -> (class_name, table_name) do
for_migration_class(class_name).where('arguments ->> 2 = ?', table_name)
end
enum status: {
pending: 0,
succeeded: 1
}
def self.mark_all_as_succeeded(class_name, arguments)
self.pending.for_migration_execution(class_name, arguments)
.update_all("status = #{statuses[:succeeded]}, updated_at = NOW()")
end
def self.normalize_class_name(class_name)
return class_name unless class_name.present? && class_name.start_with?('::')
class_name[2..]
end
def class_name=(value)
write_attribute(:class_name, self.class.normalize_class_name(value))
end
end
end
end