blob: c0e3016fd3dcd70e1baafbc76d1612fa3e9398d7 (
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
30
31
32
33
34
35
36
37
38
39
40
41
|
# frozen_string_literal: true
module Gitlab
module Database
class BackgroundMigrationJob < SharedModel
include EachBatch
include BulkInsertSafe
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
|
