lib/floe/workflow/states/wait.rb in floe-0.2.3 vs lib/floe/workflow/states/wait.rb in floe-0.3.0

@@ -8,20 +8,29 @@
 
         def initialize(workflow, name, payload)
           super
 
           @next    = payload["Next"]
+          @end     = !!payload["End"]
           @seconds = payload["Seconds"].to_i
 
           @input_path  = Path.new(payload.fetch("InputPath", "$"))
           @output_path = Path.new(payload.fetch("OutputPath", "$"))
         end
 
         def run!(input)
           input = input_path.value(context, input)
           sleep(seconds)
           output = output_path.value(context, input)
-          [@next, output]
+          [@end ? nil : @next, output]
+        end
+
+        def status
+          @end ? "success" : "running"
+        end
+
+        def end?
+          @end
         end
       end
     end
   end
 end