perfetto 0.1.4 → 0.1.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: e191222d4cb7de42dac47a09ed1e2a8db6ec203418e9420b87cf89008d2e24b2
-  data.tar.gz: e703c2fba901b08e837267ff50344d8b0f3dcfee760c752517126993e1188379
+  metadata.gz: 9989cb452ab1dd59f10629d8b29dd9ea035954d24fd013120bbeef3787adfcc9
+  data.tar.gz: 017cfdfd0a27a8e4c93d0d3ec74575b2455a4139491c6ef131608cc389487dd8
 SHA512:
-  metadata.gz: a5a9dc11b2e8ddb7c37a8024629984281f7ca38efb834bb4b939f84ecfe4f9a4197147dd782b88cd849ec4b0267aaf299b2f6ee0ac954773ca79d43b6bf90063
-  data.tar.gz: 844ee5efe6639edf96b4afcb89ec89973eb04c2818bacde53fe9f9139bfe6cfa7a89b2c4177a1f5959b34c11aa7cd5dfef984f3916d13947da11ee8eeaaba1fa
+  metadata.gz: 6af238ffdf8008e4edb49b35bd8e10de5005c7d2cd307158651ef88ba711b28e16a8d4f04d5481a87f9871ee16f5ae573151bda33f0f02475cf62b49e0b2f3be
+  data.tar.gz: a27cf64202e4731cf07a9b3752d6b3afce376585c91b9f2609aa3971ecd06c5d15cbe06ef3cee0935692cb37d4c8cfb9003efaf2afe1e5ce35b970480b5a35d0

data/README.md

