rails-profiler 0.31.3 → 0.31.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: '09422d8df091081147998c5e266018f8156874c3f2590b4269dfc1bab5b2efcf'
-  data.tar.gz: 3f0ac06620cf6d1f0297528cdce39e80962b736e835d127538968a2aff8b748f
+  metadata.gz: 3ca8756c9efed6c4619db0cbc1226f104deb5f7287f972b137f9a8bf291a40c4
+  data.tar.gz: dcb61c33b3262c081493beb1ee783923b3c7b3ac8d4cd26ef9195ea4e3ff182d
 SHA512:
-  metadata.gz: 7d74679898828ec69cea3718945c08e8c8ce4aec93d1ccb8942de28617a888ff2dfc2fbde4e4c01892f9a05a8723f713907a87b378b3be5dd11d5cb7cec7500b
-  data.tar.gz: 918333cbc7efbb6c61217adb4c52ced40be60a62b621fb828e942878598defd92d69bd36672bf515bcd5b6f9e4ed82869268fe704f64b674c309e2f61846fe6a
+  metadata.gz: 2576d031d1b4d3e6e462392ed5ca00bc83c12eaed23b9ec07ea842bd3365e373d7eff1b4c53ae73e779366fc947ad70f01f15e8b1108c8fb36e85409986ebd22
+  data.tar.gz: 4b4d5914c82c06e58fd1d8e14157a28b2d0a9bbc00d15f574a5a5bdb0e782c0e9957042f25c38ba2d6cbc93c159b0b4d2f921667d552e397f37fed896bcc2245

data/CHANGELOG.md

@@ -19,6 +19,62 @@ every commit of every tag interval is accounted for one way or the other.
 
 ## [Unreleased]
 
+## [0.31.4] - 2026-10-05
+
+<!-- stamped -->
+
+### Fixed
+
+- **Storage:** A save into the file store costs the same whatever the number of profiles on disk:
+  it listed the directory and read the size of every file on each save, about 5 ms with 500
+  profiles and 100 ms with 10,000, now under 1 ms. The store keeps an index next to the profiles
+  (`.profiles-index.jsonl`, with its lock `.profiles-index.lock`), which the processes writing to
+  the directory share; a missing or damaged one is rebuilt from the profile files, so the profiles
+  already saved are found without any migration. Temporary files left by a killed process are now
+  removed. The index relies on `flock`: a file store shared between hosts over NFS is not
+  supported, use Redis there.
+- **Storage:** `config.max_profiles` (100) is honoured by every backend, the first saved profiles
+  evicted first (a job saved when it ends counts as new, whenever it started): only the memory
+  store applied it, so the file store grew to `max_size` (100 MB,
+  tens of thousands of profiles), SQLite without limit and Redis up to its TTL. In the test
+  environment, where every example saves a profile, there is no cap on the count unless the
+  application sets one.
+- **Storage:** Finding the children of a page (its AJAX sub-requests and its jobs) reads only
+  them, through an index of their parent, instead of reading and parsing every stored profile
+  (about 260 ms with 1,500 profiles of 17 KB in the file store, now under 1 ms). Every profile page
+  and toolbar paid it, once for the child jobs and once more for the AJAX tab; they now read the
+  children once.
+- **Dashboard:** The HTTP, job, console and test lists are filtered and paged by the storage, and
+  list summaries of the profiles (without the bodies, params, headers and nested collector data):
+  they loaded the 1,000 newest profiles in full on every page (about 200 ms with 1,500 profiles of
+  17 KB, now under 10 ms), showed nothing past the thousandth, said there was no next page there,
+  and missed the profiles of one type behind 1,000 newer ones of another. The MCP tools and
+  resources that list one type of profile (jobs, console, tests) and the test report ask the
+  storage for that type too, and found nothing behind 500 newer profiles of another type.
+- **AJAX:** The AJAX tab shows the sub-requests of a page whatever `config.collectors` holds: it
+  only appeared with `AjaxCollector` in the list, which the default list does not have. The jobs a
+  page enqueued are no longer counted as AJAX requests: they stay under its child jobs. The MCP
+  tool `get_profile_ajax` finds the sub-requests too: it read the AJAX data saved with the page,
+  before any sub-request.
+- **Storage:** The storage is created once, under a lock: the first requests of a threaded server
+  could each create their own store (16 for 16 threads in the spec), and with the memory store the
+  profiles saved in all but one were lost. Several file, SQLite or Redis stores share their data
+  and lost nothing.
+- **Storage:** A store that cannot be created (its index or lock replaced by a symbolic link, say)
+  is reported once per process instead of with a backtrace on every request: the profiles are not
+  saved and no toolbar is added, the dashboard and its API answer `503` with the cause in one line
+  (no path) instead of a `500`, the MCP tools an error with it, and the store is created again
+  once the cause is gone.
+- **Upgrading:** the file, SQLite and Redis stores now keep 100 profiles outside the test
+  environment, and **the first save after the upgrade removes the profiles past the cap**, the
+  first saved first: of 3,000 profiles written by 0.31.1 in the file store, about 80 remain, and
+  that first save takes about a second, during which the other processes writing to the store
+  wait. To keep them, set `config.max_profiles` (or `storage_options[:max_profiles]`) **before**
+  upgrading: to a higher number, or to `nil` for the earlier behaviour, no cap on the count (the
+  file store then keeps up to `max_size`, Redis up to its TTL). The memory store keeps 100 with
+  `nil`, as before. A Redis store written by an earlier version is indexed once, by one process,
+  on first use, which also removes the profiles past the cap then (about 0.4 s for 20,000).
+
 ## [0.31.3] - 2026-10-05
 
 <!-- stamped -->

