lib/jflow/activity/worker.rb in jflow-0.5.0 vs lib/jflow/activity/worker.rb in jflow-0.5.1

@@ -43,10 +43,11 @@
       def process(task)
         begin
           Thread.current.set_state(:working)
           task.run!
         rescue => exception
+          Thread.current.set_state(:polling)
           task.handle_exception(exception)
           task.failed!(exception)
         end
       end
 
@@ -74,6 +75,6 @@
         !Thread.current.marked_for_shutdown?
       end
 
     end
   end
-end
\ No newline at end of file
+end