-
Notifications
You must be signed in to change notification settings - Fork 1
/
mempool_utxo_set.go
82 lines (63 loc) · 2.99 KB
/
mempool_utxo_set.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
package mempool
import (
"fmt"
"github.com/Nirvana-Chain/nirvanad/domain/consensus/utils/constants"
"github.com/Nirvana-Chain/nirvanad/domain/consensus/utils/consensushashing"
"github.com/Nirvana-Chain/nirvanad/domain/consensus/model/externalapi"
"github.com/Nirvana-Chain/nirvanad/domain/consensus/utils/utxo"
"github.com/Nirvana-Chain/nirvanad/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
}