Improve ActiveRecord connection in on_worker_boot (mastodon#9238)

This is how it looks in the example in the Puma README
dwango · Nov 23, 2018 · f73b7e7 · f73b7e7
1 parent 63f168c
commit f73b7e7
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/config/puma.rb b/config/puma.rb
@@ -13,7 +13,9 @@
 preload_app!
 
 on_worker_boot do
-  ActiveRecord::Base.establish_connection if defined?(ActiveRecord)
+  ActiveSupport.on_load(:active_record) do
+    ActiveRecord::Base.establish_connection
+  end
 end
 
 plugin :tmp_restart