Add on_booted event

README.md

@@ -157,6 +157,15 @@ before_fork do
 end
 ```
 
+You can also specify a block to be run after puma is booted using `on_booted`:
+
+```ruby
+# config/puma.rb
+on_booted do
+  # configuration here
+end
+```
+
 ### Error handling
 
 If puma encounters an error outside of the context of your application, it will respond with a 500 and a simple

lib/puma/cli.rb

@@ -93,7 +93,7 @@ def configure_control_url(command_line_arg)
     #
 
     def setup_options
-      @conf = Configuration.new do |user_config, file_config|
+      @conf = Configuration.new({}, {events: @events}) do |user_config, file_config|
         @parser = OptionParser.new do |o|
           o.on "-b", "--bind URI", "URI to bind to (tcp://, unix://, ssl://)" do |arg|
             user_config.bind arg

lib/puma/dsl.rb

@@ -663,6 +663,16 @@ def after_worker_fork(&block)
 
     alias_method :after_worker_boot, :after_worker_fork
 
+    # Code to run after puma is booted (works for both: single and clustered)
+    #
+    # @example
+    #   on_booted do
+    #     puts 'After booting...'
+    #   end
+    def on_booted(&block)
+      @config.options[:events].on_booted(&block)
+    end
+
     # When `fork_worker` is enabled, code to run in Worker 0
     # before all other workers are re-forked from this process,
     # after the server has temporarily stopped serving requests

test/config/event_on_booted.rb

@@ -0,0 +1,3 @@
+on_booted do
+  puts "on_booted called"
+end

test/config/event_on_booted_exit.rb

@@ -0,0 +1,5 @@
+on_booted do
+  pid = Process.pid
+  Process.kill :TERM, pid
+  Process.wait pid
+end

test/test_integration_cluster.rb

@@ -130,6 +130,16 @@ def test_term_suppress
     assert_equal 0, status
   end
 
+  def test_on_booted
+    cli_server "-w #{workers} -C test/config/event_on_booted.rb -C test/config/event_on_booted_exit.rb test/rackup/hello.ru", no_wait: true
+
+    output = []
+
+    output << $_ while @server.gets
+
+    assert output.any? { |msg| msg == "on_booted called\n" } != nil
+  end
+
   def test_term_worker_clean_exit
     cli_server "-w #{workers} test/rackup/hello.ru"
 

test/test_integration_single.rb

@@ -49,6 +49,16 @@ def test_term_exit_code
     assert_equal 15, status
   end
 
+  def test_on_booted
+    cli_server "-C test/config/event_on_booted.rb -C test/config/event_on_booted_exit.rb test/rackup/hello.ru", no_wait: true
+
+    output = []
+
+    output << $_ while @server.gets
+
+    assert output.any? { |msg| msg == "on_booted called\n" } != nil
+  end
+
   def test_term_suppress
     skip_unless_signal_exist? :TERM