lib/gitlab/database/as_with_materialized.rb


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36

# frozen_string_literal: true

module Gitlab
  module Database
    # This class is a special Arel node which allows optionally define the `MATERIALIZED` keyword for CTE and Recursive CTE queries.
    class AsWithMaterialized < Arel::Nodes::Binary
      extend Gitlab::Utils::StrongMemoize

      MATERIALIZED = Arel.sql(' MATERIALIZED')
      EMPTY_STRING = Arel.sql('')
      attr_reader :expr

      def initialize(left, right, materialized: true)
        @expr = if materialized && self.class.materialized_supported?
                  MATERIALIZED
                else
                  EMPTY_STRING
                end

        super(left, right)
      end

      # Note: to be deleted after the minimum PG version is set to 12.0
      def self.materialized_supported?
        strong_memoize(:materialized_supported) do
          Gitlab::Database.version.match?(/^1[2-9]\./) # version 12.x and above
        end
      end

      # Note: to be deleted after the minimum PG version is set to 12.0
      def self.materialized_if_supported
        materialized_supported? ? 'MATERIALIZED' : ''
      end
    end
  end
end