frostfs-node/pkg/local_object_storage/blobovnicza/move.go

package blobovnicza

import (
	"context"
	"fmt"

	"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"
)

var incompletedMoveBucketName = []byte("INCOMPLETED_MOVE")

type MoveInfo struct {
	Address         oid.Address
	TargetStorageID []byte
}

func (b *Blobovnicza) PutMoveInfo(ctx context.Context, prm MoveInfo) error {
	_, span := tracing.StartSpanFromContext(ctx, "Blobovnicza.PutMoveInfo",
		trace.WithAttributes(
			attribute.String("path", b.path),
			attribute.String("address", prm.Address.EncodeToString()),
			attribute.String("target_storage_id", string(prm.TargetStorageID)),
		))
	defer span.End()

	key := addressKey(prm.Address)

	return b.boltDB.Update(func(tx *bbolt.Tx) error {
		bucket, err := tx.CreateBucketIfNotExists(incompletedMoveBucketName)
		if err != nil {
			return err
		}

		if err := bucket.Put(key, prm.TargetStorageID); err != nil {
			return fmt.Errorf("(%T) failed to save move info: %w", b, err)
		}

		return nil
	})
}

func (b *Blobovnicza) DropMoveInfo(ctx context.Context, address oid.Address) error {
	_, span := tracing.StartSpanFromContext(ctx, "Blobovnicza.DropMoveInfo",
		trace.WithAttributes(
			attribute.String("path", b.path),
			attribute.String("address", address.EncodeToString()),
		))
	defer span.End()

	key := addressKey(address)

	return b.boltDB.Update(func(tx *bbolt.Tx) error {
		bucket := tx.Bucket(incompletedMoveBucketName)
		if bucket == nil {
			return nil
		}

		if err := bucket.Delete(key); err != nil {
			return fmt.Errorf("(%T) failed to drop move info: %w", b, err)
		}

		c := bucket.Cursor()
		k, v := c.First()
		bucketEmpty := k == nil && v == nil
		if bucketEmpty {
			return tx.DeleteBucket(incompletedMoveBucketName)
		}

		return nil
	})
}

func (b *Blobovnicza) ListMoveInfo(ctx context.Context) ([]MoveInfo, error) {
	_, span := tracing.StartSpanFromContext(ctx, "Blobovnicza.ListMoveInfo",
		trace.WithAttributes(
			attribute.String("path", b.path),
		))
	defer span.End()

	var result []MoveInfo
	if err := b.boltDB.View(func(tx *bbolt.Tx) error {
		bucket := tx.Bucket(incompletedMoveBucketName)
		if bucket == nil {
			return nil
		}
		return bucket.ForEach(func(k, v []byte) error {
			var addr oid.Address
			storageID := make([]byte, len(v))
			if err := addressFromKey(&addr, k); err != nil {
				return err
			}
			copy(storageID, v)
			result = append(result, MoveInfo{
				Address:         addr,
				TargetStorageID: storageID,
			})
			return nil
		})
	}); err != nil {
		return nil, err
	}

	return result, nil
}