fix context

noble-assets · Mar 4, 2024 · bd66859 · bd66859
1 parent e0370b4
commit bd66859
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/ethereum/listener.go b/ethereum/listener.go
@@ -158,6 +158,13 @@ func (e *Ethereum) WalletBalanceMetric(ctx context.Context, logger log.Logger, m
 				time.Sleep(time.Duration(queryRate) * time.Second)
 				firstTime = false
 			}
+
+			// check if context cancelled durint sleep call above
+			_, isDone := <-ctx.Done()
+			if isDone {
+				return
+			}
+
 			if createClient {
 				client, err = ethclient.DialContext(ctx, e.rpcURL)
 				if err != nil {