rails-profiler 0.29.0 → 0.30.0

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

@@ -1,10 +1,17 @@
 # frozen_string_literal: true
 
+require_relative "../slave_support"
+
 module Profiler
   module MCP
     module Tools
       class ResetAllEnvVars
-        def self.call(_params)
+        def self.call(params)
+          if (proxy = MCP::SlaveSupport.with_slave_proxy(params))
+            proxy.delete_json("/_profiler/api/env_vars/reset_all")
+            return [{ type: "text", text: "Reset all ENV overrides on slave '#{params["slave"]}'." }]
+          end
+
           overrides = Profiler.env_override_store.all_overrides
           count = overrides.size
 

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

@@ -1,5 +1,9 @@
 # frozen_string_literal: true
 
+require_relative "../slave_support"
+
+require "uri"
+
 module Profiler
   module MCP
     module Tools
@@ -9,6 +13,11 @@ module Profiler
 
           return [{ type: "text", text: "Error: key cannot be blank." }] if key.empty?
 
+          if (proxy = MCP::SlaveSupport.with_slave_proxy(params))
+            proxy.delete_json("/_profiler/api/env_vars/reset?key=#{URI.encode_www_form_component(key)}")
+            return [{ type: "text", text: "Reset #{key} on slave '#{params["slave"]}'." }]
+          end
+
           overrides = Profiler.env_override_store.all_overrides
           unless overrides.key?(key)
             return [{ type: "text", text: "No active override for #{key}." }]

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

@@ -1,5 +1,7 @@
 # frozen_string_literal: true
 
+require_relative "../slave_support"
+
 require_relative "../../test_runner/discovery"
 require_relative "../../test_runner/run_store"
 require_relative "../../test_runner/runner"
@@ -13,6 +15,10 @@ module Profiler
         POLL_TIMEOUT       = 10 # seconds per wait_for_output call
 
         def self.call(params)
+          if (proxy = MCP::SlaveSupport.with_slave_proxy(params))
+            return run_on_slave(proxy, params)
+          end
+
           files          = Array(params["files"])
           framework      = params["framework"]&.to_s
           timeout_secs   = (params["timeout_seconds"] || DEFAULT_TIMEOUT).to_i
@@ -65,6 +71,41 @@ module Profiler
 
         private
 
+        def self.run_on_slave(proxy, params)
+          files        = Array(params["files"])
+          framework    = params["framework"].to_s
+          timeout_secs = (params["timeout_seconds"] || DEFAULT_TIMEOUT).to_i
+          max_output   = (params["max_output"] || DEFAULT_MAX_OUTPUT).to_i
+
+          run_data = proxy.post_json("/_profiler/api/test_runner/runs", { files: files, framework: framework.presence || "rspec" })
+
+          if run_data["error"]
+            return [{ type: "text", text: "Error starting tests on slave: #{run_data["error"]}" }]
+          end
+
+          run_id  = run_data["id"]
+          deadline = Time.now + timeout_secs
+          timed_out = false
+
+          loop do
+            break if Time.now > deadline && (timed_out = true)
+            run_data = proxy.get_json("/_profiler/api/test_runner/runs/#{run_id}")
+            break if %w[completed failed cancelled].include?(run_data["status"])
+            sleep 2
+          end
+
+          output = run_data["output_lines"]&.join.to_s
+          output = "…(truncated)\n" + output[-(max_output)..] if output.length > max_output
+
+          text = "# Test Run on slave '#{params["slave"]}' #{timed_out ? "(timed out)" : ""}\n\n"
+          text += "| Field | Value |\n|-------|-------|\n"
+          text += "| Run ID | `#{run_id}` |\n"
+          text += "| Status | **#{run_data["status"]}** |\n"
+          text += "| Exit code | #{run_data["exit_code"].inspect} |\n\n"
+          text += "## Output\n```\n#{output.strip}\n```"
+          [{ type: "text", text: text }]
+        end
+
         def self.collect_run_profiles(since)
           Profiler.storage.list(limit: 500).select do |p|
             p.profile_type == "test" && p.started_at && p.started_at >= since

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

@@ -1,5 +1,7 @@
 # frozen_string_literal: true
 
+require_relative "../slave_support"
+
 module Profiler
   module MCP
     module Tools
