mirror of
https://github.com/kaspanet/kaspad.git
synced 2025-05-28 09:46:50 +00:00
23cccb6396
* [NOD-1497] Add missing APIs. * [NOD-1497] Rename some new APIs. * [NOD-1497] Add fields to BlockInfo. * [NOD-1497] Add comments over BlockInfo and BlockLocator. * [NOD-1497] Rename GetSelectedParent to GetVirtualSelectedParent. * [NOD-1497] Add SetPruningPointUTXOSet. * [NOD-1497] Rename GetHashesAbovePruningPoint to GetMissingBlockBodyHashes. * [NOD-1497] Fix rename error.
99 lines
2.8 KiB
Go
99 lines
2.8 KiB
Go
package blockstatusstore
|
|
|
|
import (
|
|
"github.com/golang/protobuf/proto"
|
|
"github.com/kaspanet/kaspad/domain/consensus/database/serialization"
|
|
"github.com/kaspanet/kaspad/domain/consensus/model"
|
|
"github.com/kaspanet/kaspad/domain/consensus/model/externalapi"
|
|
"github.com/kaspanet/kaspad/domain/consensus/utils/dbkeys"
|
|
)
|
|
|
|
var bucket = dbkeys.MakeBucket([]byte("block-statuses"))
|
|
|
|
// blockStatusStore represents a store of BlockStatuses
|
|
type blockStatusStore struct {
|
|
staging map[externalapi.DomainHash]externalapi.BlockStatus
|
|
}
|
|
|
|
// New instantiates a new BlockStatusStore
|
|
func New() model.BlockStatusStore {
|
|
return &blockStatusStore{
|
|
staging: make(map[externalapi.DomainHash]externalapi.BlockStatus),
|
|
}
|
|
}
|
|
|
|
// Stage stages the given blockStatus for the given blockHash
|
|
func (bss *blockStatusStore) Stage(blockHash *externalapi.DomainHash, blockStatus externalapi.BlockStatus) {
|
|
bss.staging[*blockHash] = blockStatus
|
|
}
|
|
|
|
func (bss *blockStatusStore) IsStaged() bool {
|
|
return len(bss.staging) != 0
|
|
}
|
|
|
|
func (bss *blockStatusStore) Discard() {
|
|
bss.staging = make(map[externalapi.DomainHash]externalapi.BlockStatus)
|
|
}
|
|
|
|
func (bss *blockStatusStore) Commit(dbTx model.DBTransaction) error {
|
|
for hash, status := range bss.staging {
|
|
blockStatusBytes, err := bss.serializeBlockStatus(status)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
err = dbTx.Put(bss.hashAsKey(&hash), blockStatusBytes)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
bss.Discard()
|
|
return nil
|
|
}
|
|
|
|
// Get gets the blockStatus associated with the given blockHash
|
|
func (bss *blockStatusStore) Get(dbContext model.DBReader, blockHash *externalapi.DomainHash) (externalapi.BlockStatus, error) {
|
|
if status, ok := bss.staging[*blockHash]; ok {
|
|
return status, nil
|
|
}
|
|
|
|
statusBytes, err := dbContext.Get(bss.hashAsKey(blockHash))
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
|
|
return bss.deserializeHeader(statusBytes)
|
|
}
|
|
|
|
// Exists returns true if the blockStatus for the given blockHash exists
|
|
func (bss *blockStatusStore) Exists(dbContext model.DBReader, blockHash *externalapi.DomainHash) (bool, error) {
|
|
if _, ok := bss.staging[*blockHash]; ok {
|
|
return true, nil
|
|
}
|
|
|
|
exists, err := dbContext.Has(bss.hashAsKey(blockHash))
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
|
|
return exists, nil
|
|
}
|
|
|
|
func (bss *blockStatusStore) serializeBlockStatus(status externalapi.BlockStatus) ([]byte, error) {
|
|
dbBlockStatus := serialization.DomainBlockStatusToDbBlockStatus(status)
|
|
return proto.Marshal(dbBlockStatus)
|
|
}
|
|
|
|
func (bss *blockStatusStore) deserializeHeader(statusBytes []byte) (externalapi.BlockStatus, error) {
|
|
dbBlockStatus := &serialization.DbBlockStatus{}
|
|
err := proto.Unmarshal(statusBytes, dbBlockStatus)
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
return serialization.DbBlockStatusToDomainBlockStatus(dbBlockStatus), nil
|
|
}
|
|
|
|
func (bss *blockStatusStore) hashAsKey(hash *externalapi.DomainHash) model.DBKey {
|
|
return bucket.Key(hash[:])
|
|
}
|