ree 1.2.1 → 1.2.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 26a7ec18c35a9a6d79aad3f91f111bda3338547bc61f6f979ba226d2d5d393ab
-  data.tar.gz: b330811d3d6623eeb4aab2deb85f9687c3cbe53142987cb03bb3938608d7c786
+  metadata.gz: 9856a98c27c0d4c9a5bc2c62ba473f9d6c19b1b366d1222afe705d78abbf3a4b
+  data.tar.gz: 27d745f77eeedf4dc490bf2006e8b51670df51b07d61849d0111261c176467de
 SHA512:
-  metadata.gz: 10e3c7af5fa2f7255884071f960c4795dc2eb32c7974a8ce8daf2a6d44f7a02a09e088a2ceab09fe71a738694279576c325480193e6c6b896ecb830bbc5e99a0
-  data.tar.gz: 2e3f44006a5aa0764f4ff2dc2addd46a23845ba50abec919e770de25d6d8eac02dc1fa93d9b87d5d1ac59387b74f29f66ab812b36acc057f575c48626a8ad27d
+  metadata.gz: 41161ca3747a4d15af34aebe171254a1a16e273c7d017b2c988264e14a92c5d36f4cb5ca85f131a5c9a19067ca83d2a698c7ca4d6c8095cf907a28a180ed460f
+  data.tar.gz: c3a47bb26c4a9d40b7d4ff52a145bd64c9f9e4d6b310b3b83bc1c2977701638ec399c8dddb052649dd7bbbaccc6760624d6955073ab32d820f31ea9e7721afcc

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    ree (1.2.1)
+    ree (1.2.2)
       commander (~> 5.0.0)
       logger (~> 1.6.5)
 

data/lib/ree/object_compiler.rb

@@ -160,15 +160,35 @@ class Ree::ObjectCompiler
       #{str}
     ruby_eval
 
-    if Ree.benchmark_mode? && object.fn? && !klass.instance_variable_get(:@__ree_benchmark_prepended)
-      klass.instance_variable_set(:@__ree_benchmark_prepended, true)
+    if Ree.benchmark_mode? && object.fn? && !klass.method_defined?(:__call_without_benchmark)
       benchmark_name = "#{object.package_name}/#{object.name}"
-      benchmark_mod = Module.new do
-        define_method(:call) do |*args, **kwargs, &block|
-          Ree::BenchmarkTracer.trace(benchmark_name) { super(*args, **kwargs, &block) }
+
+      wrap_call_with_benchmark = proc do |target_klass|
+        target_klass.alias_method(:__call_without_benchmark, :call)
+
+        target_klass.define_method(:call) do |*args, **kwargs, &block|
+          Ree::BenchmarkTracer.trace(benchmark_name) do
+            __call_without_benchmark(*args, **kwargs, &block)
+          end
+        end
+      end
+
+      if klass.method_defined?(:call)
+        wrap_call_with_benchmark.call(klass)
+      else
+        klass.instance_variable_set(:@__ree_benchmark_wrap, wrap_call_with_benchmark)
+
+        klass.define_singleton_method(:method_added) do |method_name|
+          if method_name == :call && !method_defined?(:__call_without_benchmark)
+            wrap = instance_variable_get(:@__ree_benchmark_wrap)
+            if wrap
+              remove_instance_variable(:@__ree_benchmark_wrap)
+              wrap.call(self)
+            end
+          end
+          super(method_name)
         end
       end
-      klass.prepend(benchmark_mod)
     end
 
     # compile all linked objects

data/lib/ree/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Ree
-  VERSION = "1.2.1"
+  VERSION = "1.2.2"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ree
 version: !ruby/object:Gem::Version
-  version: 1.2.1
+  version: 1.2.2
 platform: ruby
 authors:
 - Ruslan Gatiyatov