gitlab-foss/lib/gitlab/database/postgres_index.rb

# frozen_string_literal: true

module Gitlab
  module Database
    class PostgresIndex < SharedModel
      include Gitlab::Utils::StrongMemoize

      self.table_name = 'postgres_indexes'
      self.primary_key = 'identifier'
      self.inheritance_column = :_type_disabled

      has_one :bloat_estimate, class_name: 'Gitlab::Database::PostgresIndexBloatEstimate', foreign_key: :identifier
      has_many :reindexing_actions, class_name: 'Gitlab::Database::Reindexing::ReindexAction', foreign_key: :index_identifier
      has_many :queued_reindexing_actions, class_name: 'Gitlab::Database::Reindexing::QueuedAction', foreign_key: :index_identifier

      scope :by_identifier, ->(identifier) do
        unless Gitlab::Database::FULLY_QUALIFIED_IDENTIFIER.match?(identifier)
          raise ArgumentError, "Index name is not fully qualified with a schema: #{identifier}"
        end

        find(identifier)
      end

      # Indexes with reindexing support
      scope :reindexing_support, -> do
        where(exclusion: false, expression: false, type: Gitlab::Database::Reindexing::SUPPORTED_TYPES)
          .not_match("#{Gitlab::Database::Reindexing::ReindexConcurrently::TEMPORARY_INDEX_PATTERN}$")
      end

      scope :without_parent_partitioned_tables, -> do
        partitioned_table = PostgresPartitionedTable.arel_table
        index_table = arel_table

        parent_tables_query = PostgresPartitionedTable
          .where(partitioned_table[:schema].eq(index_table[:schema]))
          .where(partitioned_table[:name].eq(index_table[:tablename]))
          .select(1)

        where('NOT EXISTS (?)', parent_tables_query)
      end

      scope :reindexing_leftovers, -> { match("#{Gitlab::Database::Reindexing::ReindexConcurrently::TEMPORARY_INDEX_PATTERN}$").order(:name) }

      scope :not_match, ->(regex) { where("name !~ ?", regex) }

      scope :match, ->(regex) { where("name ~* ?", regex) }

      scope :not_recently_reindexed, -> do
        recent_actions = Reindexing::ReindexAction.recent.where('index_identifier = identifier')

        where('NOT EXISTS (?)', recent_actions)
      end

      def reset
        reload # rubocop:disable Cop/ActiveRecordAssociationReload
        clear_memoization(:bloat_size)
      end

      def bloat_size
        strong_memoize(:bloat_size) { bloat_estimate&.bloat_size || 0 }
      end

      def relative_bloat_level
        bloat_size / ondisk_size_bytes.to_f
      end

      def to_s
        name
      end
    end
  end
end