hotch 0.0.3 → 0.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 61f71a4ba3bef064a412f8b79c1932a84e0a1316
-  data.tar.gz: 59554006ec68fe65bae953d22a85462ec5d766e8
+  metadata.gz: 919be5a42ce8821e1317ae44c23e16f1853765ca
+  data.tar.gz: f841136247fcf929035a0e50995df061388d7ceb
 SHA512:
-  metadata.gz: 29420bf9b209a0d2a989763f24549844d3aa86d7a71420e92e48309aa213cd6357071031b9df6a936f8cfaa5e02fe16a31fa422db4a2f325f4cdb159c5f33821
-  data.tar.gz: 2d9589beac8841df9e8b1e8b9992308310ec3dd142e33a24a53e80a1826a5b92554ad1941d55f72e9526054077fda658ba5f30b7baf174264153b333783b0bdf
+  metadata.gz: 6d042be9fac36eafe88762493b6768a9871f8e482417c6eb2e71259202a487a4b46c134272db5c4a4a8d42c95c4edce0a0667848f12b9092479d0a7fcde4d974
+  data.tar.gz: 3e47fbfee879dae72763653c216ef60589c25d145dac901e411f746185a027ac1fbc029b5b06109b8d1e468375ad128e1629df4089949b1e9b6af3b8d3b61779

data/Gemfile

@@ -2,3 +2,5 @@ source 'https://rubygems.org'
 
 # Specify your gem's dependencies in hotch.gemspec
 gemspec
+
+gem 'stackprof', :github => 'splattael/stackprof', :branch => 'graphviz_filter'

data/README.md

@@ -67,6 +67,22 @@ Set envvar `HOTCH_VIEWER` to enable auto-view after profiling.
 
     $ export HOTCH_VIEWER=eog # use Gnome viewer
 
+### Filter
+
+Set envvar `HOTCH_FILTER` to (regexp) filter frames by its name.
+
+    $ export HOTCH_FILTER=ROM
+    $ export HOTCH_FILTER=Bundler
+
+### Minitest integration
+
+Load `hotch/minitest` in your `test/test_helper.rb` like this:
+
+```ruby
+require 'minitest/autorun'
+require 'hotch/minitest'
+```
+
 ## Caveat
 
 ### Using with bundler

data/examples/bundler.rb

@@ -0,0 +1,4 @@
+require 'bundler'
+require 'hotch'
+
+# run via ruby -rhotch/run bundler.rb

data/lib/hotch.rb

@@ -4,10 +4,12 @@ require 'tmpdir'
 require 'hotch/monkey_patches'
 
 class Hotch
-  attr_reader :name
+  attr_reader :name, :viewer, :filter
 
-  def initialize(name)
+  def initialize(name, viewer: nil, filter: nil)
     @name = name
+    @viewer = viewer
+    @filter = filter
     @reports = []
   end
 
@@ -45,17 +47,18 @@ class Hotch
     yield svg
   end
 
-  def report_at_exit(viewer=ENV['HOTCH_VIEWER'])
+  def report_at_exit
     return if defined? @at_exit_installed
 
     at_exit do
       stop
 
       report do |svg|
-        puts "Profile SVG: #{svg}"
-
         if viewer
+          puts "Profile SVG: #{svg}"
           Kernel.system viewer, svg
+        else
+          puts "Profile SVG: view #{svg} # no HOTCH_VIEWER set"
         end
       end
     end
@@ -80,7 +83,7 @@ class Hotch
   def report_dot(report, dir, file)
     path = File.join(dir, file)
     File.open(path, 'wb') do |fh|
-      report.print_graphviz(nil, fh)
+      report.print_graphviz(filter && Regexp.new(filter), fh)
     end
     path
   end
@@ -92,12 +95,12 @@ class Hotch
   end
 end
 
-def Hotch(aggregate: true)
+def Hotch(name: $0, aggregate: true, viewer: ENV['HOTCH_VIEWER'], filter: ENV['HOTCH_FILTER'])
   hotch = if aggregate
-    $hotch ||= Hotch.new($0)
+    $hotch ||= Hotch.new(name, viewer: viewer, filter: filter)
   else
     caller = Kernel.caller_locations(1).first
-    Hotch.new("#$0:#{caller.path}:#{caller.lineno}")
+    Hotch.new("#{name}:#{caller.path}:#{caller.lineno}", viewer: viewer, filter: filter)
   end
 
   hotch.report_at_exit

data/lib/hotch/minitest.rb

@@ -0,0 +1,20 @@
+class Hotch
+  module Minitest
+    # Usage in test/test_helper.rb:
+    #
+    #     require 'hotch/minitest'
+    #
+    def self.aggregate
+      Module.new do
+        def run_one_method(*args)
+          Hotch(aggregate: true) do
+            super(*args)
+          end
+        end
+      end
+    end
+  end
+end
+
+require "hotch"
+Minitest.singleton_class.prepend Hotch::Minitest.aggregate

data/lib/hotch/run.rb

@@ -1,8 +1,6 @@
 require 'hotch'
 
-viewer = ENV['HOTCH_VIEWER']
-
-hotch = Hotch.new($0)
+hotch = Hotch.new($0, viewer: ENV['HOTCH_VIEWER'], filter: ENV['HOTCH_FILTER'])
 hotch.start
 
-hotch.report_at_exit(viewer)
+hotch.report_at_exit

data/lib/hotch/version.rb

@@ -1,3 +1,3 @@
-module Hotch
-  VERSION = "0.0.3"
+class Hotch
+  VERSION = "0.1.0"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: hotch
 version: !ruby/object:Gem::Version
-  version: 0.0.3
+  version: 0.1.0
 platform: ruby
 authors:
 - Peter Suschlik
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2015-01-05 00:00:00.000000000 Z
+date: 2016-01-13 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: stackprof
@@ -64,9 +64,11 @@ files:
 - LICENSE.txt
 - README.md
 - Rakefile
+- examples/bundler.rb
 - examples/simple.rb
 - hotch.gemspec
 - lib/hotch.rb
+- lib/hotch/minitest.rb
 - lib/hotch/monkey_patches.rb
 - lib/hotch/run.rb
 - lib/hotch/version.rb
@@ -90,7 +92,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
 requirements: []
 rubyforge_project: 
-rubygems_version: 2.2.2
+rubygems_version: 2.4.8
 signing_key: 
 specification_version: 4
 summary: Profile helper