From d3a754e73547c09f652c6d26d4e0e7269e1084b8 Mon Sep 17 00:00:00 2001
From: Jared Wasinger <j-wasinger@hotmail.com>
Date: Mon, 23 Sep 2024 20:41:31 +0700
Subject: [PATCH] try better way

---
 core/txpool/blobpool/blobpool.go | 105 +++++++++++++++++--------------
 core/txpool/subpool.go           |   3 +
 2 files changed, 61 insertions(+), 47 deletions(-)

diff --git a/core/txpool/blobpool/blobpool.go b/core/txpool/blobpool/blobpool.go
index 685c017cbd..aa6b796703 100644
--- a/core/txpool/blobpool/blobpool.go
+++ b/core/txpool/blobpool/blobpool.go
@@ -1013,6 +1013,56 @@ func (p *BlobPool) reinject(addr common.Address, txhash common.Hash) error {
 	return nil
 }
 
+func (p *BlobPool) flushTransactionsBelowTip(tip *uint256.Int) {
+	for addr, txs := range p.index {
+		for i, tx := range txs {
+			if tx.execTipCap.Cmp(tip) < 0 {
+				// Drop the offending transaction
+				var (
+					ids    = []uint64{tx.id}
+					nonces = []uint64{tx.nonce}
+				)
+				p.spent[addr] = new(uint256.Int).Sub(p.spent[addr], txs[i].costCap)
+				p.stored -= uint64(tx.size)
+				delete(p.lookup, tx.hash)
+				txs[i] = nil
+
+				// Drop everything afterwards, no gaps allowed
+				for j, tx := range txs[i+1:] {
+					ids = append(ids, tx.id)
+					nonces = append(nonces, tx.nonce)
+
+					p.spent[addr] = new(uint256.Int).Sub(p.spent[addr], tx.costCap)
+					p.stored -= uint64(tx.size)
+					delete(p.lookup, tx.hash)
+					txs[i+1+j] = nil
+				}
+				// Clear out the dropped transactions from the index
+				if i > 0 {
+					p.index[addr] = txs[:i]
+					heap.Fix(p.evict, p.evict.index[addr])
+				} else {
+					delete(p.index, addr)
+					delete(p.spent, addr)
+
+					heap.Remove(p.evict, p.evict.index[addr])
+					p.reserve(addr, false)
+				}
+				// Clear out the transactions from the data store
+				log.Warn("Dropping underpriced blob transaction", "from", addr, "rejected", tx.nonce, "tip", tx.execTipCap, "want", tip, "drop", nonces, "ids", ids)
+				dropUnderpricedMeter.Mark(int64(len(ids)))
+
+				for _, id := range ids {
+					if err := p.store.Delete(id); err != nil {
+						log.Error("Failed to delete dropped transaction", "id", id, "err", err)
+					}
+				}
+				break
+			}
+		}
+	}
+}
+
 // SetGasTip implements txpool.SubPool, allowing the blob pool's gas requirements
 // to be kept in sync with the main transaction pool's gas requirements.
 func (p *BlobPool) SetGasTip(tip *big.Int) {
@@ -1025,59 +1075,20 @@ func (p *BlobPool) SetGasTip(tip *big.Int) {
 
 	// If the min miner fee increased, remove transactions below the new threshold
 	if old == nil || p.gasTip.Cmp(old) > 0 {
-		for addr, txs := range p.index {
-			for i, tx := range txs {
-				if tx.execTipCap.Cmp(p.gasTip) < 0 {
-					// Drop the offending transaction
-					var (
-						ids    = []uint64{tx.id}
-						nonces = []uint64{tx.nonce}
-					)
-					p.spent[addr] = new(uint256.Int).Sub(p.spent[addr], txs[i].costCap)
-					p.stored -= uint64(tx.size)
-					delete(p.lookup, tx.hash)
-					txs[i] = nil
-
-					// Drop everything afterwards, no gaps allowed
-					for j, tx := range txs[i+1:] {
-						ids = append(ids, tx.id)
-						nonces = append(nonces, tx.nonce)
-
-						p.spent[addr] = new(uint256.Int).Sub(p.spent[addr], tx.costCap)
-						p.stored -= uint64(tx.size)
-						delete(p.lookup, tx.hash)
-						txs[i+1+j] = nil
-					}
-					// Clear out the dropped transactions from the index
-					if i > 0 {
-						p.index[addr] = txs[:i]
-						heap.Fix(p.evict, p.evict.index[addr])
-					} else {
-						delete(p.index, addr)
-						delete(p.spent, addr)
-
-						heap.Remove(p.evict, p.evict.index[addr])
-						p.reserve(addr, false)
-					}
-					// Clear out the transactions from the data store
-					log.Warn("Dropping underpriced blob transaction", "from", addr, "rejected", tx.nonce, "tip", tx.execTipCap, "want", tip, "drop", nonces, "ids", ids)
-					dropUnderpricedMeter.Mark(int64(len(ids)))
-
-					for _, id := range ids {
-						if err := p.store.Delete(id); err != nil {
-							log.Error("Failed to delete dropped transaction", "id", id, "err", err)
-						}
-					}
-					break
-				}
-			}
-		}
+		p.flushTransactionsBelowTip(p.gasTip)
 	}
 	log.Debug("Blobpool tip threshold updated", "tip", tip)
 	pooltipGauge.Update(tip.Int64())
 	p.updateStorageMetrics()
 }
 
+func (p *BlobPool) FlushAllTransactions() {
+	maxUint256 := new(big.Int).Sub(new(big.Int).Lsh(common.Big1, 256), common.Big1)
+	p.lock.Lock()
+	defer p.lock.Unlock()
+	p.flushTransactionsBelowTip(maxUint256)
+}
+
 // validateTx checks whether a transaction is valid according to the consensus
 // rules and adheres to some heuristic limits of the local node (price and size).
 func (p *BlobPool) validateTx(tx *types.Transaction) error {
diff --git a/core/txpool/subpool.go b/core/txpool/subpool.go
index 9881ed1b8f..e361eabd4e 100644
--- a/core/txpool/subpool.go
+++ b/core/txpool/subpool.go
@@ -116,6 +116,9 @@ type SubPool interface {
 	// transaction, and drops all transactions below this threshold.
 	SetGasTip(tip *big.Int)
 
+	// FlushAllTransactions drops all transactions in the pool.
+	FlushAllTransactions()
+
 	// Has returns an indicator whether subpool has a transaction cached with the
 	// given hash.
 	Has(hash common.Hash) bool