debian-mirror-gitlab/lib/gitlab/diff/file_collection/merge_request_diff.rb

79 lines
2.5 KiB
Ruby
Raw Normal View History

2016-09-13 17:45:13 +05:30
module Gitlab
module Diff
module FileCollection
2016-09-29 09:46:39 +05:30
class MergeRequestDiff < Base
def initialize(merge_request_diff, diff_options:)
@merge_request_diff = merge_request_diff
2016-09-13 17:45:13 +05:30
2016-09-29 09:46:39 +05:30
super(merge_request_diff,
project: merge_request_diff.project,
2016-09-13 17:45:13 +05:30
diff_options: diff_options,
2017-09-10 17:25:29 +05:30
diff_refs: merge_request_diff.diff_refs,
fallback_diff_refs: merge_request_diff.fallback_diff_refs)
2016-09-13 17:45:13 +05:30
end
def diff_files
super.tap { |_| store_highlight_cache }
end
2017-08-17 22:00:37 +05:30
def real_size
@merge_request_diff.real_size
end
2016-09-13 17:45:13 +05:30
private
# Extracted method to highlight in the same iteration to the diff_collection.
def decorate_diff!(diff)
diff_file = super
2017-08-17 22:00:37 +05:30
cache_highlight!(diff_file) if cacheable?(diff_file)
2016-09-13 17:45:13 +05:30
diff_file
end
def highlight_diff_file_from_cache!(diff_file, cache_diff_lines)
diff_file.highlighted_diff_lines = cache_diff_lines.map do |line|
Gitlab::Diff::Line.init_from_hash(line)
end
end
#
# If we find the highlighted diff files lines on the cache we replace existing diff_files lines (no highlighted)
# for the highlighted ones, so we just skip their execution.
# If the highlighted diff files lines are not cached we calculate and cache them.
#
2016-11-03 12:29:30 +05:30
# The content of the cache is a Hash where the key identifies the file and the values are Arrays of
2016-09-13 17:45:13 +05:30
# hashes that represent serialized diff lines.
#
def cache_highlight!(diff_file)
2017-08-17 22:00:37 +05:30
item_key = diff_file.file_identifier
2016-09-13 17:45:13 +05:30
2016-11-03 12:29:30 +05:30
if highlight_cache[item_key]
highlight_diff_file_from_cache!(diff_file, highlight_cache[item_key])
2016-09-13 17:45:13 +05:30
else
2016-11-03 12:29:30 +05:30
highlight_cache[item_key] = diff_file.highlighted_diff_lines.map(&:to_hash)
2016-09-13 17:45:13 +05:30
end
end
def highlight_cache
return @highlight_cache if defined?(@highlight_cache)
@highlight_cache = Rails.cache.read(cache_key) || {}
@highlight_cache_was_empty = @highlight_cache.empty?
@highlight_cache
end
def store_highlight_cache
Rails.cache.write(cache_key, highlight_cache) if @highlight_cache_was_empty
end
2017-08-17 22:00:37 +05:30
def cacheable?(diff_file)
2017-09-10 17:25:29 +05:30
@merge_request_diff.present? && diff_file.text? && diff_file.diffable?
2016-09-13 17:45:13 +05:30
end
def cache_key
2016-09-29 09:46:39 +05:30
[@merge_request_diff, 'highlighted-diff-files', diff_options]
2016-09-13 17:45:13 +05:30
end
end
end
end
end