data/app/controllers/profiler/api/console_controller.rb

@@ -4,17 +4,7 @@ module Profiler
   module Api
     class ConsoleController < ApplicationController
       def index
-        limit  = (params[:limit]  || 50).to_i
-        offset = (params[:offset] || 0).to_i
-        all    = Profiler.storage.list(limit: 1000, offset: 0)
-        console = all.select { |p| p.profile_type == "console" }
-        page   = console.drop(offset).first(limit + 1)
-        render json: {
-          profiles: page.first(limit).map(&:to_h),
-          limit:    limit,
-          offset:   offset,
-          has_more: page.size > limit
-        }
+        render_profile_page(type: "console")
       end
 
       def show

data/app/controllers/profiler/api/jobs_controller.rb

@@ -4,17 +4,7 @@ module Profiler
   module Api
     class JobsController < ApplicationController
       def index
-        limit  = (params[:limit]  || 50).to_i
-        offset = (params[:offset] || 0).to_i
-        all    = Profiler.storage.list(limit: 1000, offset: 0)
-        jobs   = all.select { |p| p.profile_type == "job" }
-        page   = jobs.drop(offset).first(limit + 1)
-        render json: {
-          profiles: page.first(limit).map(&:to_h),
-          limit:    limit,
-          offset:   offset,
-          has_more: page.size > limit
-        }
+        render_profile_page(type: "job")
       end
 
       def show

data/app/controllers/profiler/api/profiles_controller.rb

@@ -4,20 +4,16 @@ module Profiler
   module Api
     class ProfilesController < ApplicationController
       def index
-        limit  = (params[:limit]  || 50).to_i
-        offset = (params[:offset] || 0).to_i
-        all    = Profiler.storage.list(limit: 1000, offset: 0)
         # all_types is an opt-in used by the cluster proxy so it can mirror the full
-        # storage.list contract; the dashboard relies on the default http-only filter.
-        scope  = all_types? ? all : all.select { |p| p.profile_type == "http" }
-        scope  = scope.select { |p| p.parent_token == params[:parent_token] } if params[:parent_token].present?
-        page   = scope.drop(offset).first(limit + 1)
-        render json: {
-          profiles: page.first(limit).map(&:to_h),
-          limit:    limit,
-          offset:   offset,
-          has_more: page.size > limit
-        }
+        # storage.list contract (every type, or the one given as type, full profiles); the
+        # dashboard relies on the default: http profiles, as summaries.
+        if params[:parent_token].present?
+          render_children_page
+        elsif all_types?
+          render_profile_page(type: profile_type_param, summary: false)
+        else
+          render_profile_page(type: "http")
+        end
       end
 
       def show
