frostfs-node/pkg/local_object_storage/blobstor/control.go

package blobstor

import (
	"context"
	"errors"
	"fmt"

	"git.frostfs.info/TrueCloudLab/frostfs-node/internal/logs"
	"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/local_object_storage/blobstor/compression"
	"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/local_object_storage/shard/mode"
	"go.uber.org/zap"
)

// Open opens BlobStor.
func (b *BlobStor) Open(ctx context.Context, mode mode.Mode) error {
	b.log.Debug(ctx, logs.BlobstorOpening)

	b.modeMtx.Lock()
	defer b.modeMtx.Unlock()
	b.mode = mode

	err := b.openBlobStor(ctx, mode)
	if err != nil {
		return err
	}
	b.metrics.SetMode(mode.ReadOnly())

	return nil
}

func (b *BlobStor) openBlobStor(ctx context.Context, mod mode.Mode) error {
	for i := range b.storage {
		select {
		case <-ctx.Done():
			return ctx.Err()
		default:
		}
		err := b.storage[i].Storage.Open(mode.ConvertToComponentMode(mod))
		if err != nil {
			return err
		}
	}
	return nil
}

// ErrInitBlobovniczas is returned when blobovnicza initialization fails.
var ErrInitBlobovniczas = errors.New("failure on blobovnicza initialization stage")

// Init initializes internal data structures and system resources.
//
// If BlobStor is already initialized, no action is taken.
//
// Returns wrapped ErrInitBlobovniczas on blobovnicza tree's initializaiton failure.
func (b *BlobStor) Init(ctx context.Context) error {
	b.log.Debug(ctx, logs.BlobstorInitializing)

	if !b.compression.HasValidCompressionLevel() {
		b.log.Warn(ctx, logs.UnknownCompressionLevelDefaultWillBeUsed, zap.String("level", string(b.compression.Level)))
		b.compression.Level = compression.LevelDefault
	}
	if err := b.compression.Init(); err != nil {
		return err
	}

	for i := range b.storage {
		err := b.storage[i].Storage.Init()
		if err != nil {
			return fmt.Errorf("%w: %v", ErrInitBlobovniczas, err)
		}
	}
	return nil
}

// Close releases all internal resources of BlobStor.
func (b *BlobStor) Close(ctx context.Context) error {
	b.log.Debug(ctx, logs.BlobstorClosing)

	var firstErr error
	for i := range b.storage {
		err := b.storage[i].Storage.Close(ctx)
		if err != nil {
			b.log.Info(ctx, logs.BlobstorCouldntCloseStorage, zap.Error(err))
			if firstErr == nil {
				firstErr = err
			}
			continue
		}
	}

	err := b.compression.Close()
	if firstErr == nil {
		firstErr = err
	}
	if firstErr == nil {
		b.metrics.Close()
	}
	return firstErr
}