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

gitlab.com/gitlab-org/gitlab-foss.git - Unnamed repository; edit this file 'description' to name the repository.
summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
Diffstat (limited to 'lib/gitlab/pagination/keyset/cursor_pager.rb')
-rw-r--r--lib/gitlab/pagination/keyset/cursor_pager.rb38
1 files changed, 38 insertions, 0 deletions
diff --git a/lib/gitlab/pagination/keyset/cursor_pager.rb b/lib/gitlab/pagination/keyset/cursor_pager.rb
new file mode 100644
index 00000000000..0b49aa87a02
--- /dev/null
+++ b/lib/gitlab/pagination/keyset/cursor_pager.rb
@@ -0,0 +1,38 @@
+# frozen_string_literal: true
+
+module Gitlab
+ module Pagination
+ module Keyset
+ class CursorPager < Gitlab::Pagination::Base
+ attr_reader :cursor_based_request_context, :paginator
+
+ def initialize(cursor_based_request_context)
+ @cursor_based_request_context = cursor_based_request_context
+ end
+
+ def paginate(relation)
+ @paginator ||= relation.keyset_paginate(
+ per_page: cursor_based_request_context.per_page,
+ cursor: cursor_based_request_context.cursor
+ )
+
+ paginator.records
+ end
+
+ def finalize(_records = [])
+ # can be called only after executing `paginate(relation)`
+ apply_headers
+ end
+
+ private
+
+ def apply_headers
+ return unless paginator.has_next_page?
+
+ cursor_based_request_context
+ .apply_headers(paginator.cursor_for_next_page)
+ end
+ end
+ end
+ end
+end