sampling_prof 0.1.1 → 0.1.2

checksums.yaml

@@ -1,7 +1,7 @@
 --- 
 SHA1: 
-  metadata.gz: b94654f7e00d9563e9ef60b2564e075709dadaf8
-  data.tar.gz: 074462f2581b586eb9d9a0b3a1d71120045543b0
+  metadata.gz: a0b06029c982d8f0e5935128e7db6bb16526a55a
+  data.tar.gz: 3b82d033647c5c4da28953802d751ea56f821446
 SHA512: 
-  metadata.gz: a2e1c95ffeed6251984b87fd9ffbf45e593c1b062cbb2610c9b9cc4a0925fda262e4ddcdb8d4023df160c8c769f81538fa959cebf98dafaa5b0bab6181f69e09
-  data.tar.gz: 81f8f2e61b78c7a015d57a372bf4dec89b61e9d3e3352a923759f7dd33e1961eea473eff302ee0bca9ca0bcfc376055f383e9a29900d876753bf40154db8c5dd
+  metadata.gz: 199df42d125c17554d633e51b8aaef95dafa1d114ac6650d28eafebea695d1577c2e47493a8cb9bccb4221a3ed0731cd3d89119d475286a8f60084e0148086b3
+  data.tar.gz: 3abfaf96cb25d9c6e946b7e4d51dddbf7117160ee5438ead5190f5d78a80d742c1d317f7233b1b21b76b7f7b37b138b21e0427aea2705b86f545dbf30abcab25

data/lib/sampling_prof.rb

@@ -1,9 +1,13 @@
-require File.join(File.dirname(__FILE__), 'sampling_prof.jar')
+if RUBY_PLATFORM =~ /java/
+  require 'sampling_prof.jar'
+else
+  require 'sampling_prof/internal'
+end
 
 class SamplingProf
   DEFAULT_OUTPUT_FILE = 'profile.txt'
 
-  attr_accessor :output_file
+  attr_writer :output_file
 
   def output_file
     @output_file ||= DEFAULT_OUTPUT_FILE

data/lib/sampling_prof/internal.rb

@@ -0,0 +1,82 @@
+
+class SamplingProf
+  class Sample < Struct.new(:self, :total)
+  end
+
+  class Sampling
+    def initialize(target)
+      @target = target
+      @samples = Hash.new{|h,k| h[k] = [0, 0] }
+      @call_graph = Hash.new{|h,k| h[k] = 0}
+      @nodes = {}
+    end
+
+    def result
+      [@nodes.to_a, @samples.to_a, @call_graph.to_a]
+    end
+
+    def snapshot
+      locations = @target.backtrace_locations
+      from = -1
+      paths = []
+      calls = []
+      locations.reverse.each_with_index do |loc, i|
+        node_id = node_id(loc.path)
+        if i == 0
+          @samples[node_id][0] += 1
+        end
+
+        path = [from, node_id]
+        if !paths.include?(path)
+          paths << path
+          @call_graph[path] += 1
+        end
+        if !calls.include?(node_id)
+          calls << node_id
+          @samples[node_id][1] += 1
+        end
+        from = node_id
+      end
+    end
+
+    def node_id(path)
+      @nodes[path] ||= @nodes.size
+    end
+  end
+
+  def initialize(period)
+    @period = period
+    @running = false
+    @sampling_thread = nil
+  end
+
+  def __start__(&block)
+    unless @running
+      @running = true
+      target = Thread.current
+      @sampling_thread = Thread.start do
+        sampling = Sampling.new(target)
+        loop do
+          break unless @running
+          sampling.snapshot
+          sleep @period
+        end
+        block.call(sampling.result)
+      end
+      true
+    end
+  end
+
+  def stop
+    if @running
+      @running = false
+      @sampling_thread.join
+      @sampling_thread = nil
+      true
+    end
+  end
+
+  def profiling?
+    !!@sampling_thread
+  end
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: sampling_prof
 version: !ruby/object:Gem::Version 
-  version: 0.1.1
+  version: 0.1.2
 platform: ruby
 authors: 
   - Xiao Li
@@ -9,7 +9,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2014-02-13 00:00:00 Z
+date: 2014-02-15 00:00:00 Z
 dependencies: 
   - !ruby/object:Gem::Dependency 
     name: rake-compiler
@@ -39,6 +39,7 @@ files:
   - README.md
   - lib/sampling_prof.jar
   - lib/sampling_prof.rb
+  - lib/sampling_prof/internal.rb
 homepage: https://github.com/xli/sampling_prof
 licenses: 
   - MIT