@@ -10,6 +12,11 @@ module Profiler
 
           return [{ type: "text", text: "Error: key cannot be blank." }] if key.empty?
 
+          if (proxy = MCP::SlaveSupport.with_slave_proxy(params))
+            proxy.patch_json("/_profiler/api/env_vars", { key: key, value: value })
+            return [{ type: "text", text: "Set #{key}=#{value} on slave '#{params["slave"]}'" }]
+          end
+
           Profiler.env_override_store.set(key, value)
           ENV[key] = value
 

data/lib/profiler/railtie.rb

@@ -93,6 +93,17 @@ module Profiler
       end
     end
 
+    initializer "profiler.start_cluster_client" do
+      # Check slave? inside on_load — the app's own initializers (config/initializers/profiler.rb)
+      # set master_url AFTER railtie initializers run, so the check must happen after_initialize.
+      ActiveSupport.on_load(:after_initialize) do
+        next unless Profiler.configuration.enabled && Profiler.configuration.slave?
+
+        require_relative "cluster/master_client"
+        Profiler::Cluster::MasterClient.new.start
+      end
+    end
+
     console do
       next unless Profiler.configuration.enabled && Profiler.configuration.track_console
 

data/lib/profiler/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Profiler
-  VERSION = "0.29.0"
+  VERSION = "0.30.0"
 end

data/lib/profiler.rb

@@ -29,6 +29,13 @@ module Profiler
       @env_override_store ||= EnvOverrideStore.new
     end
 
+    def slave_registry
+      @slave_registry ||= begin
+        require_relative "profiler/cluster/slave_registry"
+        Cluster::SlaveRegistry.new
+      end
+    end
+
     def enabled?
       configuration.enabled
     end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: rails-profiler
 version: !ruby/object:Gem::Version
-  version: 0.29.0
+  version: 0.30.0
 platform: ruby
 authors:
 - Sébastien Duplessy
@@ -107,6 +107,7 @@ files:
 - app/assets/builds/profiler.css
 - app/assets/builds/profiler.js
 - app/controllers/profiler/api/ajax_controller.rb
+- app/controllers/profiler/api/cluster_controller.rb
 - app/controllers/profiler/api/console_controller.rb
 - app/controllers/profiler/api/env_vars_controller.rb
 - app/controllers/profiler/api/events_controller.rb
@@ -115,6 +116,7 @@ files:
 - app/controllers/profiler/api/jobs_controller.rb
 - app/controllers/profiler/api/outbound_http_controller.rb
 - app/controllers/profiler/api/profiles_controller.rb
+- app/controllers/profiler/api/slave_proxy_controller.rb
 - app/controllers/profiler/api/test_runner_controller.rb
 - app/controllers/profiler/api/tests_controller.rb
 - app/controllers/profiler/api/toolbar_controller.rb
@@ -130,6 +132,9 @@ files:
 - config/routes.rb
 - exe/profiler-mcp
 - lib/profiler.rb
+- lib/profiler/cluster/master_client.rb
+- lib/profiler/cluster/slave_proxy.rb
+- lib/profiler/cluster/slave_registry.rb
 - lib/profiler/collectors/ajax_collector.rb
 - lib/profiler/collectors/base_collector.rb
 - lib/profiler/collectors/cache_collector.rb
@@ -172,6 +177,7 @@ files:
 - lib/profiler/mcp/resources/slow_queries.rb
 - lib/profiler/mcp/resources/slow_tests.rb
 - lib/profiler/mcp/server.rb
+- lib/profiler/mcp/slave_support.rb
 - lib/profiler/mcp/tools/analyze_queries.rb
 - lib/profiler/mcp/tools/clear_profiles.rb
 - lib/profiler/mcp/tools/delete_env_var.rb
@@ -183,6 +189,7 @@ files:
 - lib/profiler/mcp/tools/get_profile_mailers.rb
 - lib/profiler/mcp/tools/get_test_profile_detail.rb
 - lib/profiler/mcp/tools/list_env_vars.rb
+- lib/profiler/mcp/tools/list_slaves.rb
 - lib/profiler/mcp/tools/query_console_profiles.rb
 - lib/profiler/mcp/tools/query_jobs.rb
 - lib/profiler/mcp/tools/query_mailers.rb