check task status

Author: KDr2

src/tapedtask.jl

@@ -69,11 +69,13 @@ function consume(ttask::TapedTask)
 
     val = try
         take!(ttask.produce_ch)
-    catch e
-        # return nothing to indicate the finish of the task
-        isa(e, InvalidStateException) ?  nothing : rethrow()
+    catch
+        isa(e, InvalidStateException) || rethrow()
+        istaskfailed(ttask.task) && throw(ttask.task.exception)
+        # TODO: we return nothing to indicate the end of a task,
+        #       remove this when AdvancedPS is udpated.
+        istaskdone(ttask.task) && return nothing
     end
-    yield()
     isa(val, TapedTaskException) && throw(val.exc)
     return val
 end