Add helper reset internals function.

Use a little helper function to reset
the internal state of the flow when needed
instead of duplicating code that does this.

Change-Id: I51d83538a2920c7d387ffd1756e8d99413f4077e

1 files changed, 8 insertions, 6 deletions

diff --git a/taskflow/patterns/graph_flow.py b/taskflow/patterns/graph_flow.py
index 2883c48..824545a 100644
--- a/taskflow/patterns/graph_flow.py
+++ b/taskflow/patterns/graph_flow.py
@@ -49,8 +49,7 @@ class Flow(linear_flow.Flow):
         assert isinstance(task, collections.Callable)
         r = utils.Runner(task)
         self._graph.add_node(r, uuid=r.uuid)
-        self._runners = []
-        self._leftoff_at = None
+        self._reset_internals()
         return r.uuid
 
     def _add_dependency(self, provider, requirer):
@@ -64,6 +63,10 @@ class Flow(linear_flow.Flow):
         lines.append("%s" % (self.state))
         return "; ".join(lines)
 
+    def _reset_internals(self):
+        super(Flow, self)._reset_internals()
+        self._runners = []
+
     @decorators.locked
     def remove(self, uuid):
         runner = None
@@ -74,10 +77,8 @@ class Flow(linear_flow.Flow):
         if not runner:
             raise ValueError("No runner found with uuid %s" % (uuid))
         else:
-            # Ensure that we reset out internal state after said removal
             self._graph.remove_node(runner)
-            self._runners = []
-            self._leftoff_at = None
+            self._reset_internals()
 
     def _ordering(self):
         try:
@@ -94,7 +95,7 @@ class Flow(linear_flow.Flow):
         create said dependency."""
         if len(self._graph) == 0:
             return []
-        if self._runners:
+        if self._connected:
             return self._runners
 
         # Clear out all edges (since we want to do a fresh connection)
@@ -137,4 +138,5 @@ class Flow(linear_flow.Flow):
             r.runs_before = list(reversed(run_stack))
             run_stack.append(r)
         self._runners = run_order
+        self._connected = True
         return run_order