Rebase instead of replace

0chain · Feb 7, 2022 · 2353433 · 2353433
1 parent 22f5709
commit 2353433
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/code/go/0chain.net/core/util/merkle_patricia_trie.go b/code/go/0chain.net/core/util/merkle_patricia_trie.go
@@ -47,7 +47,10 @@ func CloneMPT(mpt MerklePatriciaTrieI) *MerklePatriciaTrie {
 func (mpt *MerklePatriciaTrie) SetNodeDB(ndb NodeDB) {
 	mpt.mutex.Lock()
 	defer mpt.mutex.Unlock()
-	mpt.db = ndb
+	if lndb, ok := mpt.db.(*LevelNodeDB); ok {
+		lndb.RebaseCurrentDB(ndb)
+	}
+	//mpt.db = ndb
 }
 
 /*GetNodeDB - implement interface */