mirror of
https://github.com/kaspanet/kaspad.git
synced 2025-05-30 02:36:42 +00:00
eae8bce941
* [NOD-1491] Implement block headers store * [NOD-1491] Don't commit transaction and delete from staging too
109 lines
3.2 KiB
Go
109 lines
3.2 KiB
Go
package blockheaderstore
|
|
|
|
import (
|
|
"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-headers"))
|
|
|
|
// blockHeaderStore represents a store of blocks
|
|
type blockHeaderStore struct {
|
|
staging map[externalapi.DomainHash]*externalapi.DomainBlockHeader
|
|
}
|
|
|
|
// New instantiates a new BlockHeaderStore
|
|
func New() model.BlockHeaderStore {
|
|
return &blockHeaderStore{
|
|
staging: make(map[externalapi.DomainHash]*externalapi.DomainBlockHeader),
|
|
}
|
|
}
|
|
|
|
// Stage stages the given block header for the given blockHash
|
|
func (bms *blockHeaderStore) Stage(blockHash *externalapi.DomainHash, blockHeader *externalapi.DomainBlockHeader) {
|
|
bms.staging[*blockHash] = blockHeader
|
|
}
|
|
|
|
func (bms *blockHeaderStore) IsStaged() bool {
|
|
return len(bms.staging) != 0
|
|
}
|
|
|
|
func (bms *blockHeaderStore) Discard() {
|
|
bms.staging = make(map[externalapi.DomainHash]*externalapi.DomainBlockHeader)
|
|
}
|
|
|
|
func (bms *blockHeaderStore) Commit(dbTx model.DBTransaction) error {
|
|
for hash, header := range bms.staging {
|
|
err := dbTx.Put(bms.hashAsKey(&hash), bms.serializeHeader(header))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
bms.Discard()
|
|
return nil
|
|
}
|
|
|
|
// BlockHeader gets the block header associated with the given blockHash
|
|
func (bms *blockHeaderStore) BlockHeader(dbContext model.DBReader, blockHash *externalapi.DomainHash) (*externalapi.DomainBlockHeader, error) {
|
|
if header, ok := bms.staging[*blockHash]; ok {
|
|
return header, nil
|
|
}
|
|
|
|
headerBytes, err := dbContext.Get(bms.hashAsKey(blockHash))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return bms.deserializeHeader(headerBytes)
|
|
}
|
|
|
|
// HasBlock returns whether a block header with a given hash exists in the store.
|
|
func (bms *blockHeaderStore) HasBlockHeader(dbContext model.DBReader, blockHash *externalapi.DomainHash) (bool, error) {
|
|
if _, ok := bms.staging[*blockHash]; ok {
|
|
return true, nil
|
|
}
|
|
|
|
exists, err := dbContext.Has(bms.hashAsKey(blockHash))
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
|
|
return exists, nil
|
|
}
|
|
|
|
// BlockHeaders gets the block headers associated with the given blockHashes
|
|
func (bms *blockHeaderStore) BlockHeaders(dbContext model.DBReader, blockHashes []*externalapi.DomainHash) ([]*externalapi.DomainBlockHeader, error) {
|
|
headers := make([]*externalapi.DomainBlockHeader, len(blockHashes))
|
|
for i, hash := range blockHashes {
|
|
var err error
|
|
headers[i], err = bms.BlockHeader(dbContext, hash)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
return headers, nil
|
|
}
|
|
|
|
// Delete deletes the block associated with the given blockHash
|
|
func (bms *blockHeaderStore) Delete(dbTx model.DBTransaction, blockHash *externalapi.DomainHash) error {
|
|
if _, ok := bms.staging[*blockHash]; ok {
|
|
delete(bms.staging, *blockHash)
|
|
return nil
|
|
}
|
|
return dbTx.Delete(bms.hashAsKey(blockHash))
|
|
}
|
|
|
|
func (bms *blockHeaderStore) serializeHeader(header *externalapi.DomainBlockHeader) []byte {
|
|
panic("implement me")
|
|
}
|
|
|
|
func (bms *blockHeaderStore) deserializeHeader(headerBytes []byte) (*externalapi.DomainBlockHeader, error) {
|
|
panic("implement me")
|
|
}
|
|
|
|
func (bms *blockHeaderStore) hashAsKey(hash *externalapi.DomainHash) model.DBKey {
|
|
return bucket.Key(hash[:])
|
|
}
|