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

package shard

import (
	"context"
	"errors"
	"sync"

	"git.frostfs.info/TrueCloudLab/frostfs-node/internal/logs"
	"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/local_object_storage/blobstor"
	meta "git.frostfs.info/TrueCloudLab/frostfs-node/pkg/local_object_storage/metabase"
	"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/util/logger"
	objectSDK "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/object"
	"go.uber.org/zap"
)

type rebuilder struct {
	mtx    *sync.Mutex
	wg     *sync.WaitGroup
	cancel func()
}

func newRebuilder() *rebuilder {
	return &rebuilder{
		mtx:    &sync.Mutex{},
		wg:     &sync.WaitGroup{},
		cancel: nil,
	}
}

func (r *rebuilder) Start(ctx context.Context, bs *blobstor.BlobStor, mb *meta.DB, log *logger.Logger) {
	r.mtx.Lock()
	defer r.mtx.Unlock()

	r.start(ctx, bs, mb, log)
}

func (r *rebuilder) start(ctx context.Context, bs *blobstor.BlobStor, mb *meta.DB, log *logger.Logger) {
	if r.cancel != nil {
		r.stop(log)
	}
	ctx, cancel := context.WithCancel(ctx)
	r.cancel = cancel
	r.wg.Add(1)
	go func() {
		defer r.wg.Done()

		log.Info(logs.BlobstoreRebuildStarted)
		if err := bs.Rebuild(ctx, &mbStorageIDUpdate{mb: mb}); err != nil {
			log.Warn(logs.FailedToRebuildBlobstore, zap.Error(err))
		} else {
			log.Info(logs.BlobstoreRebuildCompletedSuccessfully)
		}
	}()
}

func (r *rebuilder) Stop(log *logger.Logger) {
	r.mtx.Lock()
	defer r.mtx.Unlock()

	r.stop(log)
}

func (r *rebuilder) stop(log *logger.Logger) {
	if r.cancel == nil {
		return
	}

	r.cancel()
	r.wg.Wait()
	r.cancel = nil
	log.Info(logs.BlobstoreRebuildStopped)
}

var errMBIsNotAvailable = errors.New("metabase is not available")

type mbStorageIDUpdate struct {
	mb *meta.DB
}

func (u *mbStorageIDUpdate) UpdateStorageID(ctx context.Context, obj *objectSDK.Object, storageID []byte) error {
	if u.mb == nil {
		return errMBIsNotAvailable
	}
	var prm meta.PutPrm
	prm.SetObject(obj)
	prm.SetStorageID(storageID)
	_, err := u.mb.Put(ctx, prm)
	return err
}