Fix error

smartcontractkit · Apr 1, 2024 · b7709d7 · b7709d7
1 parent f28263e
commit b7709d7
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/core/services/ocr2/plugins/ccip/tokendata/observability/usdc_client_test.go b/core/services/ocr2/plugins/ccip/tokendata/observability/usdc_client_test.go
@@ -90,7 +90,7 @@ func testMonitoring(t *testing.T, name string, server *httptest.Server, requests
 	usdcTokenAddr := utils.RandomAddress()
 	observedHttpClient := http2.NewObservedIHttpClientWithMetric(&http2.HttpClient{}, histogram)
 	tokenDataReaderDefault := usdc.NewUSDCTokenDataReader(log, usdcReader, attestationURI, 0, usdcTokenAddr, 0)
-	tokenDataReader := usdc.NewUSDCTokenDataReaderWithHttpClient(*tokenDataReaderDefault, observedHttpClient, usdcTokenAddr)
+	tokenDataReader := usdc.NewUSDCTokenDataReaderWithHttpClient(*tokenDataReaderDefault, observedHttpClient, usdcTokenAddr, 0)
 	require.NotNil(t, tokenDataReader)
 
 	for i := 0; i < requests; i++ {

diff --git a/core/services/ocr2/plugins/ccip/tokendata/usdc/usdc.go b/core/services/ocr2/plugins/ccip/tokendata/usdc/usdc.go
@@ -131,6 +131,7 @@ func NewUSDCTokenDataReaderWithHttpClient(
 	origin TokenDataReader,
 	httpClient http.IHttpClient,
 	usdcTokenAddress common.Address,
+	requestInterval time.Duration,
 ) *TokenDataReader {
 	return &TokenDataReader{
 		lggr:                  origin.lggr,
@@ -140,6 +141,7 @@ func NewUSDCTokenDataReaderWithHttpClient(
 		attestationApiTimeout: origin.attestationApiTimeout,
 		coolDownMu:            origin.coolDownMu,
 		usdcTokenAddress:      usdcTokenAddress,
+		rate:                  rate.NewLimiter(rate.Every(requestInterval), 1),
 	}
 }
 
@@ -154,7 +156,7 @@ func (s *TokenDataReader) ReadTokenData(ctx context.Context, msg cciptypes.EVM2E
 	}
 
 	// Using 'Allow' instead of 'Wait' to avoid blocking the current goroutine.
-	if !s.rate.Allow() {
+	if s.rate != nil && !s.rate.Allow() {
 		// self rate limiting to avoid hitting the rate limit.
 		return nil, tokendata.ErrSelfRateLimit
 	}