forked from TrueCloudLab/frostfs-node
134 lines
3.4 KiB
Go
134 lines
3.4 KiB
Go
package meta
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"time"
|
|
|
|
"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/local_object_storage/internal/metaerr"
|
|
"git.frostfs.info/TrueCloudLab/frostfs-observability/tracing"
|
|
oid "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/object/id"
|
|
"go.etcd.io/bbolt"
|
|
"go.opentelemetry.io/otel/attribute"
|
|
"go.opentelemetry.io/otel/trace"
|
|
)
|
|
|
|
// StorageIDPrm groups the parameters of StorageID operation.
|
|
type StorageIDPrm struct {
|
|
addr oid.Address
|
|
}
|
|
|
|
// StorageIDRes groups the resulting values of StorageID operation.
|
|
type StorageIDRes struct {
|
|
id []byte
|
|
}
|
|
|
|
// SetAddress is a StorageID option to set the object address to check.
|
|
func (p *StorageIDPrm) SetAddress(addr oid.Address) {
|
|
p.addr = addr
|
|
}
|
|
|
|
// StorageID returns storage ID.
|
|
func (r StorageIDRes) StorageID() []byte {
|
|
return r.id
|
|
}
|
|
|
|
// StorageID returns storage descriptor for objects from the blobstor.
|
|
// It is put together with the object can makes get/delete operation faster.
|
|
func (db *DB) StorageID(ctx context.Context, prm StorageIDPrm) (res StorageIDRes, err error) {
|
|
var (
|
|
startedAt = time.Now()
|
|
success = false
|
|
)
|
|
defer func() {
|
|
db.metrics.AddMethodDuration("StorageID", time.Since(startedAt), success)
|
|
}()
|
|
|
|
_, span := tracing.StartSpanFromContext(ctx, "metabase.StorageID",
|
|
trace.WithAttributes(
|
|
attribute.String("address", prm.addr.EncodeToString()),
|
|
))
|
|
defer span.End()
|
|
|
|
db.modeMtx.RLock()
|
|
defer db.modeMtx.RUnlock()
|
|
|
|
if db.mode.NoMetabase() {
|
|
return res, ErrDegradedMode
|
|
}
|
|
|
|
err = db.boltDB.View(func(tx *bbolt.Tx) error {
|
|
res.id, err = db.storageID(tx, prm.addr)
|
|
|
|
return err
|
|
})
|
|
success = err == nil
|
|
return res, metaerr.Wrap(err)
|
|
}
|
|
|
|
func (db *DB) storageID(tx *bbolt.Tx, addr oid.Address) ([]byte, error) {
|
|
key := make([]byte, bucketKeySize)
|
|
smallBucket := tx.Bucket(smallBucketName(addr.Container(), key))
|
|
if smallBucket == nil {
|
|
return nil, nil
|
|
}
|
|
|
|
storageID := smallBucket.Get(objectKey(addr.Object(), key))
|
|
if storageID == nil {
|
|
return nil, nil
|
|
}
|
|
|
|
return bytes.Clone(storageID), nil
|
|
}
|
|
|
|
// UpdateStorageIDPrm groups the parameters of UpdateStorageID operation.
|
|
type UpdateStorageIDPrm struct {
|
|
addr oid.Address
|
|
id []byte
|
|
}
|
|
|
|
// UpdateStorageIDRes groups the resulting values of UpdateStorageID operation.
|
|
type UpdateStorageIDRes struct{}
|
|
|
|
// SetAddress is an UpdateStorageID option to set the object address to check.
|
|
func (p *UpdateStorageIDPrm) SetAddress(addr oid.Address) {
|
|
p.addr = addr
|
|
}
|
|
|
|
// SetStorageID is an UpdateStorageID option to set the storage ID.
|
|
func (p *UpdateStorageIDPrm) SetStorageID(id []byte) {
|
|
p.id = id
|
|
}
|
|
|
|
// UpdateStorageID updates storage descriptor for objects from the blobstor.
|
|
func (db *DB) UpdateStorageID(ctx context.Context, prm UpdateStorageIDPrm) (res UpdateStorageIDRes, err error) {
|
|
var (
|
|
startedAt = time.Now()
|
|
success = false
|
|
)
|
|
defer func() {
|
|
db.metrics.AddMethodDuration("UpdateStorageID", time.Since(startedAt), success)
|
|
}()
|
|
|
|
_, span := tracing.StartSpanFromContext(ctx, "metabase.UpdateStorageID",
|
|
trace.WithAttributes(
|
|
attribute.String("address", prm.addr.EncodeToString()),
|
|
attribute.String("storage_id", string(prm.id)),
|
|
))
|
|
defer span.End()
|
|
|
|
db.modeMtx.RLock()
|
|
defer db.modeMtx.RUnlock()
|
|
|
|
if db.mode.NoMetabase() {
|
|
return res, ErrDegradedMode
|
|
} else if db.mode.ReadOnly() {
|
|
return res, ErrReadOnlyMode
|
|
}
|
|
|
|
err = db.boltDB.Batch(func(tx *bbolt.Tx) error {
|
|
return setStorageID(tx, prm.addr, prm.id, true)
|
|
})
|
|
success = err == nil
|
|
return res, metaerr.Wrap(err)
|
|
}
|