ethrpc: add mutex to memoized chain id

0xsequence · Sep 6, 2024 · 6cd867a · 6cd867a
1 parent ad7994c
commit 6cd867a
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/ethrpc/ethrpc.go b/ethrpc/ethrpc.go
@@ -9,6 +9,7 @@ import (
 	"io"
 	"math/big"
 	"net/http"
+	"sync"
 	"sync/atomic"
 
 	"github.com/0xsequence/ethkit/ethrpc/jsonrpc"
@@ -30,7 +31,9 @@ type Provider struct {
 	br         breaker.Breaker
 	jwtToken   string // optional
 
-	chainID *big.Int
+	chainID   *big.Int
+	chainIDMu sync.Mutex
+
 	// cache   cachestore.Store[[]byte] // NOTE: unused for now
 	lastRequestID uint64
 
@@ -165,6 +168,8 @@ func (p *Provider) Do(ctx context.Context, calls ...Call) ([]byte, error) {
 }
 
 func (p *Provider) ChainID(ctx context.Context) (*big.Int, error) {
+	p.chainIDMu.Lock()
+	defer p.chainIDMu.Unlock()
 	if p.chainID != nil {
 		// chainID is memoized
 		return p.chainID, nil