chg : modify stateObjectsMu

maticnetwork · Aug 4, 2023 · 6220893 · 6220893
1 parent 4d5cdae
commit 6220893
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/core/state/statedb.go b/core/state/statedb.go
@@ -936,6 +936,9 @@ func (s *StateDB) getStateObject(addr common.Address) *stateObject {
 // destructed object instead of wiping all knowledge about the state object.
 func (s *StateDB) getDeletedStateObject(addr common.Address) *stateObject {
 	return MVRead(s, blockstm.NewAddressKey(addr), nil, func(s *StateDB) *stateObject {
+		s.stateObjectsMu.Lock()
+		defer s.stateObjectsMu.Unlock()
+
 		// Prefer live objects if any is available
 		if obj := s.stateObjects[addr]; obj != nil {
 			return obj
@@ -1000,9 +1003,6 @@ func (s *StateDB) getDeletedStateObject(addr common.Address) *stateObject {
 }
 
 func (s *StateDB) setStateObject(object *stateObject) {
-	s.stateObjectsMu.Lock()
-	defer s.stateObjectsMu.Unlock()
-
 	s.stateObjects[object.Address()] = object
 }