mirror of
https://github.com/kaspanet/kaspad.git
synced 2025-05-20 13:56:45 +00:00
eb693c4a86
* fix mempool accessing, rewrite get_mempool_entries_by_addresses * fix counter, add verbose * fmt * addresses as string * Define error in case utxoEntry is missing. * fix error variable to string * stop tests from failing (see in code comment) * access both pools in the same state via parameters * get rid of todo message * fmt - very important! * perf: scriptpublickey in mempool, no txscript. * address reveiw * fmt fix * mixed up isorphan bool, pass tests now * do map preallocation in mempoolbyaddresses * no proallocation for orphanpool sending. Co-authored-by: Ori Newman <orinewman1@gmail.com>
83 lines
3.0 KiB
Go
83 lines
3.0 KiB
Go
package mempool
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/kaspanet/kaspad/domain/consensus/utils/constants"
|
|
|
|
"github.com/kaspanet/kaspad/domain/consensus/utils/consensushashing"
|
|
|
|
"github.com/kaspanet/kaspad/domain/consensus/model/externalapi"
|
|
"github.com/kaspanet/kaspad/domain/consensus/utils/utxo"
|
|
"github.com/kaspanet/kaspad/domain/miningmanager/mempool/model"
|
|
)
|
|
|
|
type mempoolUTXOSet struct {
|
|
mempool *mempool
|
|
poolUnspentOutputs model.OutpointToUTXOEntryMap
|
|
transactionByPreviousOutpoint model.OutpointToTransactionMap
|
|
}
|
|
|
|
func newMempoolUTXOSet(mp *mempool) *mempoolUTXOSet {
|
|
return &mempoolUTXOSet{
|
|
mempool: mp,
|
|
poolUnspentOutputs: model.OutpointToUTXOEntryMap{},
|
|
transactionByPreviousOutpoint: model.OutpointToTransactionMap{},
|
|
}
|
|
}
|
|
|
|
func (mpus *mempoolUTXOSet) addTransaction(transaction *model.MempoolTransaction) {
|
|
outpoint := &externalapi.DomainOutpoint{TransactionID: *transaction.TransactionID()}
|
|
|
|
for i, input := range transaction.Transaction().Inputs {
|
|
outpoint.Index = uint32(i)
|
|
|
|
// Delete the output this input spends, in case it was created by mempool.
|
|
// If the outpoint doesn't exist in mpus.poolUnspentOutputs - this means
|
|
// it was created in the DAG (a.k.a. in consensus).
|
|
delete(mpus.poolUnspentOutputs, *outpoint)
|
|
|
|
mpus.transactionByPreviousOutpoint[input.PreviousOutpoint] = transaction
|
|
}
|
|
|
|
for i, output := range transaction.Transaction().Outputs {
|
|
outpoint := externalapi.DomainOutpoint{TransactionID: *transaction.TransactionID(), Index: uint32(i)}
|
|
|
|
mpus.poolUnspentOutputs[outpoint] =
|
|
utxo.NewUTXOEntry(output.Value, output.ScriptPublicKey, false, constants.UnacceptedDAAScore)
|
|
}
|
|
}
|
|
|
|
func (mpus *mempoolUTXOSet) removeTransaction(transaction *model.MempoolTransaction) {
|
|
for _, input := range transaction.Transaction().Inputs {
|
|
// If the transaction creating the output spent by this input is in the mempool - restore it's UTXO
|
|
if _, ok := mpus.mempool.transactionsPool.getTransaction(&input.PreviousOutpoint.TransactionID, false); ok {
|
|
mpus.poolUnspentOutputs[input.PreviousOutpoint] = input.UTXOEntry
|
|
}
|
|
delete(mpus.transactionByPreviousOutpoint, input.PreviousOutpoint)
|
|
}
|
|
|
|
outpoint := externalapi.DomainOutpoint{TransactionID: *transaction.TransactionID()}
|
|
for i := range transaction.Transaction().Outputs {
|
|
outpoint.Index = uint32(i)
|
|
|
|
delete(mpus.poolUnspentOutputs, outpoint)
|
|
}
|
|
}
|
|
|
|
func (mpus *mempoolUTXOSet) checkDoubleSpends(transaction *externalapi.DomainTransaction) error {
|
|
outpoint := externalapi.DomainOutpoint{TransactionID: *consensushashing.TransactionID(transaction)}
|
|
|
|
for i, input := range transaction.Inputs {
|
|
outpoint.Index = uint32(i)
|
|
|
|
if existingTransaction, exists := mpus.transactionByPreviousOutpoint[input.PreviousOutpoint]; exists {
|
|
str := fmt.Sprintf("output %s already spent by transaction %s in the memory pool",
|
|
input.PreviousOutpoint, existingTransaction.TransactionID())
|
|
return transactionRuleError(RejectDuplicate, str)
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|