done_working is called inside of the work loop, which removes the wor…

…ker from redis. this causes workers to disappear after the first fork
basecamp · Feb 1, 2011 · 93b4f6c · 93b4f6c
1 parent 7982688
commit 93b4f6c
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/lib/resque-multi-job-forks.rb b/lib/resque-multi-job-forks.rb
@@ -22,6 +22,13 @@ def shutdown_with_multi_job_forks
       end
       alias_method :shutdown_without_multi_job_forks, :shutdown?
       alias_method :shutdown?, :shutdown_with_multi_job_forks
+
+      def working_on_with_worker_registration(job)
+        register_worker
+        working_on_without_worker_registration(job)
+      end
+      alias_method :working_on_without_worker_registration, :working_on
+      alias_method :working_on, :working_on_with_worker_registration
     end
 
     def fork_hijacked?