Adding publicIp sample, only mark sub-DAGs as non-prepare - the root …

…DAG stay as preparer
begoldsm · Jun 25, 2016 · cf2fdb5 · cf2fdb5
1 parent 1d5e9c1
commit cf2fdb5
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/azure-client-runtime/src/main/java/com/microsoft/azure/DAGraph.java b/azure-client-runtime/src/main/java/com/microsoft/azure/DAGraph.java
@@ -89,8 +89,10 @@ public void prepare() {
             for (U node : graph.values()) {
                 // Prepare each node for traversal
                 node.initialize();
-                // Mark other sub-DAGs are non-preparer
-                node.setPreparer(false);
+                if (!this.isRootNode(node)) {
+                    // Mark other sub-DAGs as non-preparer
+                    node.setPreparer(false);
+                }
             }
             initializeDependentKeys();
             initializeQueue();