Fix SaveNode() to delete node from recentDB

amolabs · Jun 4, 2020 · 2e3d3bd · 2e3d3bd
1 parent 76a52fa
commit 2e3d3bd
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/nodedb.go b/nodedb.go
@@ -226,14 +226,14 @@ func (ndb *nodeDB) SaveNode(node *Node, flushToDisk bool) {
 		panic(err)
 	}
 
-	if !node.saved {
-		node.saved = true
-		ndb.recentBatch.Set(ndb.nodeKey(node.hash), buf.Bytes())
-	}
 	if flushToDisk {
 		ndb.snapshotBatch.Set(ndb.nodeKey(node.hash), buf.Bytes())
+		ndb.recentBatch.Delete(ndb.nodeKey(node.hash))
 		node.persisted = true
 		node.saved = true
+	} else if !node.saved {
+		node.saved = true
+		ndb.recentBatch.Set(ndb.nodeKey(node.hash), buf.Bytes())
 	}
 }