111 lines
3.1 KiB
Go
111 lines
3.1 KiB
Go
package shard
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
meta "git.frostfs.info/TrueCloudLab/frostfs-node/pkg/local_object_storage/metabase"
|
|
"git.frostfs.info/TrueCloudLab/frostfs-observability/tracing"
|
|
cid "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/container/id"
|
|
oid "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/object/id"
|
|
"go.opentelemetry.io/otel/attribute"
|
|
"go.opentelemetry.io/otel/trace"
|
|
)
|
|
|
|
// Lock marks objects as locked with another object. All objects from the
|
|
// specified container.
|
|
//
|
|
// Allows locking regular objects only (otherwise returns apistatus.LockNonRegularObject).
|
|
//
|
|
// Locked list should be unique. Panics if it is empty.
|
|
func (s *Shard) Lock(ctx context.Context, idCnr cid.ID, locker oid.ID, locked []oid.ID) error {
|
|
ctx, span := tracing.StartSpanFromContext(ctx, "Shard.Lock",
|
|
trace.WithAttributes(
|
|
attribute.String("shard_id", s.ID().String()),
|
|
attribute.String("container_id", idCnr.EncodeToString()),
|
|
attribute.String("locker", locker.EncodeToString()),
|
|
attribute.Int("locked_count", len(locked)),
|
|
))
|
|
defer span.End()
|
|
|
|
s.m.RLock()
|
|
defer s.m.RUnlock()
|
|
|
|
m := s.info.Mode
|
|
if m.ReadOnly() {
|
|
return ErrReadOnlyMode
|
|
} else if m.NoMetabase() {
|
|
return ErrDegradedMode
|
|
}
|
|
|
|
err := s.metaBase.Lock(ctx, idCnr, locker, locked)
|
|
if err != nil {
|
|
return fmt.Errorf("metabase lock: %w", err)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// IsLocked checks object locking relation of the provided object. Not found object is
|
|
// considered as not locked. Requires healthy metabase, returns ErrDegradedMode otherwise.
|
|
func (s *Shard) IsLocked(ctx context.Context, addr oid.Address) (bool, error) {
|
|
ctx, span := tracing.StartSpanFromContext(ctx, "Shard.IsLocked",
|
|
trace.WithAttributes(
|
|
attribute.String("shard_id", s.ID().String()),
|
|
attribute.String("address", addr.EncodeToString()),
|
|
))
|
|
defer span.End()
|
|
|
|
m := s.GetMode()
|
|
if m.NoMetabase() {
|
|
return false, ErrDegradedMode
|
|
}
|
|
|
|
var prm meta.IsLockedPrm
|
|
prm.SetAddress(addr)
|
|
|
|
res, err := s.metaBase.IsLocked(ctx, prm)
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
|
|
return res.Locked(), nil
|
|
}
|
|
|
|
// GetLocked return lock id's of the provided object. Not found object is
|
|
// considered as not locked. Requires healthy metabase, returns ErrDegradedMode otherwise.
|
|
func (s *Shard) GetLocked(ctx context.Context, addr oid.Address) ([]oid.ID, error) {
|
|
ctx, span := tracing.StartSpanFromContext(ctx, "Shard.GetLocked",
|
|
trace.WithAttributes(
|
|
attribute.String("shard_id", s.ID().String()),
|
|
attribute.String("address", addr.EncodeToString()),
|
|
))
|
|
defer span.End()
|
|
|
|
m := s.GetMode()
|
|
if m.NoMetabase() {
|
|
return nil, ErrDegradedMode
|
|
}
|
|
return s.metaBase.GetLocked(ctx, addr)
|
|
}
|
|
|
|
func (s *Shard) RemoveOrphanLocks(
|
|
ctx context.Context, locked oid.Address,
|
|
dropOrphanLocks func(ctx context.Context, cnt cid.ID, locks []oid.ID) (locksToKeep []oid.ID),
|
|
) error {
|
|
_, span := tracing.StartSpanFromContext(ctx, "Shard.RemoveOrphanLocks",
|
|
trace.WithAttributes(
|
|
attribute.String("address", locked.EncodeToString()),
|
|
))
|
|
defer span.End()
|
|
|
|
switch m := s.GetMode(); {
|
|
case m.NoMetabase():
|
|
return ErrDegradedMode
|
|
case m.ReadOnly():
|
|
return ErrReadOnlyMode
|
|
default:
|
|
}
|
|
|
|
return s.metaBase.RemoveOrphanLocks(ctx, locked, dropOrphanLocks)
|
|
}
|