[R4R] Add set version in mutable tree (#8)

* add set version * set latest version
bnb-chain · Jun 18, 2019 · ad09593 · ad09593
1 parent 0ac2dfe
commit ad09593
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/mutable_tree.go b/mutable_tree.go
@@ -224,6 +224,12 @@ func (tree *MutableTree) Load() (int64, error) {
 	return tree.LoadVersion(int64(0))
 }
 
+// SetVersion set current version of the tree. Only used in upgrade
+func (tree *MutableTree) SetVersion(version int64) {
+	tree.version = version
+	tree.ndb.latestVersion = version
+}
+
 // Returns the version number of the latest version found
 func (tree *MutableTree) LoadVersion(targetVersion int64) (int64, error) {
 	roots, err := tree.ndb.getRoots()