frostfs-node/pkg/local_object_storage/metabase/counter.go

100 lines
2.1 KiB
Go
Raw Normal View History

package meta
import (
"encoding/binary"
"fmt"
cid "github.com/nspcc-dev/neofs-sdk-go/container/id"
oid "github.com/nspcc-dev/neofs-sdk-go/object/id"
"go.etcd.io/bbolt"
)
var objectCounterKey = []byte("phy_counter")
// ObjectCounter returns object count that metabase has
// tracked since it was opened and initialized.
//
// Returns only the errors that do not allow reading counter
// in Bolt database.
func (db *DB) ObjectCounter() (counter uint64, err error) {
err = db.boltDB.View(func(tx *bbolt.Tx) error {
b := tx.Bucket(shardInfoBucket)
if b != nil {
data := b.Get(objectCounterKey)
if len(data) == 8 {
counter = binary.LittleEndian.Uint64(data)
}
}
return nil
})
return
}
// updateCounter updates the object counter. Tx MUST be writable.
// If inc == `true`, increases the counter, decreases otherwise.
func (db *DB) updateCounter(tx *bbolt.Tx, delta uint64, inc bool) error {
b := tx.Bucket(shardInfoBucket)
if b == nil {
return nil
}
var counter uint64
data := b.Get(objectCounterKey)
if len(data) == 8 {
counter = binary.LittleEndian.Uint64(data)
}
if inc {
counter += delta
} else if counter <= delta {
counter = 0
} else {
counter -= delta
}
newCounter := make([]byte, 8)
binary.LittleEndian.PutUint64(newCounter, counter)
return b.Put(objectCounterKey, newCounter)
}
// syncCounter updates object counter according to metabase state:
// it counts all the physically stored objects using internal indexes.
// Tx MUST be writable.
//
// Does nothing if counter not empty.
func syncCounter(tx *bbolt.Tx) error {
var counter uint64
b, err := tx.CreateBucketIfNotExists(shardInfoBucket)
if err != nil {
return fmt.Errorf("could not get shard info bucket: %w", err)
}
data := b.Get(objectCounterKey)
if len(data) == 8 {
return nil
}
err = iteratePhyObjects(tx, func(_ cid.ID, _ oid.ID) error {
counter++
return nil
})
if err != nil {
return fmt.Errorf("count not iterate objects: %w", err)
}
data = make([]byte, 8)
binary.LittleEndian.PutUint64(data, counter)
err = b.Put(objectCounterKey, data)
if err != nil {
return fmt.Errorf("could not update object counter: %w", err)
}
return nil
}