20 lines
429 B
Ruby
20 lines
429 B
Ruby
|
# frozen_string_literal: true
|
||
|
|
|
||
|
|
module Gitlab
|
||
|
|
module Database
|
||
|
|
module Partitioning
|
||
|
|
def self.register_models(models)
|
||
|
|
registered_models.merge(models)
|
||
|
|
end
|
||
|
|
|
||
|
|
def self.registered_models
|
||
|
|
@registered_models ||= Set.new
|
||
|
|
end
|
||
|
|
|
||
|
|
def self.sync_partitions(models_to_sync = registered_models)
|
||
|
|
MultiDatabasePartitionManager.new(models_to_sync).sync_partitions
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|