kaspad/infrastructure/db/dbaccess/acceptanceindex.go
stasatdaglabs d14809694f
[NOD-1223] Reorganize directory structure (#874)
* [NOD-1223] Delete unused files/packages.

* [NOD-1223] Move signal and limits to the os package.

* [NOD-1223] Put database and dbaccess into the db package.

* [NOD-1223] Fold the logs package into the logger package.

* [NOD-1223] Rename domainmessage to appmessage.

* [NOD-1223] Rename to/from DomainMessage to AppMessage.

* [NOD-1223] Move appmessage to the app packge.

* [NOD-1223] Move protocol to the app packge.

* [NOD-1223] Move the network package to the infrastructure packge.

* [NOD-1223] Rename cmd to executables.

* [NOD-1223] Fix go.doc in the logger package.
2020-08-18 10:26:39 +03:00

65 lines
1.7 KiB
Go

package dbaccess
import (
"github.com/kaspanet/kaspad/infrastructure/db/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)
}