Add latest changes from gitlab-org/gitlab@master

1 files changed, 21 insertions, 0 deletions

diff --git a/app/models/concerns/partitioned_table.rb b/app/models/concerns/partitioned_table.rb
new file mode 100644
index 00000000000..9f1cec5d520
--- /dev/null
+++ b/app/models/concerns/partitioned_table.rb
@@ -0,0 +1,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