Class: PrometheusExporter::Instrumentation::MethodProfiler

Inherits:
Object
  • Object
show all
Defined in:
lib/prometheus_exporter/instrumentation/method_profiler.rb

Class Method Summary collapse

Class Method Details

.clear ⇒ Object 



45
46
47
 
# File 'lib/prometheus_exporter/instrumentation/method_profiler.rb', line 45

def self.clear
  Thread.current[:_method_profiler] = nil
end

.patch(klass, methods, name) ⇒ Object 



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
 
# File 'lib/prometheus_exporter/instrumentation/method_profiler.rb', line 7

def self.patch(klass, methods, name)
  patch_source_line = __LINE__ + 3
  patches = methods.map do |method_name|
    <<~RUBY
    unless defined?(#{method_name}__mp_unpatched)
      alias_method :#{method_name}__mp_unpatched, :#{method_name}
      def #{method_name}(*args, &blk)
        unless prof = Thread.current[:_method_profiler]
          return #{method_name}__mp_unpatched(*args, &blk)
        end
        begin
          start = Process.clock_gettime(Process::CLOCK_MONOTONIC)
          #{method_name}__mp_unpatched(*args, &blk)
        ensure
          data = (prof[:#{name}] ||= {duration: 0.0, calls: 0})
          data[:duration] += Process.clock_gettime(Process::CLOCK_MONOTONIC) - start
          data[:calls] += 1
        end
      end
    end
    RUBY
  end.join("\n")

  klass.class_eval patches, __FILE__, patch_source_line
end

.start(transfer = nil) ⇒ Object 



39
40
41
42
43
 
# File 'lib/prometheus_exporter/instrumentation/method_profiler.rb', line 39

def self.start(transfer = nil)
  Thread.current[:_method_profiler] = transfer || {
    __start: Process.clock_gettime(Process::CLOCK_MONOTONIC)
  }
end

.stop ⇒ Object 



49
50
51
52
53
54
55
56
57
 
# File 'lib/prometheus_exporter/instrumentation/method_profiler.rb', line 49

def self.stop
  finish = Process.clock_gettime(Process::CLOCK_MONOTONIC)
  if data = Thread.current[:_method_profiler]
    Thread.current[:_method_profiler] = nil
    start = data.delete(:__start)
    data[:total_duration] = finish - start
  end
  data
end

.transfer ⇒ Object 



33
34
35
36
37
 
# File 'lib/prometheus_exporter/instrumentation/method_profiler.rb', line 33

def self.transfer
  result = Thread.current[:_method_profiler]
  Thread.current[:_method_profiler] = nil
  result
end