mirror of
https://github.com/kaspanet/kaspad.git
synced 2025-06-03 12:46:43 +00:00
4207c82f5a
* Add prefix to stores * Add prefix to forgotten stores * Add a special type for prefix * Rename transaction->dbTx * Change error message * Use countKeyName * Rename Temporary Consesnsus to Staging * Add DeleteStagingConsensus to Domain interface * Add lock to staging consensus * Make prefix type-safer * Use ioutil.TempDir instead of t.TempDir
63 lines
1.9 KiB
Go
63 lines
1.9 KiB
Go
package finalitystore
|
|
|
|
import (
|
|
"github.com/kaspanet/kaspad/domain/consensus/database"
|
|
"github.com/kaspanet/kaspad/domain/consensus/model"
|
|
"github.com/kaspanet/kaspad/domain/consensus/model/externalapi"
|
|
"github.com/kaspanet/kaspad/domain/consensus/utils/lrucache"
|
|
"github.com/kaspanet/kaspad/domain/prefixmanager/prefix"
|
|
)
|
|
|
|
var bucketName = []byte("finality-points")
|
|
|
|
type finalityStore struct {
|
|
cache *lrucache.LRUCache
|
|
bucket model.DBBucket
|
|
}
|
|
|
|
// New instantiates a new FinalityStore
|
|
func New(prefix *prefix.Prefix, cacheSize int, preallocate bool) model.FinalityStore {
|
|
return &finalityStore{
|
|
cache: lrucache.New(cacheSize, preallocate),
|
|
bucket: database.MakeBucket(prefix.Serialize()).Bucket(bucketName),
|
|
}
|
|
}
|
|
|
|
func (fs *finalityStore) StageFinalityPoint(stagingArea *model.StagingArea, blockHash *externalapi.DomainHash, finalityPointHash *externalapi.DomainHash) {
|
|
stagingShard := fs.stagingShard(stagingArea)
|
|
|
|
stagingShard.toAdd[*blockHash] = finalityPointHash
|
|
}
|
|
|
|
func (fs *finalityStore) FinalityPoint(dbContext model.DBReader, stagingArea *model.StagingArea, blockHash *externalapi.DomainHash) (*externalapi.DomainHash, error) {
|
|
stagingShard := fs.stagingShard(stagingArea)
|
|
|
|
if finalityPointHash, ok := stagingShard.toAdd[*blockHash]; ok {
|
|
return finalityPointHash, nil
|
|
}
|
|
|
|
if finalityPointHash, ok := fs.cache.Get(blockHash); ok {
|
|
return finalityPointHash.(*externalapi.DomainHash), nil
|
|
}
|
|
|
|
finalityPointHashBytes, err := dbContext.Get(fs.hashAsKey(blockHash))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
finalityPointHash, err := externalapi.NewDomainHashFromByteSlice(finalityPointHashBytes)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
fs.cache.Add(blockHash, finalityPointHash)
|
|
return finalityPointHash, nil
|
|
}
|
|
|
|
func (fs *finalityStore) IsStaged(stagingArea *model.StagingArea) bool {
|
|
return fs.stagingShard(stagingArea).isStaged()
|
|
}
|
|
|
|
func (fs *finalityStore) hashAsKey(hash *externalapi.DomainHash) model.DBKey {
|
|
return fs.bucket.Key(hash.ByteSlice())
|
|
}
|