wip

lib/datadog/di/hook_manager.rb

@@ -95,6 +95,25 @@ def hook_method(cls_name, meth_name, rate_limiter: nil)
         id = next_id
         serializer = self.serializer
 
+        mod = Module.new
+        mod.define_method(meth_name) do |*args, **kwargs|
+          # Arguments may be mutated by the method, therefore
+          # they need to be serialized prior to method invocation.
+          serialized_args = serializer.serialize_args(args, kwargs)
+          rv = nil
+          duration = Benchmark.realtime do
+            rv = saved.bind(self).call(*args, **kwargs)
+          end
+          # The method itself is not part of the stack trace because
+          # we are getting the stack trace from outside of the method.
+          # Add the method in manually as the top frame.
+          loc = saved.source_location
+          method_frame = "#{loc.first}:#{loc.last}:in `#{meth_name}'"
+          callers = [method_frame] + caller
+          yield rv: rv, duration: duration, callers: callers, serialized_args: serialized_args
+          rv
+        end
+
         cls.class_eval do
           saved = instance_method(meth_name)