@@ -7,7 +7,6 @@ https://ui.perfetto.dev/
 # Usage
 
 ```ruby
-
 require "perfetto"
 
 Perfetto.setup enable_tracing: true
@@ -44,6 +43,10 @@ Perfetto.trace_event_instant_with_debug_info "cpu_work", "example_instant2", "ke
 sleep 0.1
 
 class Foo
+  # Intercept instance methods
+  include Perfetto::Interceptor
+  perfetto_trace_all
+
   def bar(a, b = 1, c: 2)
     yield(a + b + c)
   end
@@ -53,15 +56,26 @@ class Foo
     sleep 0.1
   end
 
-  # Intercept instance methods
-  include Perfetto::Interceptor
-  pftrace_all
+  def self.buf
+    puts "buf"
+    sleep 0.1
+  end
+end
+
+class Bar < Foo
+  def say
+    puts "hello"
+    sleep 0.1
+  end
 end
 
 f = Foo.new
+b = Bar.new
 f.bar(1, 2, c: 3) do |n|
   n.times do |x|
     f.baz x + n
+    Foo.buf
+    b.say
   end
 end
 

data/example/example.rb

@@ -3,7 +3,7 @@
 # rubocop:disable Naming/MethodParameterName
 # rubocop:disable Style/Documentation
 
-require "perfetto"
+require_relative "../lib/perfetto"
 
 Perfetto.setup enable_tracing: true
 
@@ -39,6 +39,10 @@ Perfetto.trace_event_instant_with_debug_info "cpu_work", "example_instant2", "ke
 sleep 0.1
 
 class Foo
+  # Intercept instance methods
+  include Perfetto::Interceptor
+  perfetto_trace_all
+
   def bar(a, b = 1, c: 2)
     yield(a + b + c)
   end
@@ -48,15 +52,26 @@ class Foo
     sleep 0.1
   end
 
-  # Intercept instance methods
-  include Perfetto::Interceptor
-  pftrace_all
+  def self.buf
+    puts "buf"
+    sleep 0.1
+  end
+end
+
+class Bar < Foo
+  def say
+    puts "hello"
+    sleep 0.1
+  end
 end
 
 f = Foo.new
+b = Bar.new
 f.bar(1, 2, c: 3) do |n|
   n.times do |x|
     f.baz x + n
+    Foo.buf
+    b.say
   end
 end
 

data/lib/perfetto/interceptor.rb

@@ -4,31 +4,100 @@ module Perfetto
   # To intercept method calls in other classes
   module Interceptor
     def self.included(base)
-      base.extend(ClassMethods)
+      if Perfetto::Configure.enable_tracing
+        base.extend ImplMethods
+      else
+        base.extend StubMethods
+      end
     end
 
-    # Class methods for the interceptor
-    module ClassMethods
-      if Perfetto::Configure.enable_tracing
-        def pftrace(method_name)
-          original_method = instance_method(method_name)
-          alias_method "pftrace_#{method_name}", method_name
-          define_method(method_name) do |*args, **kwargs, &block|
-            Perfetto.trace_event_begin self.class.name, method_name.to_s
-            original_method.bind(self).call(*args, **kwargs, &block)
-          ensure
-            Perfetto.trace_event_end self.class.name
-          end
+    # Stub methods
+    module StubMethods
+      def perfetto_trace_instance_method(_method_name); end
+      def perfetto_trace_class_method(_method_name); end
+      def perfetto_trace_all; end
+
+      def perfetto_traced_instance_method?(_method_name)
+        false
+      end
+
+      def perfetto_traced_class_method?(_method_name)
+        false
+      end
+
+      def perfetto_traced_instance_methods
+        []
+      end
+
+      def perfetto_traced_class_methods
+        []
+      end
+    end
+
+    # Real Implementation
+    module ImplMethods
+      def perfetto_trace_instance_method(method_name)
+        return if method_name.to_s.start_with? "_pfi_"
+
+        # warn "Perfetto[InstanceMethod]: #{name}##{method_name}"
+        perfetto_traced_instance_methods << method_name
+
+        original_method = instance_method(method_name)
+        alias_method "_pfi_#{method_name}", method_name
+
+        define_method(method_name) do |*args, **kwargs, &block|
+          Perfetto.trace_event_begin self.class.name, method_name.to_s
+          original_method.bind(self).call(*args, **kwargs, &block)
+        ensure
+          Perfetto.trace_event_end self.class.name
         end
+      end
+
+      def perfetto_trace_class_method(method_name)
+        return if method_name.to_s == "singleton_method_added" || method_name.to_s.start_with?("_pfc_")
 
-        def pftrace_all
-          instance_methods(false).each do |method_name|
-            pftrace method_name
-          end
+        # warn "Perfetto[ClassMethod]: #{name}.#{method_name}"
+        perfetto_traced_class_methods << method_name
+
+        original_method = method(method_name)
+        singleton_class.send(:alias_method, "_pfc_#{method_name}", method_name)
+
+        define_singleton_method(method_name) do |*args, **kwargs, &block|
+          Perfetto.trace_event_begin name, method_name.to_s
+          original_method.call(*args, **kwargs, &block)
+        ensure
+          Perfetto.trace_event_end name
+        end
+      end
+
+      def perfetto_trace_all
+        define_singleton_method(:method_added) do |method_name|
+          return if perfetto_traced_instance_method?(method_name)
+
+          perfetto_trace_instance_method method_name
+        end
+
+        define_singleton_method(:singleton_method_added) do |method_name|
+          return if perfetto_traced_class_method?(method_name)
+
+          perfetto_trace_class_method method_name
         end
-      else # When tracing is disabled
-        def pftrace(_method_name); end
-        def pftrace_all; end
+      end
+
+      def perfetto_traced_instance_method?(method_name)
+        perfetto_traced_instance_methods.include? method_name
+      end
+
+      def perfetto_traced_class_method?(method_name)
+        perfetto_traced_class_methods.include? method_name
+      end
+
+      def perfetto_traced_instance_methods
+        @perfetto_traced_instance_methods ||= []
+      end
+
+      def perfetto_traced_class_methods
+        @perfetto_traced_class_methods ||= []
       end
     end
   end

data/lib/perfetto/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Perfetto
-  VERSION = "0.1.4"
+  VERSION = "0.1.5"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: perfetto
 version: !ruby/object:Gem::Version
-  version: 0.1.4
+  version: 0.1.5
 platform: ruby
 authors:
 - Kowalski Dark