frostfs-node/pkg/local_object_storage/shard/exists.go

package shard

import (
	"context"

	"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/local_object_storage/blobstor/common"
	meta "git.frostfs.info/TrueCloudLab/frostfs-node/pkg/local_object_storage/metabase"
	"git.frostfs.info/TrueCloudLab/frostfs-observability/tracing"
	oid "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/object/id"
	"go.opentelemetry.io/otel/attribute"
	"go.opentelemetry.io/otel/trace"
)

// ExistsPrm groups the parameters of Exists operation.
type ExistsPrm struct {
	addr  oid.Address
	paddr oid.Address
}

// ExistsRes groups the resulting values of Exists operation.
type ExistsRes struct {
	ex bool
	lc bool
}

// SetAddress is an Exists option to set object checked for existence.
func (p *ExistsPrm) SetAddress(addr oid.Address) {
	p.addr = addr
}

// SetParentAddress is an Exists option to set parent object checked for existence.
func (p *ExistsPrm) SetParentAddress(addr oid.Address) {
	p.paddr = addr
}

// Exists returns the fact that the object is in the shard.
func (p ExistsRes) Exists() bool {
	return p.ex
}

// Locked returns the fact that the object is locked.
func (p ExistsRes) Locked() bool {
	return p.lc
}

// Exists checks if object is presented in shard.
//
// Returns any error encountered that does not allow to
// unambiguously determine the presence of an object.
//
// Returns an error of type apistatus.ObjectAlreadyRemoved if object has been marked as removed.
// Returns the object.ErrObjectIsExpired if the object is presented but already expired.
// Returns the ErrShardDisabled if the shard is disabled.
func (s *Shard) Exists(ctx context.Context, prm ExistsPrm) (ExistsRes, error) {
	ctx, span := tracing.StartSpanFromContext(ctx, "Shard.Exists",
		trace.WithAttributes(
			attribute.String("shard_id", s.ID().String()),
			attribute.String("address", prm.addr.EncodeToString()),
		))
	defer span.End()

	var exists bool
	var locked bool
	var err error

	s.m.RLock()
	defer s.m.RUnlock()

	if s.info.Mode.Disabled() {
		return ExistsRes{}, ErrShardDisabled
	} else if s.info.Mode.NoMetabase() {
		var p common.ExistsPrm
		p.Address = prm.addr

		var res common.ExistsRes
		res, err = s.blobStor.Exists(ctx, p)
		exists = res.Exists
	} else {
		var existsPrm meta.ExistsPrm
		existsPrm.SetAddress(prm.addr)
		existsPrm.SetParent(prm.paddr)

		var res meta.ExistsRes
		res, err = s.metaBase.Exists(ctx, existsPrm)
		exists = res.Exists()
		locked = res.Locked()
	}

	return ExistsRes{
		ex: exists,
		lc: locked,
	}, err
}