feat: update nonce error handling

Bisonai · Oct 3, 2024 · 9c15751 · 9c15751
1 parent 1b15fcd
commit 9c15751
Show file tree

Hide file tree

Showing 2 changed files with 26 additions and 0 deletions.
diff --git a/node/pkg/chain/helper/helper.go b/node/pkg/chain/helper/helper.go
@@ -275,6 +275,10 @@ func (t *ChainHelper) SubmitDelegatedFallbackDirect(ctx context.Context, contrac
 				} else if errors.Is(err, errorSentinel.ErrChainTransactionFail) {
 					return err // if transaction fails, the data will probably be too old to retry
 				} else if utils.IsNonceError(err) || err == context.DeadlineExceeded {
+					err = noncemanager.ResetNonce(ctx, t.wallet, t.clients[clientIndex])
+					if err != nil {
+						return err
+					}
 					nonce, err = noncemanager.GetAndIncrementNonce(t.wallet)
 					if err != nil {
 						return err
@@ -302,6 +306,10 @@ func (t *ChainHelper) SubmitDelegatedFallbackDirect(ctx context.Context, contrac
 			} else if errors.Is(err, errorSentinel.ErrChainTransactionFail) {
 				return err // if transaction fails, the data will probably be too old to retry
 			} else if utils.IsNonceError(err) || err == context.DeadlineExceeded {
+				err = noncemanager.ResetNonce(ctx, t.wallet, t.clients[clientIndex])
+				if err != nil {
+					return err
+				}
 				nonce, err = noncemanager.GetAndIncrementNonce(t.wallet)
 				if err != nil {
 					return err

diff --git a/node/pkg/chain/noncemanager/noncemanager.go b/node/pkg/chain/noncemanager/noncemanager.go
@@ -1,8 +1,11 @@
 package noncemanager
 
 import (
+	"context"
 	"fmt"
 	"sync"
+
+	"bisonai.com/miko/node/pkg/chain/utils"
 )
 
 type NonceManager struct {
@@ -25,6 +28,10 @@ func Get() *NonceManager {
 	return Manager
 }
 
+func ResetNonce(ctx context.Context, address string, client utils.ClientInterface) error {
+	return Get().ResetNonce(ctx, address, client)
+}
+
 func Set(address string, nonce uint64) {
 	Get().SetNonce(address, nonce)
 }
@@ -33,6 +40,17 @@ func GetAndIncrementNonce(address string) (uint64, error) {
 	return Get().GetAndIncrementNonce(address)
 }
 
+func (m *NonceManager) ResetNonce(ctx context.Context, address string, client utils.ClientInterface) error {
+	m.mu.Lock()
+	defer m.mu.Unlock()
+	nonce, err := utils.GetNonceFromPk(ctx, address, client)
+	if err != nil {
+		return err
+	}
+	m.nonces[address] = nonce
+	return nil
+}
+
 func (m *NonceManager) SetNonce(address string, nonce uint64) {
 	m.mu.Lock()
 	defer m.mu.Unlock()