From 7fb4bb167d8ee263955a22f0f629d0997935ecd2 Mon Sep 17 00:00:00 2001
From: javiersuweijie <javier.su.weijie@gmail.com>
Date: Wed, 12 Jul 2023 17:33:05 +0800
Subject: [PATCH] fix: concurrent access

---
 mutable_tree.go | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/mutable_tree.go b/mutable_tree.go
index f5856125b..215ecde54 100644
--- a/mutable_tree.go
+++ b/mutable_tree.go
@@ -193,6 +193,8 @@ func (tree *MutableTree) Iterate(fn func(key []byte, value []byte) bool) (stoppe
 		return tree.ImmutableTree.Iterate(fn)
 	}
 
+	tree.mtx.Lock()
+	defer tree.mtx.Unlock()
 	itr := NewUnsavedFastIterator(nil, nil, true, tree.ndb, tree.unsavedFastNodeAdditions, tree.unsavedFastNodeRemovals)
 	defer itr.Close()
 	for ; itr.Valid(); itr.Next() {
@@ -213,6 +215,8 @@ func (tree *MutableTree) Iterator(start, end []byte, ascending bool) (dbm.Iterat
 		}
 
 		if isFastCacheEnabled {
+			tree.mtx.Lock()
+			defer tree.mtx.Unlock()
 			return NewUnsavedFastIterator(start, end, ascending, tree.ndb, tree.unsavedFastNodeAdditions, tree.unsavedFastNodeRemovals), nil
 		}
 	}
@@ -965,6 +969,8 @@ func (tree *MutableTree) getUnsavedFastNodeRemovals() map[string]interface{} {
 }
 
 func (tree *MutableTree) addUnsavedAddition(key []byte, node *FastNode) {
+	tree.mtx.Lock()
+	defer tree.mtx.Unlock()
 	skey := unsafeToStr(key)
 	delete(tree.unsavedFastNodeRemovals, skey)
 	tree.unsavedFastNodeAdditions[skey] = node
@@ -972,6 +978,8 @@ func (tree *MutableTree) addUnsavedAddition(key []byte, node *FastNode) {
 
 func (tree *MutableTree) saveFastNodeAdditions() error {
 	keysToSort := make([]string, 0, len(tree.unsavedFastNodeAdditions))
+	tree.mtx.Lock()
+	defer tree.mtx.Unlock()
 	for key := range tree.unsavedFastNodeAdditions {
 		keysToSort = append(keysToSort, key)
 	}