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
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
|
# frozen_string_literal: true
module GraphQL
module Tracing
# This implementation forwards events to a notification handler (i.e.
# ActiveSupport::Notifications or Dry::Monitor::Notifications)
# with a `graphql` suffix.
#
# @see KEYS for event names
class NotificationsTracing
# A cache of frequently-used keys to avoid needless string allocations
KEYS = {
"lex" => "lex.graphql",
"parse" => "parse.graphql",
"validate" => "validate.graphql",
"analyze_multiplex" => "analyze_multiplex.graphql",
"analyze_query" => "analyze_query.graphql",
"execute_query" => "execute_query.graphql",
"execute_query_lazy" => "execute_query_lazy.graphql",
"execute_field" => "execute_field.graphql",
"execute_field_lazy" => "execute_field_lazy.graphql",
"authorized" => "authorized.graphql",
"authorized_lazy" => "authorized_lazy.graphql",
"resolve_type" => "resolve_type.graphql",
"resolve_type_lazy" => "resolve_type.graphql",
}
MAX_KEYS_SIZE = 100
# Initialize a new NotificationsTracing instance
#
# @param [Object] notifications_engine The notifications engine to use
def initialize(notifications_engine)
@notifications_engine = notifications_engine
end
# Sends a GraphQL tracing event to the notification handler
#
# @example
# . notifications_engine = Dry::Monitor::Notifications.new(:graphql)
# . tracer = GraphQL::Tracing::NotificationsTracing.new(notifications_engine)
# . tracer.trace("lex") { ... }
#
# @param [string] key The key for the event
# @param [Hash] metadata The metadata for the event
# @yield The block to execute for the event
def trace(key, metadata, &blk)
prefixed_key = KEYS[key] || "#{key}.graphql"
# Cache the new keys while making sure not to induce a memory leak
if KEYS.size < MAX_KEYS_SIZE
KEYS[key] ||= prefixed_key
end
@notifications_engine.instrument(prefixed_key, metadata, &blk)
end
end
end
end
|
