2020-07-15 00:09:23 +00:00
|
|
|
# frozen_string_literal: true
|
|
|
|
|
|
|
|
module PartitionedTable
|
|
|
|
extend ActiveSupport::Concern
|
|
|
|
|
|
|
|
class_methods do
|
|
|
|
attr_reader :partitioning_strategy
|
|
|
|
|
|
|
|
PARTITIONING_STRATEGIES = {
|
2021-11-24 18:14:31 +00:00
|
|
|
monthly: Gitlab::Database::Partitioning::MonthlyStrategy,
|
|
|
|
sliding_list: Gitlab::Database::Partitioning::SlidingListStrategy
|
2020-07-15 00:09:23 +00:00
|
|
|
}.freeze
|
|
|
|
|
2021-07-13 15:08:38 +00:00
|
|
|
def partitioned_by(partitioning_key, strategy:, **kwargs)
|
2020-07-15 00:09:23 +00:00
|
|
|
strategy_class = PARTITIONING_STRATEGIES[strategy.to_sym] || raise(ArgumentError, "Unknown partitioning strategy: #{strategy}")
|
|
|
|
|
2021-07-13 15:08:38 +00:00
|
|
|
@partitioning_strategy = strategy_class.new(self, partitioning_key, **kwargs)
|
2020-07-15 00:09:23 +00:00
|
|
|
end
|
|
|
|
end
|
|
|
|
end
|