lib/gitlab/graphql/tracers/timer_tracer.rb


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

# frozen_string_literal: true

module Gitlab
  module Graphql
    module Tracers
      # This graphql-ruby tracer records duration for trace events and merges
      # the duration into the trace event's metadata. This way, separate tracers
      # can all use the same duration information.
      #
      # NOTE: TimerTracer should be applied last **after** other tracers, so
      # that it runs first (similar to function composition)
      class TimerTracer
        def self.use(schema)
          schema.tracer(self.new)
        end

        def trace(key, data)
          start_time = Gitlab::Metrics::System.monotonic_time

          result = yield

          duration_s = Gitlab::Metrics::System.monotonic_time - start_time

          data[:duration_s] = duration_s

          result
        end
      end
    end
  end
end