Module: TraceCalls

Defined in:
lib/trace_calls.rb,
lib/trace_calls/version.rb,
lib/trace_calls/method_call.rb

Defined Under Namespace

Classes: MethodCall

Constant Summary collapse

VERSION = 
"0.0.1"

Class Method Summary collapse

Class Method Details

.on ⇒ Object 



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
 
# File 'lib/trace_calls.rb', line 8

def self.on
  @@root_call = nil
  scope = {}
  call_stack = []
  calls = Set.new
  trace = TracePoint.new(:call, :line, :return) do |tp|
    case tp.event
    when :return
      last_call = call_stack.pop
      call_stack.last << last_call unless last_call.is_root?
    when :call
      current_call = MethodCall.new(
        tp.method_id,
        tp.defined_class,
        tp.path,
        tp.lineno,
        call_stack.last,
        scope[:lineno]
      )
      call_stack << current_call
      @@root_call = call_stack.last if call_stack.size == 1
    when :line
      scope = {
        event: :line,
        lineno: tp.lineno,
        path: tp.path,
        class: tp.defined_class,
        method_id: tp.method_id
      }
    end
  end

  trace.enable
  yield
  trace.disable
end

.root ⇒ Object 



5
6
7
 
# File 'lib/trace_calls.rb', line 5

def self.root
  @@root_call
end