41 lines
1.1 KiB
Ruby
41 lines
1.1 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
module Gitlab
|
|
module Graphql
|
|
module Limit
|
|
class FieldCallCount < ::GraphQL::Schema::FieldExtension
|
|
def resolve(object:, arguments:, context:)
|
|
raise Gitlab::Graphql::Errors::ArgumentError, 'Limit must be specified.' unless limit
|
|
raise Gitlab::Graphql::Errors::LimitError, error_message if increment_call_count(context) > limit
|
|
|
|
yield(object, arguments)
|
|
end
|
|
|
|
private
|
|
|
|
def increment_call_count(context)
|
|
query_id = fetch_query_id(context)
|
|
|
|
context[:call_count] ||= {}
|
|
context[:call_count][query_id] ||= {}
|
|
context[:call_count][query_id][field] ||= 0
|
|
context[:call_count][query_id][field] += 1
|
|
end
|
|
|
|
def fetch_query_id(context)
|
|
context.query.operation_fingerprint
|
|
rescue TypeError
|
|
''
|
|
end
|
|
|
|
def limit
|
|
options[:limit]
|
|
end
|
|
|
|
def error_message
|
|
"\"#{field.graphql_name}\" field can be requested only for #{limit} #{field.owner.graphql_name}(s) at a time."
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|