hotch 0.0.2 → 0.0.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 38cc02398ab2755f8c90da8ae6ea973db46de9b7
-  data.tar.gz: 938117da7ad53717ba52d5507c9b4b8575317068
+  metadata.gz: 61f71a4ba3bef064a412f8b79c1932a84e0a1316
+  data.tar.gz: 59554006ec68fe65bae953d22a85462ec5d766e8
 SHA512:
-  metadata.gz: 91eaa4af1f66add4591182e39b47192325b5842e7bb87176aba19491e071880394d6421288590943683f2103e970e143f07bd535b3eb758fc393ba1919187f17
-  data.tar.gz: 485b40654f50784ec3e08483fac13b4b01b41fefa47790e8d25abf9cf99d32ab720f49169517cda9b662de3ddd72128415c52054b28f1c0850b2313be4e1a8c2
+  metadata.gz: 29420bf9b209a0d2a989763f24549844d3aa86d7a71420e92e48309aa213cd6357071031b9df6a936f8cfaa5e02fe16a31fa422db4a2f325f4cdb159c5f33821
+  data.tar.gz: 2d9589beac8841df9e8b1e8b9992308310ec3dd142e33a24a53e80a1826a5b92554ad1941d55f72e9526054077fda658ba5f30b7baf174264153b333783b0bdf

data/lib/hotch.rb

@@ -1,6 +1,8 @@
 require 'stackprof'
 require 'tmpdir'
 
+require 'hotch/monkey_patches'
+
 class Hotch
   attr_reader :name
 
@@ -35,23 +37,14 @@ class Hotch
     report = @reports.inject(:+) or return
 
     dir = Dir.mktmpdir("hotch.#{name}.")
-    dump = File.open(File.join(dir, "profile.dump"), "wb")
-    svg = File.join(File.join(dir, "profile.svg"))
-
-
-    report.print_graphviz(nil, dump)
-
-    dump.close
 
-    system! "dot", "-Tsvg", "-o", svg, dump.path
+    report_dump(report, dir, "profile.dump")
+    dot = report_dot(report, dir, "profile.dot")
+    svg = convert_svg(dir, dot, "profile.svg")
 
     yield svg
   end
 
-  def system!(*args)
-    Kernel.system(*args) or raise "system call failed: #{args.join(' ')}"
-  end
-
   def report_at_exit(viewer=ENV['HOTCH_VIEWER'])
     return if defined? @at_exit_installed
 
@@ -62,7 +55,7 @@ class Hotch
         puts "Profile SVG: #{svg}"
 
         if viewer
-          system viewer, svg
+          Kernel.system viewer, svg
         end
       end
     end
@@ -75,6 +68,28 @@ class Hotch
   def name
     @name.gsub(/\W+/, '_')
   end
+
+  def report_dump(report, dir, file)
+    path = File.join(dir, file)
+    File.open(path, 'wb') do |fh|
+      report.print_dump(fh)
+    end
+    path
+  end
+
+  def report_dot(report, dir, file)
+    path = File.join(dir, file)
+    File.open(path, 'wb') do |fh|
+      report.print_graphviz(nil, fh)
+    end
+    path
+  end
+
+  def convert_svg(dir, dot, file)
+    svg = File.join(dir, file)
+    system("dot", "-Tsvg", "-o", svg, dot) or raise "dot failed"
+    svg
+  end
 end
 
 def Hotch(aggregate: true)

data/lib/hotch/monkey_patches.rb

@@ -0,0 +1,7 @@
+# Monkey patches
+
+StackProf::Report.class_eval do
+  def print_dump(f = STDOUT)
+    f.puts Marshal.dump(@data.reject{|k,v| k == :files })
+  end
+end

data/lib/hotch/version.rb

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

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: hotch
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.3
 platform: ruby
 authors:
 - Peter Suschlik
@@ -67,6 +67,7 @@ files:
 - examples/simple.rb
 - hotch.gemspec
 - lib/hotch.rb
+- lib/hotch/monkey_patches.rb
 - lib/hotch/run.rb
 - lib/hotch/version.rb
 homepage: https://github.com/splattael/hotch