@@ -57,22 +53,16 @@ module Profiler
         %w[1 true].include?(params[:all_types].to_s)
       end
 
-      def recalculate_ajax_data(profile)
-        # Find AJAX collector in the configured collectors
-        ajax_collector_class = Profiler::Collectors::AjaxCollector
-
-        if Profiler.configuration.collectors.include?(ajax_collector_class)
-          collector = ajax_collector_class.new(profile)
-          collector.collect
+      def profile_type_param
+        params[:type].to_s.match?(/\A[a-z_]{1,32}\z/) ? params[:type].to_s : nil
+      end
 
-          # Update tab metadata to reflect has_data status
-          if profile.instance_variable_get(:@collectors_metadata)
-            ajax_tab = profile.instance_variable_get(:@collectors_metadata).find { |tab| tab[:key] == 'ajax' }
-            if ajax_tab
-              ajax_tab[:has_data] = collector.has_data?
-            end
-          end
-        end
+      # The children of a page, oldest last, read through the store's parent index.
+      def render_children_page
+        limit, offset = page_params
+        children = Profiler.storage.find_by_parent(params[:parent_token].to_s).reverse
+        children = children.select { |p| p.profile_type == "http" } unless all_types?
+        render_page(children.drop(offset).first(limit + 1), limit, offset)
       end
     end
   end

data/app/controllers/profiler/api/tests_controller.rb

@@ -4,17 +4,7 @@ module Profiler
   module Api
     class TestsController < ApplicationController
       def index
-        limit  = (params[:limit]  || 50).to_i
-        offset = (params[:offset] || 0).to_i
-        all    = Profiler.storage.list(limit: 1000, offset: 0)
-        tests  = all.select { |p| p.profile_type == "test" }
-        page   = tests.drop(offset).first(limit + 1)
-        render json: {
-          profiles: page.first(limit).map(&:to_h),
-          limit:    limit,
-          offset:   offset,
-          has_more: page.size > limit
-        }
+        render_profile_page(type: "test")
       end
 
       def show

data/app/controllers/profiler/api/toolbar_controller.rb

@@ -24,26 +24,6 @@ module Profiler
           events_cursor: events_cursor
         }
       end
-
-      private
-
-      def recalculate_ajax_data(profile)
-        # Find AJAX collector in the configured collectors
-        ajax_collector_class = Profiler::Collectors::AjaxCollector
-
-        if Profiler.configuration.collectors.include?(ajax_collector_class)
-          collector = ajax_collector_class.new(profile)
-          collector.collect
-
-          # Update tab metadata to reflect has_data status
-          if profile.instance_variable_get(:@collectors_metadata)
-            ajax_tab = profile.instance_variable_get(:@collectors_metadata).find { |tab| tab[:key] == 'ajax' }
-            if ajax_tab
-              ajax_tab[:has_data] = collector.has_data?
-            end
-          end
-        end
-      end
     end
   end
 end

data/app/controllers/profiler/application_controller.rb

@@ -6,6 +6,9 @@ module Profiler
   class ApplicationController < ActionController::Base
     layout "profiler/application"
 
+    # A store that cannot be created: the cause in one line, as a 503, for every page and API.
+    rescue_from Profiler::Storage::Unavailable::Error, with: :storage_unavailable
+
     before_action :check_authorization
     before_action :authorize_request
     # Declared after the access checks, so that an unauthorized request gets its 403 first.
@@ -48,6 +51,14 @@ module Profiler
       deny("Missing #{Profiler::FORGERY_PROTECTION_HEADER} header or CSRF token")
     end
 
+    def storage_unavailable(error)
+      if controller_path.start_with?("profiler/api/")
+        render json: { error: error.message }, status: :service_unavailable
+      else
+        render plain: error.message, status: :service_unavailable
+      end
+    end
+
     def deny(message)
       if controller_path.start_with?("profiler/api/")
         render json: { error: message }, status: :forbidden
