fix save node

cosmos · Nov 29, 2022 · 03baaf5 · 03baaf5
1 parent dcc0d07
commit 03baaf5
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 9 deletions.
diff --git a/mutable_tree.go b/mutable_tree.go
@@ -1206,18 +1206,24 @@ func (tree *MutableTree) saveNewNodes() error {
 
 	var recursiveSave func(*Node) error
 	recursiveSave = func(node *Node) error {
+		if node.nodeKey.version < version {
+			return nil
+		}
 		if err := tree.ndb.SaveNode(node); err != nil {
 			return err
 		}
-		if err := recursiveSave(node.leftNode); err != nil {
-			return err
+		if node.leftNode != nil {
+			if err := recursiveSave(node.leftNode); err != nil {
+				return err
+			}
+			node.leftNode = nil
 		}
-		if err := recursiveSave(node.rightNode); err != nil {
-			return err
+		if node.rightNode != nil {
+			if err := recursiveSave(node.rightNode); err != nil {
+				return err
+			}
+			node.rightNode = nil
 		}
-
-		node.leftNode = nil
-		node.rightNode = nil
 		return nil
 	}
 

diff --git a/nodedb.go b/nodedb.go
@@ -208,8 +208,6 @@ func (ndb *nodeDB) SaveNode(node *Node) error {
 			return err
 		}
 	}
-	node.leftNode = nil
-	node.rightNode = nil
 
 	logger.Debug("BATCH SAVE %+v\n", node)
 	ndb.nodeCache.Add(node)