From 2904387287f81eae8b8e1abf8f36d4ec5695eda6 Mon Sep 17 00:00:00 2001
From: Alexei Mikhailov <amikhailov83@gmail.com>
Date: Sat, 27 Apr 2013 13:16:04 +0600
Subject: [PATCH] Backport hooks behaviour from 1.24.1

---
 lib/resque/integration.rb               |  4 ++
 lib/resque/integration/hooks.rb         | 72 +++++++++++++++++++++++++
 lib/resque/integration/tasks/hooks.rake |  4 +-
 3 files changed, 78 insertions(+), 2 deletions(-)
 create mode 100644 lib/resque/integration/hooks.rb

diff --git a/lib/resque/integration.rb b/lib/resque/integration.rb
index 63fcf9a..c7c5f3f 100644
--- a/lib/resque/integration.rb
+++ b/lib/resque/integration.rb
@@ -12,11 +12,15 @@
 
 require 'active_support/concern'
 
+require 'resque/integration/hooks'
 require 'resque/integration/engine'
 
 require 'active_support/core_ext/module/attribute_accessors'
 
 module Resque
+  include Integration::Hooks
+  extend Integration::Hooks
+
   # Resque.config is available now
   mattr_accessor :config
 
diff --git a/lib/resque/integration/hooks.rb b/lib/resque/integration/hooks.rb
new file mode 100644
index 0000000..5752542
--- /dev/null
+++ b/lib/resque/integration/hooks.rb
@@ -0,0 +1,72 @@
+# coding: utf-8
+
+module Resque
+  module Integration
+    # Backport of resque-1.24.x hooks
+    # @see https://github.com/resque/resque/pull/680/files
+    module Hooks
+      # We do not want to patch Worker, so we declare callable Array
+      class CallableArray < Array
+        def call(*args)
+          each { |hook| hook.call(*args) }
+        end
+      end
+      # Call with a block to register a hook.
+      # Call with no arguments to return all registered hooks.
+      def before_first_fork(&block)
+        block ? register_hook(:before_first_fork, block) : hooks(:before_first_fork)
+      end
+
+      # Register a before_first_fork proc.
+      def before_first_fork=(block)
+        register_hook(:before_first_fork, block)
+      end
+
+      # Call with a block to register a hook.
+      # Call with no arguments to return all registered hooks.
+      def before_fork(&block)
+        block ? register_hook(:before_fork, block) : hooks(:before_fork)
+      end
+
+      # Register a before_fork proc.
+      def before_fork=(block)
+        register_hook(:before_fork, block)
+      end
+
+      # Call with a block to register a hook.
+      # Call with no arguments to return all registered hooks.
+      def after_fork(&block)
+        block ? register_hook(:after_fork, block) : hooks(:after_fork)
+      end
+
+      # Register an after_fork proc.
+      def after_fork=(block)
+        register_hook(:after_fork, block)
+      end
+
+      private
+
+      # Register a new proc as a hook. If the block is nil this is the
+      # equivalent of removing all hooks of the given name.
+      #
+      # `name` is the hook that the block should be registered with.
+      def register_hook(name, block)
+        return clear_hooks(name) if block.nil?
+
+        @hooks ||= {}
+        @hooks[name] ||= CallableArray.new
+        @hooks[name] << block
+      end
+
+      # Clear all hooks given a hook name.
+      def clear_hooks(name)
+        @hooks && @hooks[name] = CallableArray.new
+      end
+
+      # Retrieve all hooks of a given name.
+      def hooks(name)
+        (@hooks && @hooks[name]) || CallableArray.new
+      end
+    end # module Hooks
+  end # module Integration
+end # module Resque
\ No newline at end of file
diff --git a/lib/resque/integration/tasks/hooks.rake b/lib/resque/integration/tasks/hooks.rake
index 35f5da5..8289c36 100644
--- a/lib/resque/integration/tasks/hooks.rake
+++ b/lib/resque/integration/tasks/hooks.rake
@@ -19,10 +19,10 @@ namespace :resque do
     #   Resque.after_fork { ActiveRecord::Base.connection_handler.verify_active_connections! }
     #
     # Это работает
-    Resque.before_fork { ActiveRecord::Base.connection_handler.clear_all_connections! }
+    Resque.before_first_fork { ActiveRecord::Base.connection_handler.clear_all_connections! }
 
     # Нужно также закрыть соединение к memcache
-    Resque.before_fork { Rails.cache.reset if Rails.cache.respond_to?(:reset) }
+    Resque.before_first_fork { Rails.cache.reset if Rails.cache.respond_to?(:reset) }
 
     # Support for resque-multi-job-forks
     if ENV['JOBS_PER_FORK'] || ENV['MINUTES_PER_FORK']