judoscale-rails 1.10.0 → 1.11.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 8126459482ea24a61ab251dc19235c8c52f3087d9104ff68d1442e4fb73d4698
-  data.tar.gz: 3a25a9b8ced78ffdb8d21351085ccadfb533ecddda22b735b37c8d278977e88d
+  metadata.gz: 5af8c7881148adca3b0b924b30918cc0a99962c71d02c54b4ea66a19313c3845
+  data.tar.gz: 061baca32fd8dc5f05ff67524e7147d9cb803c9a3966935ff764141215a2d382
 SHA512:
-  metadata.gz: b23c16010590b9270a1b1369e95b64eac963993391dfeb415a7e37544d2c06e199d562014b4aa9226087a5ae34c38ae60ba43d59e9455efa60789e71bcc4046c
-  data.tar.gz: a3af90edce5f9b7124dea6d4e599f2c1543f6655fe50cfe341ece06e15e9dd4738671f0886270fd2dd5f791165924018a3cd86da89c789ae4fab3cd861c81a8a
+  metadata.gz: 58163866506c2ef25a3306a344f96ffa41ce83c8d3802e249b4914601e2787eb160e3509507c10bd73da6a86fbde08212547875f7239526bb3fcc95aa0da7da8
+  data.tar.gz: 3de499c651b8744a05911834bcc40a85340ea2c4b6cbe7f480c6afd126bf5cc21996563b2e85c57109015c5f1e4a4be57fa4d2860f633891b72c84067c7b5f36

data/lib/judoscale/rails/config.rb

@@ -3,12 +3,15 @@ require "judoscale/config"
 module Judoscale
   module Rails
     module Config
-      attr_accessor :start_reporter_after_initialize, :rake_task_ignore_regex
+      attr_accessor :start_reporter_after_initialize, :rake_task_ignore_regex, :utilization_enabled, :utilization_interval
 
       def reset
         super
         @start_reporter_after_initialize = true
         @rake_task_ignore_regex = /assets:|db:/
+
+        @utilization_enabled = ENV["JUDOSCALE_UTILIZATION_ENABLED"] == "true"
+        @utilization_interval = (ENV["JUDOSCALE_UTILIZATION_INTERVAL"] || 1.0).to_f
       end
     end
 

data/lib/judoscale/rails/railtie.rb

@@ -4,6 +4,7 @@ require "rails"
 require "rails/railtie"
 require "judoscale/request_middleware"
 require "judoscale/rails/config"
+require "judoscale/rails/utilization_middleware"
 require "judoscale/logger"
 require "judoscale/reporter"
 
@@ -27,14 +28,20 @@ module Judoscale
         ::Judoscale::Config.instance
       end
 
-      initializer "Judoscale.logger" do |app|
+      initializer "judoscale.logger" do |app|
         judoscale_config.logger = ::Rails.logger
       end
 
-      initializer "Judoscale.request_middleware" do |app|
+      initializer "judoscale.request_middleware" do |app|
         app.middleware.insert_before Rack::Runtime, RequestMiddleware
       end
 
+      initializer "judoscale.utilization_middleware", after: :load_config_initializers do |app|
+        if judoscale_config.utilization_enabled
+          app.middleware.insert_before RequestMiddleware, UtilizationMiddleware, interval: judoscale_config.utilization_interval
+        end
+      end
+
       config.after_initialize do
         if in_rails_console_or_runner?
           logger.debug "No reporting since we're in a Rails console or runner process"

data/lib/judoscale/rails/utilization_middleware.rb

@@ -0,0 +1,70 @@
+# frozen_string_literal: true
+
+require "singleton"
+require "concurrent"
+require "judoscale/metrics_store"
+
+module Judoscale
+  module Rails
+    class UtilizationMiddleware
+      def initialize(app, interval:)
+        @app = app
+        @interval = interval
+      end
+
+      def call(env)
+        tracker = UtilizationTracker.instance
+        tracker.start!(interval: @interval)
+        tracker.incr
+
+        @app.call(env)
+      ensure
+        tracker.decr
+      end
+    end
+
+    class UtilizationTracker
+      include Singleton
+
+      def initialize
+        @active_request_counter = Concurrent::AtomicFixnum.new(0)
+        @thread_ref = Concurrent::AtomicReference.new(nil)
+      end
+
+      def start!(interval:)
+        @thread_ref.update do |current_thread|
+          next current_thread if current_thread&.alive?
+
+          Thread.new do
+            # Advise multi-threaded app servers to ignore this thread for the purposes of fork safety warnings.
+            Thread.current.thread_variable_set(:fork_safe, true)
+
+            loop do
+              sleep interval
+              track_current_state
+            end
+          end
+        end
+      end
+
+      def incr
+        @active_request_counter.increment
+      end
+
+      def decr
+        @active_request_counter.decrement
+      end
+
+      def track_current_state
+        active_requests = @active_request_counter.value
+        active_processes = (active_requests > 0) ? 1 : 0
+        time = Time.now.utc
+
+        MetricsStore.instance.tap do |store|
+          store.push :pu, active_processes, time # pu = process utilization
+          store.push :ru, active_requests, time  # ru = request utilization
+        end
+      end
+    end
+  end
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: judoscale-rails
 version: !ruby/object:Gem::Version
-  version: 1.10.0
+  version: 1.11.0
 platform: ruby
 authors:
 - Adam McCrea
@@ -10,7 +10,7 @@ authors:
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2025-02-17 00:00:00.000000000 Z
+date: 2025-04-28 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: judoscale-ruby
@@ -18,14 +18,14 @@ dependencies:
     requirements:
     - - '='
       - !ruby/object:Gem::Version
-        version: 1.10.0
+        version: 1.11.0
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - '='
       - !ruby/object:Gem::Version
-        version: 1.10.0
+        version: 1.11.0
 - !ruby/object:Gem::Dependency
   name: railties
   requirement: !ruby/object:Gem::Requirement
@@ -51,6 +51,7 @@ files:
 - lib/judoscale/rails.rb
 - lib/judoscale/rails/config.rb
 - lib/judoscale/rails/railtie.rb
+- lib/judoscale/rails/utilization_middleware.rb
 homepage: https://judoscale.com
 licenses:
 - MIT