rperf 0.7.0 → 0.9.0

data/lib/rperf/active_job.rb

@@ -1,4 +1,5 @@
 require "rperf"
+require "active_support/concern"
 
 module Rperf::ActiveJobMiddleware
   extend ActiveSupport::Concern

data/lib/rperf/rack.rb

@@ -1,13 +1,35 @@
-require "rperf"
+require_relative "../rperf"
 
 class Rperf::RackMiddleware
-  def initialize(app, label_key: :endpoint)
+  # 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 = "#{env["REQUEST_METHOD"]} #{env["PATH_INFO"]}"
+    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

data/lib/rperf/version.rb

@@ -1,3 +1,3 @@
 module Rperf
-  VERSION = "0.7.0"
+  VERSION = "0.9.0"
 end