Mirroristas/kaspad
2
0
Fork 0
mirror of https://github.com/kaspanet/kaspad.git synced 2025-06-05 05:36:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

Change removeTransactionAndItsChainedTransactions to be non-recursive (#1696)

Browse Source 
* Change removeTransactionAndItsChainedTransactions to be non-recursive

* Split the variables assigning.

* Change names of function and variables.

* Append the correct queue.

Co-authored-by: tal <tal@daglabs.com>
Co-authored-by: stasatdaglabs <39559713+stasatdaglabs@users.noreply.github.com>
This commit is contained in:
talelbaz 2021-05-10 15:21:48 +03:00 committed by GitHub
parent 28a8e96e65
commit 19718ac102
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 35 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

55
domain/miningmanager/mempool/mempool.go
View File

@ -406,34 +406,49 @@ func (mp *mempool) removeTransactionsFromPool(txs []*consensusexternalapi.Domain
return nil
}
type transactionAndOutpoint struct {
transaction *consensusexternalapi.DomainTransaction
outpoint *consensusexternalapi.DomainOutpoint
}
// removeTransactionAndItsChainedTransactions removes a transaction and all of its chained transaction from the mempool.
//
// This function MUST be called with the mempool lock held (for writes).
func (mp *mempool) removeTransactionAndItsChainedTransactions(tx *consensusexternalapi.DomainTransaction) error {
txID := consensushashing.TransactionID(tx)
func (mp *mempool) removeTransactionAndItsChainedTransactions(transaction *consensusexternalapi.DomainTransaction) error {
transactionAndOutpointQueue := make([]*transactionAndOutpoint, 0)
transactionAndOutpointQueue = appendTransactionToTransactionAndOutpointQueue(transactionAndOutpointQueue, transaction)
// Remove any transactions which rely on this one.
for i := uint32(0); i < uint32(len(tx.Outputs)); i++ {
prevOut := consensusexternalapi.DomainOutpoint{TransactionID: *txID, Index: i}
if txRedeemer, exists := mp.mempoolUTXOSet.poolTransactionBySpendingOutpoint(prevOut); exists {
err := mp.removeTransactionAndItsChainedTransactions(txRedeemer)
if err != nil {
return err
}
for len(transactionAndOutpointQueue) > 0 {
txAndOutpoint := transactionAndOutpointQueue[0]
transactionAndOutpointQueue = transactionAndOutpointQueue[1:]
if txRedeemer, exists := mp.mempoolUTXOSet.poolTransactionBySpendingOutpoint(*txAndOutpoint.outpoint); exists {
transactionAndOutpointQueue = appendTransactionToTransactionAndOutpointQueue(transactionAndOutpointQueue, txRedeemer)
}
transactionID := txAndOutpoint.outpoint.TransactionID
if _, exists := mp.chainedTransactions[transactionID]; exists {
mp.removeChainTransaction(txAndOutpoint.transaction)
}
err := mp.cleanTransactionFromSets(txAndOutpoint.transaction)
if err != nil {
return err
}
}
if _, exists := mp.chainedTransactions[*txID]; exists {
mp.removeChainTransaction(tx)
}
err := mp.cleanTransactionFromSets(tx)
if err != nil {
return err
}
return nil
}
func appendTransactionToTransactionAndOutpointQueue(queue []*transactionAndOutpoint,
transaction *consensusexternalapi.DomainTransaction) []*transactionAndOutpoint {
transactionID := consensushashing.TransactionID(transaction)
queueWithAddedTransactionAndOutpoint := queue
for i := uint32(0); i < uint32(len(transaction.Outputs)); i++ {
previousOutpoint := consensusexternalapi.DomainOutpoint{TransactionID: *transactionID, Index: i}
queueWithAddedTransactionAndOutpoint = append(queueWithAddedTransactionAndOutpoint,
&transactionAndOutpoint{transaction: transaction, outpoint: &previousOutpoint})
}
return queueWithAddedTransactionAndOutpoint
}
// cleanTransactionFromSets removes the transaction from all mempool related transaction sets.
// It assumes that any chained transaction is already cleaned from the mempool.
//