36 lines
849 B
Ruby
36 lines
849 B
Ruby
|
# 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
|