mirror of
https://github.com/kaspanet/kaspad.git
synced 2025-05-24 07:46:45 +00:00
7402f3fb0e
* [NOD-1492] Implement some data stores * [NOD-1492] Remove pointers to acceptance data * [NOD-1492] Fix receiver names * [NOD-1492] Implement delete for acceptanceDataStore * [NOD-1492] In blockRelationStore rename IsAnythingStaged to IsStaged * [NOD-1492] Rename bucket name
82 lines
2.4 KiB
Go
82 lines
2.4 KiB
Go
package acceptancedatastore
|
|
|
|
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("acceptance-data"))
|
|
|
|
// acceptanceDataStore represents a store of AcceptanceData
|
|
type acceptanceDataStore struct {
|
|
staging map[externalapi.DomainHash]model.AcceptanceData
|
|
}
|
|
|
|
// New instantiates a new AcceptanceDataStore
|
|
func New() model.AcceptanceDataStore {
|
|
return &acceptanceDataStore{
|
|
staging: make(map[externalapi.DomainHash]model.AcceptanceData),
|
|
}
|
|
}
|
|
|
|
// Stage stages the given acceptanceData for the given blockHash
|
|
func (ads *acceptanceDataStore) Stage(blockHash *externalapi.DomainHash, acceptanceData model.AcceptanceData) {
|
|
ads.staging[*blockHash] = acceptanceData
|
|
}
|
|
|
|
func (ads *acceptanceDataStore) IsStaged() bool {
|
|
return len(ads.staging) != 0
|
|
}
|
|
|
|
func (ads *acceptanceDataStore) Discard() {
|
|
ads.staging = make(map[externalapi.DomainHash]model.AcceptanceData)
|
|
}
|
|
|
|
func (ads *acceptanceDataStore) Commit(dbTx model.DBTransaction) error {
|
|
for hash, acceptanceData := range ads.staging {
|
|
err := dbTx.Put(ads.hashAsKey(&hash), ads.serializeAcceptanceData(acceptanceData))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
ads.Discard()
|
|
return nil
|
|
}
|
|
|
|
// Get gets the acceptanceData associated with the given blockHash
|
|
func (ads *acceptanceDataStore) Get(dbContext model.DBReader, blockHash *externalapi.DomainHash) (model.AcceptanceData, error) {
|
|
if acceptanceData, ok := ads.staging[*blockHash]; ok {
|
|
return acceptanceData, nil
|
|
}
|
|
|
|
acceptanceDataBytes, err := dbContext.Get(ads.hashAsKey(blockHash))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return ads.deserializeAcceptanceData(acceptanceDataBytes)
|
|
}
|
|
|
|
// Delete deletes the acceptanceData associated with the given blockHash
|
|
func (ads *acceptanceDataStore) Delete(dbTx model.DBTransaction, blockHash *externalapi.DomainHash) error {
|
|
if _, ok := ads.staging[*blockHash]; ok {
|
|
delete(ads.staging, *blockHash)
|
|
return nil
|
|
}
|
|
return dbTx.Delete(ads.hashAsKey(blockHash))
|
|
}
|
|
|
|
func (ads *acceptanceDataStore) serializeAcceptanceData(acceptanceData model.AcceptanceData) []byte {
|
|
panic("implement me")
|
|
}
|
|
|
|
func (ads *acceptanceDataStore) deserializeAcceptanceData(acceptanceDataBytes []byte) (model.AcceptanceData, error) {
|
|
panic("implement me")
|
|
}
|
|
|
|
func (ads *acceptanceDataStore) hashAsKey(hash *externalapi.DomainHash) model.DBKey {
|
|
return bucket.Key(hash[:])
|
|
}
|