Implemented a manage state for keeping track of nonces

1 files changed, 56 insertions, 0 deletions

diff --git a/state/managed_state.go b/state/managed_state.go
new file mode 100644
index 000000000..b0a129af0
--- /dev/null
+++ b/state/managed_state.go
@@ -0,0 +1,56 @@
+package state
+
+import "sync"
+
+type ManagedState struct {
+	*StateDB
+
+	mu sync.RWMutex
+
+	accounts map[string]*StateObject
+}
+
+func ManageState(statedb *StateDB) *ManagedState {
+	return &ManagedState{
+		StateDB:  statedb,
+		accounts: make(map[string]*StateObject),
+	}
+}
+
+func (ms *ManagedState) IncrementNonce(addr []byte) {
+	ms.mu.Lock()
+	defer ms.mu.Unlock()
+
+	ms.getAccount(addr).nonce++
+}
+
+func (ms *ManagedState) DecrementNonce(addr []byte) {
+	// Decrementing a nonce does not mean we are interested in the account
+	// incrementing only happens if you control the account, therefor
+	// incrementing  behaves differently from decrementing
+	if ms.hasAccount(addr) {
+		ms.mu.Lock()
+		defer ms.mu.Unlock()
+
+		ms.getAccount(addr).nonce--
+	}
+}
+
+func (ms *ManagedState) GetNonce(addr []byte) uint64 {
+	ms.mu.RLock()
+	defer ms.mu.RUnlock()
+	return ms.getAccount(addr).nonce
+}
+
+func (ms *ManagedState) hasAccount(addr []byte) bool {
+	_, ok := ms.accounts[string(addr)]
+	return ok
+}
+
+func (ms *ManagedState) getAccount(addr []byte) *StateObject {
+	if _, ok := ms.accounts[string(addr)]; !ok {
+		ms.accounts[string(addr)] = ms.GetOrNewStateObject(addr)
+	}
+
+	return ms.accounts[string(addr)]
+}