rperf 0.6.0 → 0.8.0

data/lib/rperf/active_job.rb

@@ -5,7 +5,7 @@ module Rperf::ActiveJobMiddleware
 
   included do
     around_perform do |job, block|
-      Rperf.label(job: job.class.name) do
+      Rperf.profile(job: job.class.name) do
         block.call
       end
     end

data/lib/rperf/rack.rb

@@ -0,0 +1,37 @@
+require_relative "../rperf"
+
+class Rperf::RackMiddleware
+  # Options:
+  #   label_key: - Symbol key for the endpoint label (default: :endpoint)
+  #   label:     - Proc(env) -> String to customize the label value.
+  #               Default: "METHOD /path" with dynamic segments normalized
+  #               (numeric IDs → :id, UUIDs → :uuid) to keep label cardinality low.
+  #               Set label: :raw to use PATH_INFO as-is (not recommended for
+  #               routes with dynamic segments — each unique path persists in
+  #               memory for the profiling session).
+  #
+  def initialize(app, label_key: :endpoint, label: nil)
+    @app = app
+    @label_key = label_key
+    @label_proc = label
+  end
+
+  UUID_RE = %r{/[0-9a-f]{8}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{12}}i
+  NUMERIC_RE = %r{/\d+}
+
+  def call(env)
+    endpoint = if @label_proc == :raw
+      "#{env["REQUEST_METHOD"]} #{env["PATH_INFO"]}"
+    elsif @label_proc
+      @label_proc.call(env)
+    else
+      path = env["PATH_INFO"]
+        .gsub(UUID_RE, "/:uuid")
+        .gsub(NUMERIC_RE, "/:id")
+      "#{env["REQUEST_METHOD"]} #{path}"
+    end
+    Rperf.profile(@label_key => endpoint) do
+      @app.call(env)
+    end
+  end
+end

data/lib/rperf/sidekiq.rb

@@ -2,7 +2,7 @@ require "rperf"
 
 class Rperf::SidekiqMiddleware
   def call(_worker, job, _queue)
-    Rperf.label(job: job["class"]) do
+    Rperf.profile(job: job["class"]) do
       yield
     end
   end

data/lib/rperf/version.rb

@@ -1,3 +1,3 @@
 module Rperf
-  VERSION = "0.6.0"
+  VERSION = "0.8.0"
 end