@@ -56,22 +67,51 @@ module Profiler
       end
     end
 
+    # A page of one type of profile, filtered, ordered and paged by the store: one profile more
+    # than the limit tells whether there is a next page, at any offset.
+    def render_profile_page(type:, summary: true)
+      limit, offset = page_params
+      render_page(Profiler.storage.list(limit: limit + 1, offset: offset, type: type, summary: summary), limit, offset)
+    end
+
+    def render_page(profiles, limit, offset)
+      render json: {
+        profiles: profiles.first(limit).map(&:to_h),
+        limit:    limit,
+        offset:   offset,
+        has_more: profiles.size > limit
+      }
+    end
+
+    def page_params
+      [[(params[:limit] || 50).to_i, 0].max, [(params[:offset] || 0).to_i, 0].max]
+    end
+
+    # The children of a page, read once per request: the child jobs and the AJAX tab share them.
+    def child_profiles(profile)
+      @child_profiles ||= {}
+      @child_profiles[profile.token] ||= (@resolved_storage || Profiler.storage).find_by_parent(profile.token)
+    end
+
+    # The AJAX sub-requests are saved after the page: their tab is computed here, whatever the
+    # configured collector list (Profiler::AjaxData).
+    def recalculate_ajax_data(profile)
+      Profiler::AjaxData.attach(profile, child_profiles(profile))
+    end
+
     def build_child_jobs(profile)
-      storage = @resolved_storage || Profiler.storage
-      storage.find_by_parent(profile.token)
-             .select { |p| p.profile_type == "job" }
-             .map do |j|
-               job_data = j.collector_data("job") || {}
-               {
-                 token: j.token,
-                 job_class: j.path,
-                 job_id: job_data["job_id"],
-                 queue: job_data["queue"],
-                 status: job_data["status"],
-                 duration: j.duration,
-                 started_at: j.started_at&.iso8601
-               }
-             end
+      child_profiles(profile).select { |p| p.profile_type == "job" }.map do |j|
+        job_data = j.collector_data("job") || {}
+        {
+          token: j.token,
+          job_class: j.path,
+          job_id: job_data["job_id"],
+          queue: job_data["queue"],
+          status: job_data["status"],
+          duration: j.duration,
+          started_at: j.started_at&.iso8601
+        }
+      end
     end
 
     def build_parent_summary(profile)

data/app/controllers/profiler/profiles_controller.rb

@@ -144,22 +144,6 @@ module Profiler
       end
     end
 
-    def recalculate_ajax_data(profile)
-      ajax_collector_class = Profiler::Collectors::AjaxCollector
-
-      if Profiler.configuration.collectors.include?(ajax_collector_class)
-        collector = ajax_collector_class.new(profile, storage: @resolved_storage || Profiler.storage)
-        collector.collect
-
-        if profile.instance_variable_get(:@collectors_metadata)
-          ajax_tab = profile.instance_variable_get(:@collectors_metadata).find { |tab| tab[:key] == 'ajax' }
-          if ajax_tab
-            ajax_tab[:has_data] = collector.has_data?
-          end
-        end
-      end
-    end
-
     def filter_profiles(profiles)
       filtered = profiles
 

data/lib/profiler/ajax_data.rb

