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
 |