fix: data race of latestVersion (backport #834) (#835)

Co-authored-by: cool-developer <51834436+cool-develope@users.noreply.github.com>
cosmos · Sep 13, 2023 · fdc599d · fdc599d
1 parent f528c1f
commit fdc599d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/mutable_tree.go b/mutable_tree.go
@@ -797,7 +797,6 @@ func (tree *MutableTree) saveFastNodeVersion(isGenesis bool) error {
 	return tree.ndb.setFastStorageVersionToBatch()
 }
 
-// nolint: unused
 func (tree *MutableTree) getUnsavedFastNodeAdditions() map[string]*fastnode.Node {
 	additions := make(map[string]*fastnode.Node)
 	tree.unsavedFastNodeAdditions.Range(func(key, value interface{}) bool {

diff --git a/nodedb.go b/nodedb.go
@@ -733,6 +733,8 @@ func (ndb *nodeDB) getLatestVersion() (int64, error) {
 }
 
 func (ndb *nodeDB) resetLatestVersion(version int64) {
+	ndb.mtx.Lock()
+	defer ndb.mtx.Unlock()
 	ndb.latestVersion = version
 }