kaspad/infrastructure/dbaccess/acceptanceindex.go
stasatdaglabs 8a4ece1101
[NOD-1223] Reorganize project (#868)
* [NOD-1223] Move all network stuff into a new network package.

* [NOD-1223] Delete the unused package testutil.

* [NOD-1223] Move infrastructure stuff into a new instrastructure package.

* [NOD-1223] Move domain stuff into a new domain package.
2020-08-13 17:27:25 +03:00

65 lines
1.7 KiB
Go

package dbaccess
import (
"github.com/kaspanet/kaspad/infrastructure/database"
"github.com/kaspanet/kaspad/util/daghash"
"github.com/pkg/errors"
)
var (
acceptanceIndexBucket = database.MakeBucket([]byte("acceptance-index"))
)
func acceptanceIndexKey(hash *daghash.Hash) *database.Key {
return acceptanceIndexBucket.Key(hash[:])
}
// StoreAcceptanceData stores the given acceptanceData in the database.
func StoreAcceptanceData(context Context, hash *daghash.Hash, acceptanceData []byte) error {
accessor, err := context.accessor()
if err != nil {
return err
}
key := acceptanceIndexKey(hash)
return accessor.Put(key, acceptanceData)
}
// HasAcceptanceData returns whether the acceptanceData of the given hash
// has been previously inserted into the database.
func HasAcceptanceData(context Context, hash *daghash.Hash) (bool, error) {
accessor, err := context.accessor()
if err != nil {
return false, err
}
key := acceptanceIndexKey(hash)
return accessor.Has(key)
}
// FetchAcceptanceData returns the acceptanceData of the given hash.
// Returns ErrNotFound if the acceptanceData had not been previously
// inserted into the database.
func FetchAcceptanceData(context Context, hash *daghash.Hash) ([]byte, error) {
accessor, err := context.accessor()
if err != nil {
return nil, err
}
key := acceptanceIndexKey(hash)
acceptanceData, err := accessor.Get(key)
if err != nil {
if database.IsNotFoundError(err) {
return nil, errors.Wrapf(err, "acceptance data not found for hash %s", hash)
}
return nil, err
}
return acceptanceData, nil
}
// DropAcceptanceIndex completely removes all acceptanceData entries.
func DropAcceptanceIndex(dbTx *TxContext) error {
return clearBucket(dbTx, acceptanceIndexBucket)
}