frostfs-node/pkg/local_object_storage/metabase/graveyard.go
Evgenii Stratonikov 95893927aa *: replace neofs-api-go with neofs-sdk-go
Signed-off-by: Evgenii Stratonikov <evgeniy@nspcc.ru>
2021-11-12 17:29:09 +03:00

78 lines
1.6 KiB
Go

package meta
import (
"bytes"
"errors"
"fmt"
"github.com/nspcc-dev/neofs-sdk-go/object"
"go.etcd.io/bbolt"
)
// Grave represents descriptor of DB's graveyard item.
type Grave struct {
gcMark bool
addr *object.Address
}
// WithGCMark returns true if grave marked for GC to be removed.
func (g *Grave) WithGCMark() bool {
return g.gcMark
}
// Address returns buried object address.
func (g *Grave) Address() *object.Address {
return g.addr
}
// GraveHandler is a Grave handling function.
type GraveHandler func(*Grave) error
// IterateOverGraveyard iterates over all graves in DB.
//
// If h returns ErrInterruptIterator, nil returns immediately.
// Returns other errors of h directly.
func (db *DB) IterateOverGraveyard(h GraveHandler) error {
return db.boltDB.View(func(tx *bbolt.Tx) error {
return db.iterateOverGraveyard(tx, h)
})
}
func (db *DB) iterateOverGraveyard(tx *bbolt.Tx, h GraveHandler) error {
// get graveyard bucket
bktGraveyard := tx.Bucket(graveyardBucketName)
if bktGraveyard == nil {
return nil
}
// iterate over all graves
err := bktGraveyard.ForEach(func(k, v []byte) error {
// parse Grave
g, err := graveFromKV(k, v)
if err != nil {
return fmt.Errorf("could not parse Grave: %w", err)
}
// handler Grave
return h(g)
})
if errors.Is(err, ErrInterruptIterator) {
err = nil
}
return err
}
func graveFromKV(k, v []byte) (*Grave, error) {
addr, err := addressFromKey(k)
if err != nil {
return nil, fmt.Errorf("could not parse address: %w", err)
}
return &Grave{
gcMark: bytes.Equal(v, []byte(inhumeGCMarkValue)),
addr: addr,
}, nil
}