lib/openwfe/expressions/timeout.rb in openwferu-0.9.4 vs lib/openwfe/expressions/timeout.rb in openwferu-0.9.5

@@ -71,13 +71,10 @@
 
             timeout = OpenWFE::parse_time_string(timeout)
             @timeout_at = Time.new.to_f + timeout
         end
 
-        #
-        # Overrides the parent method to make sure a potential
-        #
         def reschedule (scheduler)
             to_reschedule(scheduler)
         end
 
         #
@@ -85,9 +82,17 @@
         # timeout schedules gets removed.
         #
         def reply_to_parent (workitem)
             unschedule_timeout()
             super(workitem)
+        end
+
+        def set_timedout_flag (workitem)
+            workitem.attributes["__timed_out__"] = "true"
+        end
+
+        def remove_timedout_flag (workitem)
+            workitem.attributes.delete("__timed_out__")
         end
 
         protected
 
             #