trace_tree 0.1.3 → 0.1.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: f3387337a1f93ab9355cc6813b288acfd9272a62
-  data.tar.gz: 8f9398c42e22c9e156689665f08999e25fcf07e0
+  metadata.gz: 5ee0aca382b4cf453d1a686ee889f6dd7717bc55
+  data.tar.gz: 45677cc8e22cf0b570c956ef0ddb4900fab9856f
 SHA512:
-  metadata.gz: a2d35a39fcfb8333ebc8c505ff6a518071c092e8e29fe451ec9935024b5bc12001d726c98231b1387157ebdbb3b5fdd6c8193d6a754f8eaae85eaa626a74cbbb
-  data.tar.gz: 3b2f25e2b6e3d49788ecba875a18bbf1c70dc4c5b92822d2b57c9c979f96facb494b435f618a22db251229d7fccc686a86f9e6630f21439ce354d3cc7ba0a2bf
+  metadata.gz: bfa005442e08b5293d2c6a8b26eb98155e4c678b96e015ed41deb863dbefa9534f17fb09d8a7eaa75c5b6ba09f7d5420dd61ea6054f2b2b959df8898d6e2206a
+  data.tar.gz: beca1bd006605d215528f349ea34cf49dbece736d403b2e8402ce4261320370bc3641e748275e545def5b176b80ed534228074f7aebee73958845468b2815890

data/README.md

@@ -43,6 +43,7 @@ end
 * `:gem => true` by default. Replace the gem paths in source_location with $GemPathN, can make the lines shorter. To see what are replaced, inspect `TraceTree::GemPaths`.
 * `:html => nil` by default. Set it true to generate a html in which a tree constructed with `<ul>`, `<li>`. (No need to set `color`).
 * `:tmp => nil` by default. Set it true or an array of string to specify a tmp file under the default tmp dir of your system. (No need to provide `file` argument)
+* `:timer => nil` by default. Set it true if you want to know how much time spent in tracing and drawing tree. Notice the `file` should be appendable, otherwise the time will overwrite the tree.
 
 ### Example
 

data/lib/trace_tree/node.rb

@@ -13,7 +13,7 @@ class TraceTree
     def initialize trace_point
       @event = trace_point.event
       @method_id = trace_point.method_id
-      @bindings = filter_call_stack trace_point.binding.of_callers!
+      @bindings = filter_call_stack trace_point.binding.of_callers
     end
 
     def << node

data/lib/trace_tree/timer.rb

@@ -0,0 +1,27 @@
+class TraceTree
+  class Timer
+
+    attr_reader :record
+
+    def initialize
+      @record = Hash.new do |h, k|
+        h[k] = []
+      end
+    end
+
+    def [](name)
+      record[name] << Time.now
+    end
+
+    def to_s
+      record.map{|k,v| "#{k}: #{ftime v[0]} ~ #{ftime v[-1]} = #{v[-1] - v[0]}"}
+    end
+
+    private
+
+    def ftime time
+      time.strftime '%F %T %L'
+    end
+
+  end
+end

data/lib/trace_tree/tmp_file.rb

@@ -12,7 +12,7 @@ class TraceTree
     end
 
     def puts *content
-      File.open @tmp, 'w' do |f|
+      File.open @tmp, 'a' do |f|
         f.puts *content
       end
     end

data/lib/trace_tree/version.rb

@@ -1,3 +1,3 @@
 class TraceTree
-  VERSION = "0.1.3"
+  VERSION = "0.1.4"
 end

data/lib/trace_tree.rb

@@ -4,6 +4,7 @@ require 'trace_tree/node'
 require 'trace_tree/short_gem_path'
 require 'trace_tree/color'
 require 'trace_tree/tmp_file'
+require 'trace_tree/timer'
 
 class Binding
   def trace_tree *log, **opt, &to_do
@@ -16,28 +17,39 @@ class TraceTree
   def initialize bi
     @bi = bi
     @trace_points = []
+    @timer = Timer.new
   end
 
   def generate *log, **opt, &to_do
-    @log = dump_location *log, **opt
-    node_class = optional_node opt
+    @opt = opt
+    @log = dump_location *log
+    @node_class = optional_node **opt
     @build_command = opt[:html] ? :tree_html_full : :tree_graph
-
-    tp = TracePoint.trace(:call, :b_call, :raise, :c_call) do |point|
-      trace_points << node_class.new(point) if wanted? point
-    end
-
+    start_trace
     bi.eval('self').instance_eval &to_do
   ensure
-    tp.disable
-    dump_trace_tree
+    stop_trace
   end
 
   private
 
-  attr_reader :bi, :trace_points, :log, :build_command
+  attr_reader :bi, :trace_points, :log, :build_command, :timer, :opt
+
+  def start_trace
+    timer[:to_do]
+    @tp = TracePoint.trace(:call, :b_call, :raise, :c_call) do |point|
+      trace_points << @node_class.new(point) if wanted? point
+    end
+  end
+
+  def stop_trace
+    return unless @tp
+    @tp.disable
+    timer[:to_do]
+    dump_trace_tree
+  end
 
-  def dump_location *log, **opt
+  def dump_location *log
     return TmpFile.new opt[:tmp] if opt[:tmp]
     log.empty? ? STDOUT : log[0]
   end
@@ -50,8 +62,11 @@ class TraceTree
   end
 
   def dump_trace_tree
+    timer[:tree]
     tree = sort(trace_points).send build_command
+    timer[:tree]
     log.puts tree
+    log.puts timer.to_s if opt[:timer]
   end
 
   def wanted? trace_point

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: trace_tree
 version: !ruby/object:Gem::Version
-  version: 0.1.3
+  version: 0.1.4
 platform: ruby
 authors:
 - ken
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2017-02-16 00:00:00.000000000 Z
+date: 2017-02-17 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -115,6 +115,7 @@ files:
 - lib/trace_tree/gem_paths.rb
 - lib/trace_tree/node.rb
 - lib/trace_tree/short_gem_path.rb
+- lib/trace_tree/timer.rb
 - lib/trace_tree/tmp_file.rb
 - lib/trace_tree/tree_graphable.rb
 - lib/trace_tree/tree_htmlable.rb
@@ -140,7 +141,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
 requirements: []
 rubyforge_project: 
-rubygems_version: 2.6.8
+rubygems_version: 2.5.2
 signing_key: 
 specification_version: 4
 summary: Print TracePoint(normal ruby call, block call, raise call, throw call) in