add mutex for baseNode

multiversx · Dec 16, 2024 · 354812c · 354812c
1 parent 4d5f3e2
commit 354812c
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/trie/baseNode.go b/trie/baseNode.go
@@ -16,15 +16,16 @@ type baseNode struct {
 }
 
 func (bn *baseNode) getHash() []byte {
-	//TODO add mutex protection for all methods
-
 	bn.mutex.RLock()
 	defer bn.mutex.RUnlock()
 
 	return bn.hash
 }
 
 func (bn *baseNode) setGivenHash(hash []byte) {
+	bn.mutex.Lock()
+	defer bn.mutex.Unlock()
+
 	bn.hash = hash
 }
 
@@ -36,6 +37,9 @@ func (bn *baseNode) isDirty() bool {
 }
 
 func (bn *baseNode) setDirty(dirty bool) {
+	bn.mutex.Lock()
+	defer bn.mutex.Unlock()
+
 	bn.dirty = dirty
 }