@@ -0,0 +1,40 @@
+# frozen_string_literal: true
+
+require_relative "collectors/ajax_collector"
+
+module Profiler
+  # The AJAX tab of a page: its sub-requests are saved after the page itself, so the tab is
+  # computed when the page is shown, from its children, whatever the configured collector list.
+  # The children are the ones the caller already read (the same find_by_parent gives the child
+  # jobs): only the http ones are sub-requests, the jobs stay in child_jobs.
+  module AjaxData
+    # Hands the children already read to the AjaxCollector in place of the storage.
+    Children = Struct.new(:profiles) do
+      def find_by_parent(_parent_token)
+        profiles
+      end
+    end
+
+    module_function
+
+    # Adds the ajax collector data to profile, and its tab when it has none: a page without
+    # sub-requests and without the tab is left as it is.
+    def attach(profile, children)
+      requests = children.select { |child| child.profile_type == "http" }
+      # A copy: the memory store hands out the very tabs it keeps.
+      tabs = profile.collectors_metadata = (profile.collectors_metadata || []).map(&:dup)
+      tab = tabs.find { |entry| (entry[:key] || entry["key"]).to_s == "ajax" }
+      return if requests.empty? && tab.nil?
+
+      collector = Collectors::AjaxCollector.new(profile, storage: Children.new(requests))
+      collector.collect
+      if tab.nil?
+        profile.add_collector_metadata(collector)
+      elsif tab.key?("key")
+        tab["has_data"] = collector.has_data?
+      else
+        tab[:has_data] = collector.has_data?
+      end
+    end
+  end
+end

data/lib/profiler/cluster/slave_proxy.rb

@@ -33,11 +33,13 @@ module Profiler
 
       # Storage-compatible interface for MCP query tools
 
-      def list(limit: 50, offset: 0, **)
-        # all_types mirrors Profiler.storage.list, which returns every profile type;
-        # without it the proxy would only ever see http profiles (see ProfilesController#index).
-        data = get_json("/_profiler/api/profiles", limit: limit, offset: offset, all_types: true)
-        Array(data["profiles"]).map { |h| profile_from_api(h) }
+      def list(limit: 50, offset: 0, type: nil, **)
+        # all_types mirrors Profiler.storage.list, which returns every profile type, or the one
+        # given as type; without it the proxy would only ever see http profiles (see
+        # ProfilesController#index). A slave of an earlier version ignores type: filtered here too.
+        data = get_json("/_profiler/api/profiles", limit: limit, offset: offset, all_types: true, type: type)
+        profiles = Array(data["profiles"]).map { |h| profile_from_api(h) }
+        type ? profiles.select { |p| p.profile_type == type.to_s } : profiles
       end
 
       # A token that is not one the gem issues is not found, without asking the slave.

data/lib/profiler/configuration.rb

@@ -29,13 +29,15 @@ module Profiler
 
     # Options whose default depends on the Rails environment: the railtie sets them only when
     # the application has not assigned them itself (see #default).
-    RAILS_DEFAULTED = %i[enabled storage track_tests].freeze
+    RAILS_DEFAULTED = %i[enabled storage track_tests max_profiles].freeze
+
+    BACKEND_LOCK = Mutex.new
 
     attr_accessor :storage_options, :collectors,
                   :skip_paths, :slow_query_threshold, :max_queries_warning, :sql_backtrace,
                   :track_memory, :allocated_objects_warning_threshold,
                   :mcp_enabled, :mcp_transport, :mcp_port,
-                  :authorization_mode, :max_profiles, :extension_cors_enabled,
+                  :authorization_mode, :extension_cors_enabled,
                   :cors_allowed_origins, :api_forgery_protection, :frame_ancestors,
                   :track_ajax, :ajax_skip_paths,
                   :track_http, :slow_http_threshold, :http_skip_hosts, :http_backtrace_depth,
@@ -51,7 +53,7 @@ module Profiler
                   :cluster_master, :cluster_secret, :cluster_require_secret,
                   :cluster_allow_insecure_http
 
-    attr_reader :authorize_block, :enabled, :track_tests
+    attr_reader :authorize_block, :enabled, :track_tests, :max_profiles
 
     def initialize
       @assigned = []
@@ -181,6 +183,13 @@ module Profiler
       @track_tests = value
     end
 
+    # How many profiles a store keeps, the oldest evicted first; nil for no cap on the count (the
+    # memory store still keeps 100). The railtie leaves it nil in the test environment.
+    def max_profiles=(value)
+      @assigned |= [:max_profiles]
+      @max_profiles = value
+    end
+
     # Sets one of RAILS_DEFAULTED to value unless the application assigned it, in
     # config/application.rb for instance, which runs before the railtie's initializers.
     def default(option, value)
