Module: Gitlab::NoteableMetadata

Included in:
Dashboard::SnippetsController, Explore::SnippetsController, SnippetsActions, UsersController
Defined in:
lib/gitlab/noteable_metadata.rb

Instance Method Summary collapse

Instance Method Details

#noteable_meta_data(noteable_collection, collection_type) ⇒ Object


5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
# File 'lib/gitlab/noteable_metadata.rb', line 5

def (noteable_collection, collection_type)
  # ActiveRecord uses Object#extend for null relations.
  if !(noteable_collection.singleton_class < ActiveRecord::NullRelation) &&
      noteable_collection.respond_to?(:limit_value) &&
      noteable_collection.limit_value.nil?

    raise 'Collection must have a limit applied for preloading meta-data'
  end

  # map has to be used here since using pluck or select will
  # throw an error when ordering noteables which inserts
  # a new order into the collection.
  # We cannot use reorder to not mess up the paginated collection.
  noteable_ids = noteable_collection.map(&:id)

  return {} if noteable_ids.empty?

  noteable_notes_count = ::Note.count_for_collection(noteable_ids, collection_type)

  noteable_ids.each_with_object({}) do |id, noteable_meta|
    notes = noteable_notes_count.find { |notes| notes.noteable_id == id }

    noteable_meta[id] = ::Noteable::NoteableMeta.new(
      notes.try(:count).to_i
    )
  end
end