fix: fix potential goroutine leak (bnb-chain#1067)

tolunghk · Aug 31, 2022 · b485651 · b485651
1 parent f457598
commit b485651
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/core/state_prefetcher.go b/core/state_prefetcher.go
@@ -138,9 +138,14 @@ func (p *statePrefetcher) PrefetchMining(txs *types.TransactionsByPriceAndNonce,
 				if tx == nil {
 					return
 				}
-				txCh <- tx
-				txset.Shift()
 
+				select {
+				case <-interruptCh:
+					return
+				case txCh <- tx:
+				}
+
+				txset.Shift()
 			}
 		}
 	}(txs)