1
2
3
4
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
32
33
34
|
# frozen_string_literal: true
module Resolvers
class UserNotesCountResolver < BaseResolver
include Gitlab::Graphql::Authorize::AuthorizeResource
type GraphQL::Types::Int, null: true
def resolve
authorize!(object)
load_notes_counts
end
def authorized_resource?(object)
ability = "read_#{object.class.name.underscore}".to_sym
Ability.allowed?(context[:current_user], ability, object)
end
private
def load_notes_counts
BatchLoader::GraphQL.for(object.id).batch(key: :user_notes_count) do |ids, loader, args|
counts = Note.count_for_collection(ids, object.class.name).index_by(&:noteable_id)
ids.each do |id|
loader.call(id, counts[id]&.count || 0)
end
end
end
end
end
::Resolvers::UserNotesCountResolver.prepend_mod
|
