74 lines
1.7 KiB
Ruby
74 lines
1.7 KiB
Ruby
|
module Gitlab
|
||
|
|
module Graphql
|
||
|
|
module Connections
|
||
|
|
class KeysetConnection < GraphQL::Relay::BaseConnection
|
||
|
|
def cursor_from_node(node)
|
||
|
|
encode(node[order_field].to_s)
|
||
|
|
end
|
||
|
|
|
||
|
|
def sliced_nodes
|
||
|
|
@sliced_nodes ||=
|
||
|
|
begin
|
||
|
|
sliced = nodes
|
||
|
|
|
||
|
|
sliced = sliced.where(before_slice) if before.present?
|
||
|
|
sliced = sliced.where(after_slice) if after.present?
|
||
|
|
|
||
|
|
sliced
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
def paged_nodes
|
||
|
|
if first && last
|
||
|
|
raise Gitlab::Graphql::Errors::ArgumentError.new("Can only provide either `first` or `last`, not both")
|
||
|
|
end
|
||
|
|
|
||
|
|
if last
|
||
|
|
sliced_nodes.last(limit_value)
|
||
|
|
else
|
||
|
|
sliced_nodes.limit(limit_value)
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
private
|
||
|
|
|
||
|
|
def before_slice
|
||
|
|
if sort_direction == :asc
|
||
|
|
table[order_field].lt(decode(before))
|
||
|
|
else
|
||
|
|
table[order_field].gt(decode(before))
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
def after_slice
|
||
|
|
if sort_direction == :asc
|
||
|
|
table[order_field].gt(decode(after))
|
||
|
|
else
|
||
|
|
table[order_field].lt(decode(after))
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
def limit_value
|
||
|
|
@limit_value ||= [first, last, max_page_size].compact.min
|
||
|
|
end
|
||
|
|
|
||
|
|
def table
|
||
|
|
nodes.arel_table
|
||
|
|
end
|
||
|
|
|
||
|
|
def order_info
|
||
|
|
@order_info ||= nodes.order_values.first
|
||
|
|
end
|
||
|
|
|
||
|
|
def order_field
|
||
|
|
@order_field ||= order_info&.expr&.name || nodes.primary_key
|
||
|
|
end
|
||
|
|
|
||
|
|
def sort_direction
|
||
|
|
@order_direction ||= order_info&.direction || :desc
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|