lib/gorgon/callback_handler.rb in gorgon-0.7.1 vs lib/gorgon/callback_handler.rb in gorgon-0.8.0

@@ -1,25 +1,35 @@
 class CallbackHandler
   def initialize(config)
     @config = config || {}
+    load(@config[:callbacks_class_file]) if @config[:callbacks_class_file]
   end
 
-  def before_start
-    load(@config[:before_start]) if @config[:before_start]
+  def before_originate
+    cluster_id = Gorgon.callbacks.before_originate
+    return cluster_id if cluster_id.is_a?(String)
   end
 
-  def after_complete
-    load(@config[:after_complete]) if @config[:after_complete]
+  def after_sync
+    Gorgon.callbacks.after_sync
   end
 
   def before_creating_workers
-    load(@config[:before_creating_workers]) if @config[:before_creating_workers]
+    Gorgon.callbacks.before_creating_workers
   end
 
-  def after_sync
-    load(@config[:after_sync]) if @config[:after_sync]
+  def before_start
+    Gorgon.callbacks.before_start
   end
 
   def after_creating_workers
-    load(@config[:after_creating_workers]) if @config[:after_creating_workers]
+    Gorgon.callbacks.after_creating_workers
+  end
+
+  def after_complete
+    Gorgon.callbacks.after_complete
+  end
+
+  def after_job_finishes
+    Gorgon.callbacks.after_job_finishes
   end
 end