yuslow 0.0.3 → 0.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 5246c2535d0277df7d0117c0abdb1a9ce9eeb49c434d0df1db0bfa85f71fbdd5
-  data.tar.gz: e2408ca24e820cd5d91cfa10b441bbbf8246e7476f7aab48f657bc6cd82fe1b8
+  metadata.gz: 30c27065a08a4b7140a74cf6aa520a5785648d21516ebc45e80ccaad063df9b0
+  data.tar.gz: 3e664aeffaee96cdc09632ae470df633976e0f811c4f5cde6a16a560aab190ef
 SHA512:
-  metadata.gz: a89ddd6dd6063d40a3614904226728771d15f746a29d326e2b01cc96b2f6895f01e0366c375b6dccb6862a1492cebb94386e4e0a176bf88266d8f61c0c26b482
-  data.tar.gz: 9c0c568bcd841ac8b0e78927b00fa7cc73d1bdf4cb0bc2a6e140679de948d5e853210798137158c6403583f1fa07d6cf7a1a1172418284dbbc108e4126538ec9
+  metadata.gz: adf2247d500446139937b95b7f663caac86cfada0be0a2ba57241e48e149cd9751b8df41608911675f40252965ff01053b8780add96433e5b137d245e8fb586e
+  data.tar.gz: ccd64c90d53730765d6e62c6c86bb2a75e33dae05ea212f07d06722cd06eb49f027a61f512fd85f7356efc4d8a937d04c60346c5125beba4e5e451df8b9806bb

data/README.md

@@ -42,3 +42,10 @@ Thread[1]:
       Object#root_cause elapsed 1000 ms
         Kernel#sleep elapsed 1000 ms
 ```
+
+## Parameters
+| Parameter     | Options           | Default   | Description                                         |
+|---------------|-------------------|-----------|-----------------------------------------------------|
+| **output**    | `false` `:stdout` | `:stdout` | Defines how the profiling results will be presented |
+| **max_level** | `number`, `nil`   | `nil`     | Specifies maximum profiling depth                   |
+| **debug**     | `true` `false`    | false     | Prints debug information while profiling            |

data/lib/yuslow.rb

@@ -5,18 +5,18 @@ require_relative 'yuslow/stdout_printer'
 module Yuslow
   extend self
 
-  def run(debug: false, output: nil)
+  def run(debug: false, output: nil, max_depth: nil)
     raise 'Block expected' unless block_given?
 
-    investigation = Investigation.new debug: debug, printer: printer_from(output)
+    investigation = Investigation.new debug: debug, printer: printer_from(output), max_depth: max_depth
     investigation.start
     yield
     investigation.finish
     investigation
   end
 
-  def investigation(debug: false, output: nil)
-    Investigation.new debug: debug, printer: printer_from(output)
+  def investigation(debug: false, output: nil, max_depth: nil)
+    Investigation.new debug: debug, printer: printer_from(output), max_depth: max_depth
   end
 
   private

data/lib/yuslow/investigation.rb

@@ -1,11 +1,17 @@
 module Yuslow
   class Investigation
-    def initialize(debug: false, printer: nil)
+    def initialize(debug: false, printer: nil, max_depth:)
+      if max_depth
+        raise ArgumentError, 'max_depth must be a number'      if !max_depth.is_a? Numeric
+        raise ArgumentError, 'max_depth cannot be less than 1' if max_depth < 1
+      end
+
       @printer = printer
       @debug = debug
       @tracing = false
       @trace = nil
-      @indent = 0
+      @depth = 0
+      @max_depth = max_depth
       @execution = {}
     end
 
@@ -19,26 +25,28 @@ module Yuslow
         elsif trace_point.defined_class == self.class && trace_point.callee_id == :finish
           @tracing = false
         elsif @tracing
-          if %i(call c_call).include? trace_point.event
-            if @execution[thread_id][:current]
-              @execution[thread_id][:current] =
-                @execution[thread_id][:current].fork object: trace_point.defined_class,
-                                                     method: trace_point.callee_id
-            else
-              operation = Operation.new object: trace_point.defined_class, method: trace_point.callee_id
-              @execution[thread_id][:root]    = operation
-              @execution[thread_id][:current] = operation
+          if %i(call c_call).include?(trace_point.event)
+            if @depth < @max_depth
+              if @execution[thread_id][:current]
+                @execution[thread_id][:current] = @execution[thread_id][:current].fork object: trace_point.defined_class,
+                                                                                       method: trace_point.callee_id
+              else
+                operation = Operation.new object: trace_point.defined_class, method: trace_point.callee_id
+                @execution[thread_id][:root]    = operation
+                @execution[thread_id][:current] = operation
+              end
             end
 
-            debug trace_point, @indent
-            @indent += 1
-
+            debug trace_point, @depth
+            @depth += 1
           elsif %i(return c_return).include? trace_point.event
-            @indent -= 1
-            debug trace_point, @indent
+            @depth -= 1
+            debug trace_point, @depth
 
-            @execution[thread_id][:current]&.complete
-            @execution[thread_id][:current] = @execution[thread_id][:current]&.parent
+            if @depth < @max_depth
+              @execution[thread_id][:current]&.complete
+              @execution[thread_id][:current] = @execution[thread_id][:current]&.parent
+            end
           end
         end
       end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: yuslow
 version: !ruby/object:Gem::Version
-  version: 0.0.3
+  version: 0.1.0
 platform: ruby
 authors:
 - Deyan Dobrinov