summaryrefslogtreecommitdiff
path: root/app/models/concerns/partitioned_table.rb
blob: 9f1cec5d52046c4d26879f97aeef88ca89ff87e7 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
# frozen_string_literal: true

module PartitionedTable
  extend ActiveSupport::Concern

  class_methods do
    attr_reader :partitioning_strategy

    PARTITIONING_STRATEGIES = {
      monthly: Gitlab::Database::Partitioning::MonthlyStrategy
    }.freeze

    def partitioned_by(partitioning_key, strategy:)
      strategy_class = PARTITIONING_STRATEGIES[strategy.to_sym] || raise(ArgumentError, "Unknown partitioning strategy: #{strategy}")

      @partitioning_strategy = strategy_class.new(self, partitioning_key)

      Gitlab::Database::Partitioning::PartitionCreator.register(self)
    end
  end
end