puma 2.2.0 → 8.0.2

data/lib/puma/server_plugin_control.rb

@@ -0,0 +1,32 @@
+module Puma
+  # ServerPluginControl provides a control interface for server plugins to
+  # interact with and manage server settings dynamically.
+  #
+  # This class acts as a facade between plugins and the Puma server,
+  # allowing plugins to safely modify server configuration and thread pool
+  # settings without direct access to the server's internal state.
+  #
+  class ServerPluginControl
+    def initialize(server)
+      @server = server
+    end
+
+    # Returns the maximum number of threads in the thread pool.
+    def max_threads
+      @server.max_threads
+    end
+
+    # Returns the minimum number of threads in the thread pool.
+    def min_threads
+      @server.min_threads
+    end
+
+    # Updates the minimum and maximum number of threads in the thread pool.
+    #
+    # @see Puma::Server#update_thread_pool_min_max
+    #
+    def update_thread_pool_min_max(min: max_threads, max: min_threads)
+      @server.update_thread_pool_min_max(min: min, max: max)
+    end
+  end
+end

data/lib/puma/single.rb

@@ -0,0 +1,72 @@
+# frozen_string_literal: true
+
+require_relative 'runner'
+require_relative 'detect'
+require_relative 'plugin'
+
+module Puma
+  # This class is instantiated by the `Puma::Launcher` and used
+  # to boot and serve a Ruby application when no puma "workers" are needed
+  # i.e. only using "threaded" mode. For example `$ puma -t 1:5`
+  #
+  # At the core of this class is running an instance of `Puma::Server` which
+  # gets created via the `start_server` method from the `Puma::Runner` class
+  # that this inherits from.
+  class Single < Runner
+    # @!attribute [r] stats
+    def stats
+      {
+        started_at: utc_iso8601(@started_at)
+      }.merge(@server&.stats || {}).merge(super)
+    end
+
+    def restart
+      @server&.begin_restart
+    end
+
+    def stop
+      @server&.stop false
+    end
+
+    def halt
+      @server&.halt
+    end
+
+    def stop_blocked
+      log "- Gracefully stopping, waiting for requests to finish"
+      @control&.stop true
+      @server&.stop true
+    end
+
+    def run
+      output_header "single"
+
+      load_and_bind
+
+      Plugins.fire_background
+
+      @launcher.write_state
+
+      start_control
+
+      @server = start_server
+      server_thread = @server.run
+
+      log "Use Ctrl-C to stop"
+
+      warn_ruby_mn_threads
+
+      redirect_io
+
+      @events.fire_after_booted!
+
+      debug_loaded_extensions("Loaded Extensions:") if @log_writer.debug?
+
+      begin
+        server_thread.join
+      rescue Interrupt
+        # Swallow it
+      end
+    end
+  end
+end

data/lib/puma/state_file.rb

@@ -0,0 +1,69 @@
+# frozen_string_literal: true
+
+module Puma
+
+  # Puma::Launcher uses StateFile to write a yaml file for use with Puma::ControlCLI.
+  #
+  # In previous versions of Puma, YAML was used to read/write the state file.
+  # Since Puma is similar to Bundler/RubyGems in that it may load before one's app
+  # does, minimizing the dependencies that may be shared with the app is desired.
+  #
+  # At present, it only works with numeric and string values.  It is still a valid
+  # yaml file, and the CI tests parse it with Psych.
+  #
+  class StateFile
+
+    ALLOWED_FIELDS = %w!control_url control_auth_token pid running_from!
+
+    def initialize
+      @options = {}
+    end
+
+    def save(path, permission = nil)
+      contents = +"---\n"
+      @options.each do |k,v|
+        next unless ALLOWED_FIELDS.include? k
+        case v
+        when Numeric
+          contents << "#{k}: #{v}\n"
+        when String
+          next if v.strip.empty?
+          contents << (k == 'running_from' || v.to_s.include?(' ') ?
+            "#{k}: \"#{v}\"\n" : "#{k}: #{v}\n")
+        end
+      end
+
+      if permission
+        File.write path, contents, mode: 'wb:UTF-8', perm: permission
+      else
+        File.write path, contents, mode: 'wb:UTF-8'
+      end
+    end
+
+    def load(path)
+      File.read(path).lines.each do |line|
+        next if line.start_with? '#'
+        k,v = line.split ':', 2
+        next unless v && ALLOWED_FIELDS.include?(k)
+        v = v.strip
+        @options[k] =
+          case v
+          when ''              then nil
+          when /\A\d+\z/       then v.to_i
+          when /\A\d+\.\d+\z/  then v.to_f
+          else                      v.gsub(/\A"|"\z/, '')
+          end
+      end
+    end
+
+    ALLOWED_FIELDS.each do |f|
+      define_method f.to_sym do
+        @options[f]
+      end
+
+      define_method :"#{f}=" do |v|
+        @options[f] = v
+      end
+    end
+  end
+end