57 lines
1.6 KiB
Ruby
57 lines
1.6 KiB
Ruby
|
# frozen_string_literal: true
|
||
|
|
|
||
|
|
module Gitlab
|
||
|
|
module Database
|
||
|
|
module BackgroundMigration
|
||
|
|
class BatchedMigration < ActiveRecord::Base # rubocop:disable Rails/ApplicationRecord
|
||
|
|
JOB_CLASS_MODULE = 'Gitlab::BackgroundMigration'
|
||
|
|
BATCH_CLASS_MODULE = "#{JOB_CLASS_MODULE}::BatchingStrategies".freeze
|
||
|
|
|
||
|
|
self.table_name = :batched_background_migrations
|
||
|
|
|
||
|
|
has_many :batched_jobs, foreign_key: :batched_background_migration_id
|
||
|
|
has_one :last_job, -> { order(id: :desc) },
|
||
|
|
class_name: 'Gitlab::Database::BackgroundMigration::BatchedJob',
|
||
|
|
foreign_key: :batched_background_migration_id
|
||
|
|
|
||
|
|
scope :queue_order, -> { order(id: :asc) }
|
||
|
|
|
||
|
|
enum status: {
|
||
|
|
paused: 0,
|
||
|
|
active: 1,
|
||
|
|
aborted: 2,
|
||
|
|
finished: 3
|
||
|
|
}
|
||
|
|
|
||
|
|
def interval_elapsed?
|
||
|
|
last_job.nil? || last_job.created_at <= Time.current - interval
|
||
|
|
end
|
||
|
|
|
||
|
|
def create_batched_job!(min, max)
|
||
|
|
batched_jobs.create!(min_value: min, max_value: max, batch_size: batch_size, sub_batch_size: sub_batch_size)
|
||
|
|
end
|
||
|
|
|
||
|
|
def next_min_value
|
||
|
|
last_job&.max_value&.next || min_value
|
||
|
|
end
|
||
|
|
|
||
|
|
def job_class
|
||
|
|
"#{JOB_CLASS_MODULE}::#{job_class_name}".constantize
|
||
|
|
end
|
||
|
|
|
||
|
|
def batch_class
|
||
|
|
"#{BATCH_CLASS_MODULE}::#{batch_class_name}".constantize
|
||
|
|
end
|
||
|
|
|
||
|
|
def job_class_name=(class_name)
|
||
|
|
write_attribute(:job_class_name, class_name.demodulize)
|
||
|
|
end
|
||
|
|
|
||
|
|
def batch_class_name=(class_name)
|
||
|
|
write_attribute(:batch_class_name, class_name.demodulize)
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|