@@ -246,8 +255,10 @@ module Profiler
       @storage_backend = nil
     end
 
+    # Under a lock of its own, for a caller that asks the configuration directly; Profiler.storage
+    # takes it inside its own, always in that order.
     def storage_backend
-      @storage_backend ||= build_storage_backend
+      @storage_backend || BACKEND_LOCK.synchronize { @storage_backend ||= build_storage_backend }
     end
 
     private

data/lib/profiler/mcp/resources/failing_tests.rb

@@ -5,7 +5,7 @@ module Profiler
     module Resources
       class FailingTests
         def self.call
-          profiles = Profiler.storage.list(limit: 500)
+          profiles = Profiler.storage.list(limit: 500, type: "test")
           failing = profiles.select do |p|
             next false unless p.profile_type == "test"
             test_data = p.collector_data("test") || {}

data/lib/profiler/mcp/resources/recent_console.rb

@@ -5,8 +5,8 @@ module Profiler
     module Resources
       class RecentConsole
         def self.call
-          profiles = Profiler.storage.list(limit: 200)
-          consoles = profiles.select { |p| p.profile_type == "console" }.first(50)
+          profiles = Profiler.storage.list(limit: 50, type: "console")
+          consoles = profiles.select { |p| p.profile_type == "console" }
 
           data = consoles.map do |profile|
             console_data = profile.collector_data("console") || {}

data/lib/profiler/mcp/resources/recent_jobs.rb

@@ -5,8 +5,8 @@ module Profiler
     module Resources
       class RecentJobs
         def self.call
-          profiles = Profiler.storage.list(limit: 200)
-          jobs = profiles.select { |p| p.profile_type == "job" }.first(50)
+          profiles = Profiler.storage.list(limit: 50, type: "job")
+          jobs = profiles.select { |p| p.profile_type == "job" }
 
           data = jobs.map do |profile|
             job_data = profile.collector_data("job") || {}

data/lib/profiler/mcp/resources/slow_tests.rb

@@ -5,7 +5,7 @@ module Profiler
     module Resources
       class SlowTests
         def self.call
-          profiles = Profiler.storage.list(limit: 500)
+          profiles = Profiler.storage.list(limit: 500, type: "test")
           tests = profiles.select { |p| p.profile_type == "test" }
                           .sort_by { |p| -p.duration }
                           .first(10)

data/lib/profiler/mcp/server.rb

@@ -35,8 +35,12 @@ module Profiler
           handler = @resource_handlers[uri]
           next [{ uri: uri, mimeType: "application/json", text: "Resource not found: #{uri}" }] unless handler
 
-          result = handler.call
-          [{ uri: result[:uri], mimeType: result[:mimeType], text: result[:text] }]
+          begin
+            result = handler.call
+            [{ uri: result[:uri], mimeType: result[:mimeType], text: result[:text] }]
+          rescue Profiler::Storage::Unavailable::Error => e
+            [{ uri: uri, mimeType: "text/plain", text: e.message }]
+          end
         end
       end
 
@@ -393,6 +397,8 @@ module Profiler
         ::MCP::Tool.define(name: name, description: description, input_schema: input_schema) do |server_context: nil, **args|
           result = handler.call(args.transform_keys(&:to_s))
           result.is_a?(::MCP::Tool::Response) ? result : ::MCP::Tool::Response.new(result)
+        rescue Profiler::Storage::Unavailable::Error => e
+          ::MCP::Tool::Response.new([{ type: "text", text: e.message }], error: true)
         end
       end
 

data/lib/profiler/mcp/tools/get_profile_ajax.rb

@@ -22,6 +22,8 @@ module Profiler
             return [{ type: "text", text: "Profile not found: #{token}" }]
           end
 
+          # The sub-requests are saved after the page: computed from its children, as the page does.
+          Profiler::AjaxData.attach(profile, storage.find_by_parent(profile.token))
           ajax_data = profile.collector_data("ajax")
           unless ajax_data && ajax_data["total_requests"].to_i > 0
             return [{ type: "text", text: "No AJAX requests found in this profile" }]

