fix: data race of latestVersion (#834)

cosmos · Sep 13, 2023 · 4de71fd · 4de71fd
1 parent 6aff580
commit 4de71fd
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
 }