1 files changed, 35 insertions, 0 deletions
diff --git a/lib/gitlab/database/postgres_partitioned_table.rb b/lib/gitlab/database/postgres_partitioned_table.rb
new file mode 100644
index 00000000000..5d2eaa22ee4
--- /dev/null
+++ b/lib/gitlab/database/postgres_partitioned_table.rb
@@ -0,0 +1,35 @@
+# frozen_string_literal: true
+
+module Gitlab
+  module Database
+    class PostgresPartitionedTable < ActiveRecord::Base
+      DYNAMIC_PARTITION_STRATEGIES = %w[range list].freeze
+
+      self.primary_key = :identifier
+
+      has_many :postgres_partitions, foreign_key: 'parent_identifier', primary_key: 'identifier'
+
+      scope :by_identifier, ->(identifier) do
+        raise ArgumentError, "Table name is not fully qualified with a schema: #{identifier}" unless identifier =~ /^\w+\.\w+$/
+
+        find(identifier)
+      end
+
+      def self.find_by_name_in_current_schema(name)
+        find_by("identifier = concat(current_schema(), '.', ?)", name)
+      end
+
+      def dynamic?
+        DYNAMIC_PARTITION_STRATEGIES.include?(strategy)
+      end
+
+      def static?
+        !dynamic?
+      end
+
+      def to_s
+        name
+      end
+    end
+  end
+end