an attempt at making the editor more efficient

whyrusleeping · whyrusleeping · commit aa122a50aa14 · 2015-08-12T13:44:23.000-07:00
License: MIT
Signed-off-by: Jeromy &lt;jeromyj@gmail.com&gt;
diff --git a/merkledag/utils/utils.go b/merkledag/utils/utils.go
@@ -153,3 +153,32 @@ func rmLink(ctx context.Context, ds dag.DAGService, root *dag.Node, path []strin
 
 	return root, nil
 }
+
+func (e *Editor) WriteOutputTo(ds dag.DAGService) error {
+	return copyDag(e.GetNode(), e.ds, ds)
+}
+
+func copyDag(nd *dag.Node, from, to dag.DAGService) error {
+	_, err := to.Add(nd)
+	if err != nil {
+		return err
+	}
+
+	for _, lnk := range nd.Links {
+		child, err := lnk.GetNode(context.Background(), from)
+		if err != nil {
+			if err == dag.ErrNotFound {
+				// not found means we didnt modify it, and it should
+				// already be in the target datastore
+				continue
+			}
+			return err
+		}
+
+		err = copyDag(child, from, to)
+		if err != nil {
+			return err
+		}
+	}
+	return nil
+}
