before_fork and after_fork hooks are now arrays (no user API change)

Author: defunkt

README.markdown

@@ -687,6 +687,9 @@ All hooks can also be set using a setter, e.g.
 
     Resque.after_fork = proc { puts "called" }
 
+To clear all hooks, set nil or false:
+
+    Resque.after_fork = nil
 
 Namespaces
 ----------

lib/resque.rb

@@ -56,13 +56,14 @@ def redis
   # Call with a block to set the hook.
   # Call with no arguments to return the hook.
   def before_first_fork(&block)
-    block ? (@before_first_fork = block) : @before_first_fork
+    @before_first_fork ||= []
+    block ? (@before_first_fork << block) : @before_first_fork
   end
 
   # Set a proc that will be called in the parent process before the
   # worker forks for the first time.
-  def before_first_fork=(before_first_fork)
-    @before_first_fork = before_first_fork
+  def before_first_fork=(block)
+    block ? (before_first_fork << block) : before_first_fork.clear
   end
 
   # The `before_fork` hook will be run in the **parent** process
@@ -72,12 +73,13 @@ def before_first_fork=(before_first_fork)
   # Call with a block to set the hook.
   # Call with no arguments to return the hook.
   def before_fork(&block)
-    block ? (@before_fork = block) : @before_fork
+    @before_fork ||= []
+    block ? (@before_fork << block) : @before_fork
   end
 
   # Set the before_fork proc.
-  def before_fork=(before_fork)
-    @before_fork = before_fork
+  def before_fork=(block)
+    block ? (before_fork << block) : before_fork.clear
   end
 
   # The `after_fork` hook will be run in the child process and is passed
@@ -87,12 +89,13 @@ def before_fork=(before_fork)
   # Call with a block to set the hook.
   # Call with no arguments to return the hook.
   def after_fork(&block)
-    block ? (@after_fork = block) : @after_fork
+    @after_fork ||= []
+    block ? (@after_fork << block) : @after_fork
   end
 
   # Set the after_fork proc.
-  def after_fork=(after_fork)
-    @after_fork = after_fork
+  def after_fork=(block)
+    block ? (after_fork << block) : after_fork.clear
   end
 
   def to_s

lib/resque/worker.rb

@@ -312,12 +312,13 @@ def register_worker
 
     # Runs a named hook, passing along any arguments.
     def run_hook(name, *args)
-      return unless hook = Resque.send(name)
-      msg = "Running #{name} hook"
-      msg << " with #{args.inspect}" if args.any?
-      log msg
+      Resque.send(name).each do |hook|
+        msg = "Running #{name} hook"
+        msg << " with #{args.inspect}" if args.any?
+        log msg
 
-      args.any? ? hook.call(*args) : hook.call
+        args.any? ? hook.call(*args) : hook.call
+      end
     end
 
     # Unregisters ourself as a worker. Useful when shutting down.