Welcome to mirror list, hosted at ThFree Co, Russian Federation.

last_items.rb « keyset « pagination « graphql « gitlab « lib - gitlab.com/gitlab-org/gitlab-foss.git - Unnamed repository; edit this file 'description' to name the repository.
summaryrefslogtreecommitdiff
blob: 45bf15236c17c2cf9cfa9f54bc2d9c2b95eb478d (plain)
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
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
# frozen_string_literal: true

module Gitlab
  module Graphql
    module Pagination
      module Keyset
        # This class handles the last(N) ActiveRecord call even if a special ORDER BY configuration is present.
        # For the last(N) call, ActiveRecord calls reverse_order, however for some cases it raises
        # ActiveRecord::IrreversibleOrderError error.
        class LastItems
          # rubocop: disable CodeReuse/ActiveRecord
          def self.take_items(scope, count)
            if custom_order = lookup_custom_reverse_order(scope.order_values)
              items = scope.reorder(*custom_order).first(count) # returns a single record when count is nil
              items.is_a?(Array) ? items.reverse : items
            else
              scope.last(count)
            end
          end
          # rubocop: enable CodeReuse/ActiveRecord

          # Detect special ordering and provide the reversed order
          def self.lookup_custom_reverse_order(order_values)
            if ordering_by_merged_at_and_mr_id_desc?(order_values)
              [
                Gitlab::Database.nulls_first_order('merge_request_metrics.merged_at', 'ASC'), # reversing the order
                MergeRequest.arel_table[:id].asc
              ]
            elsif ordering_by_merged_at_and_mr_id_asc?(order_values)
              [
                Gitlab::Database.nulls_first_order('merge_request_metrics.merged_at', 'DESC'),
                MergeRequest.arel_table[:id].asc
              ]
            end
          end

          def self.ordering_by_merged_at_and_mr_id_desc?(order_values)
            order_values.size == 2 &&
              order_values.first.to_s == Gitlab::Database.nulls_last_order('merge_request_metrics.merged_at', 'DESC') &&
              order_values.last.is_a?(Arel::Nodes::Descending) &&
              order_values.last.to_sql == MergeRequest.arel_table[:id].desc.to_sql
          end

          def self.ordering_by_merged_at_and_mr_id_asc?(order_values)
            order_values.size == 2 &&
              order_values.first.to_s == Gitlab::Database.nulls_last_order('merge_request_metrics.merged_at', 'ASC') &&
              order_values.last.is_a?(Arel::Nodes::Descending) &&
              order_values.last.to_sql == MergeRequest.arel_table[:id].desc.to_sql
          end

          private_class_method :ordering_by_merged_at_and_mr_id_desc?
          private_class_method :ordering_by_merged_at_and_mr_id_asc?
        end
      end
    end
  end
end