52 lines
1.7 KiB
Ruby
52 lines
1.7 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
module Gitlab
|
|
module Pagination
|
|
module Keyset
|
|
class Iterator
|
|
UnsupportedScopeOrder = Class.new(StandardError)
|
|
|
|
def initialize(scope:, use_union_optimization: true, in_operator_optimization_options: nil)
|
|
@scope, success = Gitlab::Pagination::Keyset::SimpleOrderBuilder.build(scope)
|
|
raise(UnsupportedScopeOrder, 'The order on the scope does not support keyset pagination') unless success
|
|
|
|
@order = Gitlab::Pagination::Keyset::Order.extract_keyset_order_object(scope)
|
|
@use_union_optimization = in_operator_optimization_options ? false : use_union_optimization
|
|
@in_operator_optimization_options = in_operator_optimization_options
|
|
end
|
|
|
|
# rubocop: disable CodeReuse/ActiveRecord
|
|
def each_batch(of: 1000)
|
|
cursor_attributes = {}
|
|
|
|
loop do
|
|
current_scope = scope.dup
|
|
relation = order.apply_cursor_conditions(current_scope, cursor_attributes, keyset_options)
|
|
relation = relation.reorder(order) unless @in_operator_optimization_options
|
|
relation = relation.limit(of)
|
|
|
|
yield relation
|
|
|
|
last_record = relation.last
|
|
break unless last_record
|
|
|
|
cursor_attributes = order.cursor_attributes_for_node(last_record)
|
|
end
|
|
end
|
|
# rubocop: enable CodeReuse/ActiveRecord
|
|
|
|
private
|
|
|
|
attr_reader :scope, :order
|
|
|
|
def keyset_options
|
|
{
|
|
use_union_optimization: @use_union_optimization,
|
|
in_operator_optimization_options: @in_operator_optimization_options
|
|
}
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|