mirror of
https://github.com/kaspanet/kaspad.git
synced 2025-11-24 06:25:55 +00:00
102 lines
2.9 KiB
Go
102 lines
2.9 KiB
Go
package utxodiffstore
|
|
|
|
import (
|
|
"github.com/kaspanet/kaspad/domain/consensus/model"
|
|
"github.com/kaspanet/kaspad/domain/consensus/model/externalapi"
|
|
"github.com/kaspanet/kaspad/domain/consensus/utils/utxo"
|
|
"math/rand"
|
|
"testing"
|
|
)
|
|
|
|
func BenchmarkUTXODiffSerialization(b *testing.B) {
|
|
utxoDiffStore := New(0).(*utxoDiffStore)
|
|
|
|
b.ResetTimer()
|
|
for i := 0; i < b.N; i++ {
|
|
b.StopTimer()
|
|
testUTXODiff := buildTestUTXODiff(b)
|
|
b.StartTimer()
|
|
_, err := utxoDiffStore.serializeUTXODiff(testUTXODiff)
|
|
if err != nil {
|
|
b.Fatalf("Could not serialize UTXO diff: %s", err)
|
|
}
|
|
}
|
|
}
|
|
|
|
func BenchmarkUTXODiffDeserialization(b *testing.B) {
|
|
utxoDiffStore := New(0).(*utxoDiffStore)
|
|
|
|
b.ResetTimer()
|
|
for i := 0; i < b.N; i++ {
|
|
b.StopTimer()
|
|
testUTXODiff := buildTestUTXODiff(b)
|
|
serializedUTXODiff, err := utxoDiffStore.serializeUTXODiff(testUTXODiff)
|
|
if err != nil {
|
|
b.Fatalf("Could not serialize UTXO diff: %s", err)
|
|
}
|
|
b.StartTimer()
|
|
_, err = utxoDiffStore.deserializeUTXODiff(serializedUTXODiff)
|
|
if err != nil {
|
|
b.Fatalf("Could not deserialize UTXO diff: %s", err)
|
|
}
|
|
}
|
|
}
|
|
|
|
func BenchmarkUTXODiffSerializationAndDeserialization(b *testing.B) {
|
|
utxoDiffStore := New(0).(*utxoDiffStore)
|
|
|
|
b.ResetTimer()
|
|
for i := 0; i < b.N; i++ {
|
|
b.StopTimer()
|
|
testUTXODiff := buildTestUTXODiff(b)
|
|
b.StartTimer()
|
|
serializedUTXODiff, err := utxoDiffStore.serializeUTXODiff(testUTXODiff)
|
|
if err != nil {
|
|
b.Fatalf("Could not serialize UTXO diff: %s", err)
|
|
}
|
|
_, err = utxoDiffStore.deserializeUTXODiff(serializedUTXODiff)
|
|
if err != nil {
|
|
b.Fatalf("Could not deserialize UTXO diff: %s", err)
|
|
}
|
|
}
|
|
}
|
|
|
|
func buildTestUTXODiff(b *testing.B) model.UTXODiff {
|
|
toAdd := buildTestUTXOCollection()
|
|
toRemove := buildTestUTXOCollection()
|
|
|
|
utxoDiff, err := utxo.NewUTXODiffFromCollections(toAdd, toRemove)
|
|
if err != nil {
|
|
b.Fatalf("Could not create UTXODiff from toAdd and toRemove collections: %s", err)
|
|
}
|
|
return utxoDiff
|
|
}
|
|
|
|
func buildTestUTXOCollection() model.UTXOCollection {
|
|
utxoMap := make(map[externalapi.DomainOutpoint]externalapi.UTXOEntry)
|
|
|
|
for i := 0; i < 100_000; i++ {
|
|
var outpointTransactionIDBytes [32]byte
|
|
rand.Read(outpointTransactionIDBytes[:])
|
|
outpointTransactionID := externalapi.NewDomainTransactionIDFromByteArray(&outpointTransactionIDBytes)
|
|
outpointIndex := rand.Uint32()
|
|
outpoint := externalapi.NewDomainOutpoint(outpointTransactionID, outpointIndex)
|
|
|
|
utxoEntryAmount := rand.Uint64()
|
|
var utxoEntryScriptPublicKeyScript [256]byte
|
|
rand.Read(utxoEntryScriptPublicKeyScript[:])
|
|
utxoEntryScriptPublicKeyVersion := uint16(0)
|
|
utxoEntryScriptPublicKey := &externalapi.ScriptPublicKey{
|
|
Script: utxoEntryScriptPublicKeyScript[:],
|
|
Version: utxoEntryScriptPublicKeyVersion,
|
|
}
|
|
isCoinbase := rand.Float32() > 0.5
|
|
blockBlueScore := rand.Uint64()
|
|
utxoEntry := utxo.NewUTXOEntry(utxoEntryAmount, utxoEntryScriptPublicKey, isCoinbase, blockBlueScore)
|
|
|
|
utxoMap[*outpoint] = utxoEntry
|
|
}
|
|
|
|
return utxo.NewUTXOCollection(utxoMap)
|
|
}
|