80 lines
2.7 KiB
Ruby
80 lines
2.7 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
module Gitlab
|
|
module Diff
|
|
module FileCollection
|
|
# Builds a paginated diff file collection and collects pagination
|
|
# metadata.
|
|
#
|
|
# It doesn't handle caching yet as we're not prepared to write/read
|
|
# separate file keys (https://gitlab.com/gitlab-org/gitlab/issues/30550).
|
|
#
|
|
class MergeRequestDiffBatch < MergeRequestDiffBase
|
|
DEFAULT_BATCH_PAGE = 1
|
|
DEFAULT_BATCH_SIZE = 30
|
|
|
|
attr_reader :pagination_data
|
|
|
|
def initialize(merge_request_diff, batch_page, batch_size, diff_options:)
|
|
super(merge_request_diff, diff_options: diff_options)
|
|
|
|
@paginated_collection = load_paginated_collection(batch_page, batch_size, diff_options)
|
|
|
|
@pagination_data = {
|
|
total_pages: @paginated_collection.blank? ? nil : relation.size
|
|
}
|
|
end
|
|
|
|
override :diffs
|
|
def diffs
|
|
strong_memoize(:diffs) do
|
|
@merge_request_diff.opening_external_diff do
|
|
# Avoiding any extra queries.
|
|
collection = @paginated_collection.to_a
|
|
|
|
# The offset collection and calculation is required so that we
|
|
# know how much has been loaded in previous batches, collapsing
|
|
# the current paginated set accordingly (collection limit calculation).
|
|
# See: https://docs.gitlab.com/ee/development/diffs.html#diff-collection-limits
|
|
#
|
|
offset_index = collection.first&.index
|
|
options = diff_options.dup
|
|
|
|
collection =
|
|
if offset_index && offset_index > 0
|
|
offset_collection = relation.limit(offset_index) # rubocop:disable CodeReuse/ActiveRecord
|
|
options[:offset_index] = offset_index
|
|
offset_collection + collection
|
|
else
|
|
collection
|
|
end
|
|
|
|
Gitlab::Git::DiffCollection.new(collection.map(&:to_hash), options)
|
|
end
|
|
end
|
|
end
|
|
|
|
private
|
|
|
|
def relation
|
|
@merge_request_diff.merge_request_diff_files
|
|
end
|
|
|
|
# rubocop: disable CodeReuse/ActiveRecord
|
|
def load_paginated_collection(batch_page, batch_size, diff_options)
|
|
batch_page ||= DEFAULT_BATCH_PAGE
|
|
batch_size ||= DEFAULT_BATCH_SIZE
|
|
|
|
paths = diff_options&.fetch(:paths, nil)
|
|
|
|
paginated_collection = relation.offset(batch_page).limit([batch_size.to_i, DEFAULT_BATCH_SIZE].min)
|
|
paginated_collection = paginated_collection.by_paths(paths) if paths
|
|
|
|
paginated_collection
|
|
end
|
|
# rubocop: enable CodeReuse/ActiveRecord
|
|
end
|
|
end
|
|
end
|
|
end
|