mirror of
https://github.com/kaspanet/kaspad.git
synced 2025-05-23 07:16:47 +00:00
05941a76e7
* Increase size of reachability cache * Change DomainHash to struct with unexported hashArray * Fixing compilation errors stemming from new DomainHash structure * Remove obsolete Read/WriteElement methods in appmessage * Fix all tests * Fix all tests * Add comments * A few renamings * go mod tidy
108 lines
3.2 KiB
Go
108 lines
3.2 KiB
Go
package blockrelationstore
|
|
|
|
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"
|
|
"github.com/kaspanet/kaspad/domain/consensus/utils/lrucache"
|
|
)
|
|
|
|
var bucket = dbkeys.MakeBucket([]byte("block-relations"))
|
|
|
|
// blockRelationStore represents a store of BlockRelations
|
|
type blockRelationStore struct {
|
|
staging map[externalapi.DomainHash]*model.BlockRelations
|
|
cache *lrucache.LRUCache
|
|
}
|
|
|
|
// New instantiates a new BlockRelationStore
|
|
func New(cacheSize int) model.BlockRelationStore {
|
|
return &blockRelationStore{
|
|
staging: make(map[externalapi.DomainHash]*model.BlockRelations),
|
|
cache: lrucache.New(cacheSize),
|
|
}
|
|
}
|
|
|
|
func (brs *blockRelationStore) StageBlockRelation(blockHash *externalapi.DomainHash, blockRelations *model.BlockRelations) {
|
|
brs.staging[*blockHash] = blockRelations.Clone()
|
|
}
|
|
|
|
func (brs *blockRelationStore) IsStaged() bool {
|
|
return len(brs.staging) != 0
|
|
}
|
|
|
|
func (brs *blockRelationStore) Discard() {
|
|
brs.staging = make(map[externalapi.DomainHash]*model.BlockRelations)
|
|
}
|
|
|
|
func (brs *blockRelationStore) Commit(dbTx model.DBTransaction) error {
|
|
for hash, blockRelations := range brs.staging {
|
|
blockRelationBytes, err := brs.serializeBlockRelations(blockRelations)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
err = dbTx.Put(brs.hashAsKey(&hash), blockRelationBytes)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
brs.cache.Add(&hash, blockRelations)
|
|
}
|
|
|
|
brs.Discard()
|
|
return nil
|
|
}
|
|
|
|
func (brs *blockRelationStore) BlockRelation(dbContext model.DBReader, blockHash *externalapi.DomainHash) (*model.BlockRelations, error) {
|
|
if blockRelations, ok := brs.staging[*blockHash]; ok {
|
|
return blockRelations.Clone(), nil
|
|
}
|
|
|
|
if blockRelations, ok := brs.cache.Get(blockHash); ok {
|
|
return blockRelations.(*model.BlockRelations).Clone(), nil
|
|
}
|
|
|
|
blockRelationsBytes, err := dbContext.Get(brs.hashAsKey(blockHash))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
blockRelations, err := brs.deserializeBlockRelations(blockRelationsBytes)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
brs.cache.Add(blockHash, blockRelations)
|
|
return blockRelations.Clone(), nil
|
|
}
|
|
|
|
func (brs *blockRelationStore) Has(dbContext model.DBReader, blockHash *externalapi.DomainHash) (bool, error) {
|
|
if _, ok := brs.staging[*blockHash]; ok {
|
|
return true, nil
|
|
}
|
|
|
|
if brs.cache.Has(blockHash) {
|
|
return true, nil
|
|
}
|
|
|
|
return dbContext.Has(brs.hashAsKey(blockHash))
|
|
}
|
|
|
|
func (brs *blockRelationStore) hashAsKey(hash *externalapi.DomainHash) model.DBKey {
|
|
return bucket.Key(hash.ByteSlice())
|
|
}
|
|
|
|
func (brs *blockRelationStore) serializeBlockRelations(blockRelations *model.BlockRelations) ([]byte, error) {
|
|
dbBlockRelations := serialization.DomainBlockRelationsToDbBlockRelations(blockRelations)
|
|
return proto.Marshal(dbBlockRelations)
|
|
}
|
|
|
|
func (brs *blockRelationStore) deserializeBlockRelations(blockRelationsBytes []byte) (*model.BlockRelations, error) {
|
|
dbBlockRelations := &serialization.DbBlockRelations{}
|
|
err := proto.Unmarshal(blockRelationsBytes, dbBlockRelations)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return serialization.DbBlockRelationsToDomainBlockRelations(dbBlockRelations)
|
|
}
|