visual_call_html 0.0.3 → 0.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 84e84e85f85f91218502afb13d75d4ffc10f1c3fa93e18159d359f125692d55e
-  data.tar.gz: bb666f91822caf00d747522e852a0a0a7efa46a9b232e949802396bc8b25d2a6
+  metadata.gz: 0b3bc8a9e40282d73db18c4fc8c5fe335f67b56215cae90e16798677586c8bf3
+  data.tar.gz: 53ce8bf54cc935ec9e38be6de0c803c99cc6845dc556c28058b35d70666e014d
 SHA512:
-  metadata.gz: d6429ec055f5531109624e23702150a195cd2048aad6057e72c8c5b2a1b964cd3db1d230c3a1c493e1051d6cdac90ef3892c10d8d8d8c77500369de010f81f18
-  data.tar.gz: 7c7c3de40d91b8736046000502a241f355b9e5b83885b4789372d9b208b3e6d1836936e49438b7e4906d4a4642c81baab201bdf8503e09e91d539a9a4e8c1021
+  metadata.gz: 18039c0ce4a18ef08e3a3d22e2ed04162f087f244b65eb64dfa062f165c2991adf9b400a4e647ace2e67a81bc4ef252deb338b0837cd61acdc11b2139568f7b1
+  data.tar.gz: 30b9cd2347484b739fc46d03d2016e054eb2dc3da6b906c19ead252e58df32403a10cd97aab549deb791e294f2764fddd3b92da4ec7acfa2899731a9aa161853

data/lib/call.rb

@@ -0,0 +1,30 @@
+class Call
+  attr_reader :key, :class_id, :method_id
+  attr_accessor :start_time, :finish_time
+
+  def initialize(event)
+    @start_time = Time.now.to_f
+    @class_id = event.defined_class
+    @method_id = event.method_id
+    @key = generate_key
+  end
+
+  def finish
+    @finish_time ||= Time.now.to_f
+    execute_time
+  end
+
+  def execute_time
+    @finish_time.to_f - @start_time
+  end
+
+  def key?(key)
+    @key == key
+  end
+
+  private
+
+  def generate_key
+    [('a'..'z'), (0..9)].map(&:to_a).flatten.shuffle.join
+  end
+end

data/lib/html.rb

@@ -47,7 +47,7 @@ class Html
             // the template we defined earlier
             myDiagram.nodeTemplate =
               $(go.Node, "Horizontal",
-                { background: "#44CCFF" },
+                new go.Binding("background", "color"),
                 $(go.TextBlock, "Default Text",
                   { margin: 12, stroke: "white", font: "bold 16px sans-serif" },
                   new go.Binding("text", "text"))

data/lib/tracer.rb

@@ -1,3 +1,5 @@
+require_relative 'call'
+
 class Tracer
   attr_reader :path, :data
 
@@ -18,28 +20,46 @@ class Tracer
 
   private
 
-  def uniq_key
-    rand(1..1_000).to_s + Time.now.to_f.to_s
-  end
-
   def build_tracer
     TracePoint.new(:call, :return) do |event|
-      next if event.defined_class == self.class
+      next if not_able_to_add?(event)
 
       case event.event
       when :return
-        @options[:display_return_value] && event.return_value && @data.push(
-          key: uniq_key,
-          parent: @path.last,
-          text: "Return: #{event.return_value}"
-        )
-        @path.pop
+        run_when_return(event)
       when :call
-        key = uniq_key
-        data = { key: key, parent: @path.last, text: event.method_id }.compact
-        @data.push(data)
-        @path.push(key)
+        run_when_call(event)
       end
     end
   end
+  
+  def not_able_to_add?(event)
+    event.defined_class == self.class || event.defined_class == TracePoint
+  end
+
+  def run_when_call(event)
+    call = Call.new(event)
+    
+    data = {
+      key: call.key,
+      parent: @path.last&.key,
+      call: call,
+      level: @path.size
+    }.compact
+    
+    @data.push(data)
+    @path.push(call)
+  end
+
+  def run_when_return(event)
+    call = @path.pop
+    call&.finish
+    if @options[:display_return_value] && event.return_value && call
+      @data.push(
+        key: "#{call.key}_return",
+        parent: call.key,
+        return: event.return_value
+      )
+    end
+  end
 end

data/lib/visual_call_html.rb

@@ -1,5 +1,6 @@
 require_relative 'tracer'
 require_relative 'html'
+require_relative 'call'
 
 module VisualCallHtml
   class << self
@@ -11,7 +12,36 @@ module VisualCallHtml
       yield
       tracer.disable
 
-      Html.new(tracer.data).print
+      data = tracer.data.map do |o|
+        next if o[:call].nil? && o[:return].nil?
+
+        text = if !o[:return].nil?
+          o[:return]
+        else
+          r = o[:call]&.class_id.to_s + "#" + o[:call]&.method_id.to_s
+          r += " (#{(o[:call].execute_time * 1000).round}ms)" if options[:display_time]
+        end
+
+        color = if o[:return]
+          "red"
+        else
+          "#44CCFF"
+        end
+
+        {
+          key: o[:key],
+          parent: o[:parent],
+          text: text,
+          color: color,
+          level: o[:level]
+      }.compact
+      end
+
+      if options[:level]
+        data = data.select{|o| o[:level].to_i <= options[:level]}
+      end
+
+      Html.new(data.compact).print
     end
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: visual_call_html
 version: !ruby/object:Gem::Version
-  version: 0.0.3
+  version: 0.1.0
 platform: ruby
 authors:
 - hungkieu
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2020-12-22 00:00:00.000000000 Z
+date: 2022-03-29 00:00:00.000000000 Z
 dependencies: []
 description: Uses the TracePoint class & GoJS to generate a visual representation
   of all the methods called by another method.
@@ -17,6 +17,7 @@ executables: []
 extensions: []
 extra_rdoc_files: []
 files:
+- lib/call.rb
 - lib/html.rb
 - lib/tracer.rb
 - lib/visual_call_html.rb
@@ -39,7 +40,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.2.2
+rubygems_version: 3.3.3
 signing_key:
 specification_version: 4
 summary: This gem helps you see all the other methods called by another method.