summaryrefslogtreecommitdiff
path: root/lib/gitlab/metrics/method_call.rb
blob: faf0d9b6318f8a9da2f260ce9827e1f1597cb0af (plain)
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
module Gitlab
  module Metrics
    # Class for tracking timing information about method calls
    class MethodCall
      attr_reader :real_time, :cpu_time, :call_count

      # name - The full name of the method (including namespace) such as
      #        `User#sign_in`.
      #
      # series - The series to use for storing the data.
      def initialize(name, series)
        @name = name
        @series = series
        @real_time = 0.0
        @cpu_time = 0.0
        @call_count = 0
      end

      # Measures the real and CPU execution time of the supplied block.
      def measure
        start_real = Time.now
        start_cpu = System.cpu_time
        retval = yield

        @real_time += (Time.now - start_real) * 1000.0
        @cpu_time += System.cpu_time.to_f - start_cpu
        @call_count += 1

        retval
      end

      # Returns a Metric instance of the current method call.
      def to_metric
        Metric.new(
          @series,
          {
            duration:     real_time,
            cpu_duration: cpu_time,
            call_count:   call_count
          },
          method: @name
        )
      end

      # Returns true if the total runtime of this method exceeds the method call
      # threshold.
      def above_threshold?
        real_time >= Metrics.method_call_threshold
      end
    end
  end
end