perfetto 0.1.6 → 0.1.8

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 336d2d67b2921461a66687b0a06fca3ef0e925cbdc83e028bdbf2863ef3012d4
-  data.tar.gz: 3dc3d8b63d9029dcc10d5250f156663d5248da727652aa0b3cce42a55ac856af
+  metadata.gz: fd838b9d196922765c51a65336d1304c6d9ac9aca456fce60d42b8c72b8b62d4
+  data.tar.gz: 9c1972ee1b588bd0dacaaca9cd81c39d90fd892ff0c57b0286157cf8944c6ae8
 SHA512:
-  metadata.gz: c2771fce6b5aa3b804f6c79d7e64fde0af8238ba25d6325707634d5612a05254df7c6dbf0b34d78d8c2f9c9f502f331e0dd954805167993378f28540cc474c57
-  data.tar.gz: 2a006c86b49bad07a0a0d7117c65a5813dd9a957cb24992aee9b419efe0ac3644a92f6d63338a994b97e9ef800b25dc2bc941d0434a42c20443eb04c5b7aad5d
+  metadata.gz: 92e2da1eceb1263f9d9590fed6130623deee27ab0caa9e4faaf701e3ad340456a6c3730abba3b84da93df94f732a9136ddc8fa9eb7558344701fc132a7f6b69e
+  data.tar.gz: f2f6ba2c22ca06fdaea0e09c6ad5e555d7b580a6630a77ff0132c0de1ee188fd33c6ab76b35fc5819a8ea1f5aed5863323207b232c0dfabbb633b6b524136247

data/README.md

@@ -4,7 +4,9 @@
 
 https://ui.perfetto.dev/
 
-# Usage
+## Usage
+
+### Basic
 
 ```ruby
 require "perfetto"
@@ -83,3 +85,25 @@ Perfetto.stop_tracing "example.pftrace"
 ```
 
 ![example](./example/example.png)
+
+### Rack Middleware
+
+```ruby
+# frozen_string_literal: true
+
+require "sinatra/base"
+require "perfetto"
+
+class Server < Sinatra::Base
+  use Perfetto::Middleware, env_proc: ->(env) { env.to_json }
+
+  get "/" do
+    "Hello World"
+  end
+end
+
+Perfetto.setup enable_tracing: true
+Perfetto.start_tracing
+Server.run!
+Perfetto.stop_tracing "middleware.pftrace"
+```

data/lib/perfetto/interceptor.rb

@@ -1,5 +1,8 @@
 # frozen_string_literal: true
 
+# rubocop:disable Metrics/AbcSize
+# rubocop:disable Metrics/MethodLength
+
 module Perfetto
   # To intercept method calls in other classes
   module Interceptor
@@ -39,7 +42,6 @@ module Perfetto
       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)
@@ -56,17 +58,16 @@ module Perfetto
       end
 
       def perfetto_trace_class_method(method_name)
-        return if method_name.to_s == "singleton_method_added" || method_name.to_s.start_with?("_pfc_")
+        return if method_name.to_s.start_with? "_pfc_"
 
-        # 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|
-          category = self.name
-          task_name = "#{self.name}.#{method_name}"
+          category = name
+          task_name = "#{name}.#{method_name}"
           Perfetto.trace_event_begin category, task_name
           original_method.call(*args, **kwargs, &block)
         ensure
@@ -106,3 +107,5 @@ module Perfetto
     end
   end
 end
+# rubocop:enable Metrics/AbcSize
+# rubocop:enable Metrics/MethodLength

data/lib/perfetto/middleware.rb

@@ -0,0 +1,35 @@
+# frozen_string_literal: true
+
+module Perfetto
+  # Rack middleware
+  class Middleware
+    def initialize(app, options = {})
+      @app = app
+      @options = options
+      @env_proc = options[:env_proc]
+    end
+
+    # rubocop:disable Metrics/MethodLength
+    def perfetto_traced_call(env)
+      category = "RackMiddleware"
+      method = env["REQUEST_METHOD"] || "UNKNOWN"
+      path = env["PATH_INFO"] || "UNKNOWN PATH"
+      task_name = "#{method} #{path}"
+      env_str = @env_proc&.call(env) || { env: "unknown" }.to_json
+
+      Perfetto.trace_event_begin_with_debug_info category, task_name, "env", env_str
+      begin
+        @app.call(env)
+      ensure
+        Perfetto.trace_event_end category
+      end
+    end
+    # rubocop:enable Metrics/MethodLength
+
+    def call(env)
+      @app.call(env) unless Perfetto::Configure.enable_tracing
+
+      perfetto_traced_call(env)
+    end
+  end
+end

data/lib/perfetto/perfetto.rb

@@ -13,7 +13,7 @@ module Perfetto
       start_tracing_native Configure.buffer_size_kb
     end
 
-    def self.stop_tracing(trace_file_name = "#{Time.now.strftime("%Y%m%d%H%M%S")}.pftrace")
+    def self.stop_tracing(trace_file_name = "#{Time.now.strftime("%Y%m%d-%H-%M-%S")}.pftrace")
       stop_tracing_native trace_file_name
     end
 

data/lib/perfetto/version.rb

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

data/lib/perfetto.rb

@@ -1,10 +1,15 @@
 # frozen_string_literal: true
 
+require "json"
+
 require_relative "perfetto/version"
 
 require_relative "perfetto/core_ext/configurable"
 require_relative "perfetto/configure"
 
+# Rack middleware
+require_relative "perfetto/middleware"
+
 # To minimize the overhead of tracing at runtime
 # we determine whether to enable instrumentations
 # at the first call to 'setup' method instead of

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: perfetto
 version: !ruby/object:Gem::Version
-  version: 0.1.6
+  version: 0.1.8
 platform: ruby
 authors:
 - Kowalski Dark
@@ -35,6 +35,7 @@ files:
 - lib/perfetto/configure.rb
 - lib/perfetto/core_ext/configurable.rb
 - lib/perfetto/interceptor.rb
+- lib/perfetto/middleware.rb
 - lib/perfetto/perfetto.rb
 - lib/perfetto/version.rb
 - sig/perfetto.rbs