mirror of
https://github.com/kaspanet/kaspad.git
synced 2025-06-06 06:06:49 +00:00
5566aaf95a
* [NOD-1512] Implement UTXO set deserialization * [NOD-1512] Remove redundant file * [NOD-1512] Don't use big endian for serialization * [NOD-1512] Use Read/Write element * [NOD-1512] Unexport ReadElement * [NOD-1512] Fix StageVirtualUTXOSet * [NOD-1512] Get rid of dagParams in consensusStateManager * [NOD-1512] Get rid of dagParams in consensusStateManager
100 lines
2.7 KiB
Go
100 lines
2.7 KiB
Go
package consensusstatemanager
|
|
|
|
import (
|
|
"github.com/kaspanet/kaspad/domain/consensus/model"
|
|
"github.com/kaspanet/kaspad/domain/consensus/model/externalapi"
|
|
"github.com/kaspanet/kaspad/domain/consensus/utils/consensusserialization"
|
|
"github.com/kaspanet/kaspad/domain/consensus/utils/multiset"
|
|
"github.com/kaspanet/kaspad/domain/consensus/utils/utxoserialization"
|
|
)
|
|
|
|
func (csm *consensusStateManager) calculateMultiset(
|
|
acceptanceData model.AcceptanceData, blockGHOSTDAGData *model.BlockGHOSTDAGData) (model.Multiset, error) {
|
|
|
|
ms, err := csm.multisetStore.Get(csm.databaseContext, blockGHOSTDAGData.SelectedParent)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
for _, blockAcceptanceData := range acceptanceData {
|
|
for i, transactionAcceptanceData := range blockAcceptanceData.TransactionAcceptanceData {
|
|
if !transactionAcceptanceData.IsAccepted {
|
|
continue
|
|
}
|
|
|
|
transaction := transactionAcceptanceData.Transaction
|
|
|
|
isCoinbase := i == 0
|
|
var err error
|
|
err = addTransactionToMultiset(ms, transaction, blockGHOSTDAGData.BlueScore, isCoinbase)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
}
|
|
|
|
return ms, nil
|
|
}
|
|
|
|
func addTransactionToMultiset(multiset model.Multiset, transaction *externalapi.DomainTransaction,
|
|
blockBlueScore uint64, isCoinbase bool) error {
|
|
|
|
for _, input := range transaction.Inputs {
|
|
err := removeUTXOFromMultiset(multiset, input.UTXOEntry, &input.PreviousOutpoint)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
for i, output := range transaction.Outputs {
|
|
outpoint := &externalapi.DomainOutpoint{
|
|
TransactionID: *consensusserialization.TransactionID(transaction),
|
|
Index: uint32(i),
|
|
}
|
|
utxoEntry := &externalapi.UTXOEntry{
|
|
Amount: output.Value,
|
|
ScriptPublicKey: output.ScriptPublicKey,
|
|
BlockBlueScore: blockBlueScore,
|
|
IsCoinbase: isCoinbase,
|
|
}
|
|
err := addUTXOToMultiset(multiset, utxoEntry, outpoint)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func addUTXOToMultiset(multiset model.Multiset, entry *externalapi.UTXOEntry,
|
|
outpoint *externalapi.DomainOutpoint) error {
|
|
|
|
serializedUTXO, err := consensusserialization.SerializeUTXO(entry, outpoint)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
multiset.Add(serializedUTXO)
|
|
|
|
return nil
|
|
}
|
|
|
|
func removeUTXOFromMultiset(multiset model.Multiset, entry *externalapi.UTXOEntry,
|
|
outpoint *externalapi.DomainOutpoint) error {
|
|
|
|
serializedUTXO, err := consensusserialization.SerializeUTXO(entry, outpoint)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
multiset.Remove(serializedUTXO)
|
|
|
|
return nil
|
|
}
|
|
|
|
func calcMultisetFromProtoUTXOSet(protoUTXOSet *utxoserialization.ProtoUTXOSet) (model.Multiset, error) {
|
|
ms := multiset.New()
|
|
for _, utxo := range protoUTXOSet.Utxos {
|
|
ms.Add(utxo.EntryOutpointPair)
|
|
}
|
|
return ms, nil
|
|
}
|