data/lib/profiler/mcp/tools/get_test_profile_detail.rb

@@ -14,7 +14,7 @@ module Profiler
 
           storage = MCP::SlaveSupport.resolve_storage(params)
           profile = if token == "latest"
-            profiles = storage.list(limit: 200)
+            profiles = storage.list(limit: 1, type: "test")
             profiles.find { |p| p.profile_type == "test" }
           else
             storage.load(token)

data/lib/profiler/mcp/tools/query_console_profiles.rb

@@ -12,7 +12,7 @@ module Profiler
           limit = params["limit"]&.to_i || 20
           fetch_size = [limit * 5, 500].min
           storage = MCP::SlaveSupport.resolve_storage(params)
-          profiles = storage.list(limit: fetch_size)
+          profiles = storage.list(limit: fetch_size, type: "console")
 
           consoles = profiles.select { |p| p.profile_type == "console" }
 

data/lib/profiler/mcp/tools/query_jobs.rb

@@ -12,7 +12,7 @@ module Profiler
           limit = params["limit"]&.to_i || 20
           fetch_size = [limit * 5, 500].min
           storage = MCP::SlaveSupport.resolve_storage(params)
-          profiles = storage.list(limit: fetch_size)
+          profiles = storage.list(limit: fetch_size, type: "job")
 
           jobs = profiles.select { |p| p.profile_type == "job" }
 

data/lib/profiler/mcp/tools/query_profiles.rb

@@ -12,7 +12,7 @@ module Profiler
           storage = MCP::SlaveSupport.resolve_storage(params)
           limit = params["limit"]&.to_i || 20
           fetch_size = [limit * 5, 500].min
-          profiles = storage.list(limit: fetch_size)
+          profiles = storage.list(limit: fetch_size, type: params["profile_type"])
 
           profiles = profiles.select { |p| p.path&.include?(params["path"]) } if params["path"]
           profiles = profiles.select { |p| p.method == params["method"]&.upcase } if params["method"]

data/lib/profiler/mcp/tools/query_test_profiles.rb

@@ -12,7 +12,7 @@ module Profiler
           limit = params["limit"]&.to_i || 20
           fetch_size = [limit * 5, 500].min
           storage = MCP::SlaveSupport.resolve_storage(params)
-          profiles = storage.list(limit: fetch_size)
+          profiles = storage.list(limit: fetch_size, type: "test")
 
           tests = profiles.select { |p| p.profile_type == "test" }
 

data/lib/profiler/mcp/tools/run_tests.rb

@@ -114,7 +114,7 @@ module Profiler
         end
 
         def self.collect_run_profiles(since)
-          Profiler.storage.list(limit: 500).select do |p|
+          Profiler.storage.list(limit: 500, type: "test").select do |p|
             p.profile_type == "test" && p.started_at && p.started_at >= since
           end.map(&:token)
         rescue

data/lib/profiler/middleware/profiler_middleware.rb

@@ -104,7 +104,10 @@ module Profiler
         record_response(env, profile, request_body, status, headers, content, content.bytesize, true)
         collect_all(profile, collectors)
 
-        Profiler.storage.save(profile.token, profile)
+        storage = Profiler.storage
+        storage.save(profile.token, profile)
+        # Not saved: no token to send, no toolbar to inject for it.
+        return [status, headers, body] if storage.is_a?(Profiler::Storage::Unavailable)
 
         set_header(headers, TOKEN_HEADER, profile.token)
 

data/lib/profiler/railtie.rb

@@ -15,6 +15,9 @@ module Profiler
         config.default(:enabled, development_or_test)
         config.default(:storage, development_or_test ? :file : :memory)
         config.default(:track_tests, Rails.env.test?)
+        # One test profile per example: no cap on the count in the test environment, so that the
+        # report at the end of a suite sees all of them.
+        config.default(:max_profiles, Rails.env.test? ? nil : 100)
 
         app.config.profiler.each do |key, value|
           if config.respond_to?("#{key}=")