mirror of
https://github.com/kaspanet/kaspad.git
synced 2025-05-23 07:16:47 +00:00
ad9c213a06
* Add TestValidateAndInsertPruningPointWithSideBlocks * Optimize infrastracture bucket paths * Update infrastracture tests * Refactor the consensus/database layer * Remove utils/dbkeys * Use consensus/database in consensus instead of infrastructure * Fix a bug in dbBucketToDatabaseBucket and MakeBucket combination Co-authored-by: Elichai Turkel <elichai.turkel@gmail.com> Co-authored-by: stasatdaglabs <39559713+stasatdaglabs@users.noreply.github.com>
101 lines
2.4 KiB
Go
101 lines
2.4 KiB
Go
package headersselectedtipstore
|
|
|
|
import (
|
|
"github.com/golang/protobuf/proto"
|
|
"github.com/kaspanet/kaspad/domain/consensus/database"
|
|
"github.com/kaspanet/kaspad/domain/consensus/database/serialization"
|
|
"github.com/kaspanet/kaspad/domain/consensus/model"
|
|
"github.com/kaspanet/kaspad/domain/consensus/model/externalapi"
|
|
)
|
|
|
|
var headerSelectedTipKey = database.MakeBucket(nil).Key([]byte("headers-selected-tip"))
|
|
|
|
type headerSelectedTipStore struct {
|
|
staging *externalapi.DomainHash
|
|
cache *externalapi.DomainHash
|
|
}
|
|
|
|
// New instantiates a new HeaderSelectedTipStore
|
|
func New() model.HeaderSelectedTipStore {
|
|
return &headerSelectedTipStore{}
|
|
}
|
|
|
|
func (hts *headerSelectedTipStore) Has(dbContext model.DBReader) (bool, error) {
|
|
if hts.staging != nil {
|
|
return true, nil
|
|
}
|
|
|
|
if hts.cache != nil {
|
|
return true, nil
|
|
}
|
|
|
|
return dbContext.Has(headerSelectedTipKey)
|
|
}
|
|
|
|
func (hts *headerSelectedTipStore) Discard() {
|
|
hts.staging = nil
|
|
}
|
|
|
|
func (hts *headerSelectedTipStore) Commit(dbTx model.DBTransaction) error {
|
|
if hts.staging == nil {
|
|
return nil
|
|
}
|
|
|
|
selectedTipBytes, err := hts.serializeHeadersSelectedTip(hts.staging)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
err = dbTx.Put(headerSelectedTipKey, selectedTipBytes)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
hts.cache = hts.staging
|
|
|
|
hts.Discard()
|
|
return nil
|
|
}
|
|
|
|
func (hts *headerSelectedTipStore) Stage(selectedTip *externalapi.DomainHash) {
|
|
hts.staging = selectedTip
|
|
}
|
|
|
|
func (hts *headerSelectedTipStore) IsStaged() bool {
|
|
return hts.staging != nil
|
|
}
|
|
|
|
func (hts *headerSelectedTipStore) HeadersSelectedTip(dbContext model.DBReader) (*externalapi.DomainHash, error) {
|
|
if hts.staging != nil {
|
|
return hts.staging, nil
|
|
}
|
|
|
|
if hts.cache != nil {
|
|
return hts.cache, nil
|
|
}
|
|
|
|
selectedTipBytes, err := dbContext.Get(headerSelectedTipKey)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
selectedTip, err := hts.deserializeHeadersSelectedTip(selectedTipBytes)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
hts.cache = selectedTip
|
|
return hts.cache, nil
|
|
}
|
|
|
|
func (hts *headerSelectedTipStore) serializeHeadersSelectedTip(selectedTip *externalapi.DomainHash) ([]byte, error) {
|
|
return proto.Marshal(serialization.DomainHashToDbHash(selectedTip))
|
|
}
|
|
|
|
func (hts *headerSelectedTipStore) deserializeHeadersSelectedTip(selectedTipBytes []byte) (*externalapi.DomainHash, error) {
|
|
dbHash := &serialization.DbHash{}
|
|
err := proto.Unmarshal(selectedTipBytes, dbHash)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return serialization.DbHashToDomainHash(dbHash)
|
|
}
|