Class: Gitlab::Git::DiffCollection
- Inherits:
-
Object
- Object
- Gitlab::Git::DiffCollection
- Includes:
- Enumerable
- Defined in:
- lib/gitlab_git/diff_collection.rb
Constant Summary collapse
- DEFAULT_LIMITS =
{ max_files: 100, max_lines: 5000 }.freeze
Instance Method Summary collapse
- #decorate! ⇒ Object
- #each ⇒ Object
- #empty? ⇒ Boolean
-
#initialize(iterator, options = {}) ⇒ DiffCollection
constructor
A new instance of DiffCollection.
- #overflow? ⇒ Boolean
- #real_size ⇒ Object
- #size ⇒ Object
Constructor Details
#initialize(iterator, options = {}) ⇒ DiffCollection
Returns a new instance of DiffCollection.
8 9 10 11 12 13 14 15 16 17 |
# File 'lib/gitlab_git/diff_collection.rb', line 8 def initialize(iterator, ={}) @iterator = iterator @max_files = .fetch(:max_files, DEFAULT_LIMITS[:max_files]) @max_lines = .fetch(:max_lines, DEFAULT_LIMITS[:max_lines]) @all_diffs = !!.fetch(:all_diffs, false) @line_count = 0 @overflow = false @array = Array.new end |
Instance Method Details
#decorate! ⇒ Object
84 85 86 87 88 |
# File 'lib/gitlab_git/diff_collection.rb', line 84 def decorate! each_with_index do |element, i| @array[i] = yield(element) end end |
#each ⇒ Object
19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 |
# File 'lib/gitlab_git/diff_collection.rb', line 19 def each @iterator.each_with_index do |raw, i| # First yield cached Diff instances from @array if @array[i] yield @array[i] next end # We have exhausted @array, time to create new Diff instances or stop. break if @overflow if !@all_diffs && i >= @max_files @overflow = true break end # Going by the number of files alone it is OK to create a new Diff instance. diff = Gitlab::Git::Diff.new(raw) # If a diff is too large we still want to display some information # about it (e.g. the file path) without keeping the raw data around # (as this would be a waste of memory usage). # # This also removes the line count (from the diff itself) so it # doesn't add up to the total amount of lines. if diff.too_large? diff.prune_large_diff! end @line_count += diff.line_count if !@all_diffs && @line_count >= @max_lines # This last Diff instance pushes us over the lines limit. We stop and # discard it. @overflow = true break end yield @array[i] = diff end end |
#empty? ⇒ Boolean
61 62 63 |
# File 'lib/gitlab_git/diff_collection.rb', line 61 def empty? !@iterator.any? end |
#overflow? ⇒ Boolean
65 66 67 68 |
# File 'lib/gitlab_git/diff_collection.rb', line 65 def overflow? populate! !!@overflow end |
#real_size ⇒ Object
74 75 76 77 78 79 80 81 82 |
# File 'lib/gitlab_git/diff_collection.rb', line 74 def real_size populate! if @overflow "#{size}+" else size.to_s end end |
#size ⇒ Object
70 71 72 |
# File 'lib/gitlab_git/diff_collection.rb', line 70 def size @size ||= count # forces a loop through @iterator end |