frostfs-node/pkg/local_object_storage/blobstor/blobovniczatree/option.go

package blobovniczatree

import (
	"io/fs"
	"time"

	"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/local_object_storage/blobovnicza"
	"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/local_object_storage/blobstor/compression"
	"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/util/logger"
	"go.uber.org/zap"
)

type cfg struct {
	log             *logger.Logger
	perm            fs.FileMode
	readOnly        bool
	rootPath        string
	openedCacheSize int
	blzShallowDepth uint64
	blzShallowWidth uint64
	blzLeafWidth    uint64
	compression     *compression.Config
	blzOpts         []blobovnicza.Option
	// reportError is the function called when encountering disk errors.
	reportError      func(string, error)
	metrics          Metrics
	waitBeforeDropDB time.Duration
}

type Option func(*cfg)

const (
	defaultPerm             = 0700
	defaultOpenedCacheSize  = 50
	defaultBlzShallowDepth  = 2
	defaultBlzShallowWidth  = 16
	defaultWaitBeforeDropDB = 10 * time.Second
)

func initConfig(c *cfg) {
	*c = cfg{
		log:              &logger.Logger{Logger: zap.L()},
		perm:             defaultPerm,
		openedCacheSize:  defaultOpenedCacheSize,
		blzShallowDepth:  defaultBlzShallowDepth,
		blzShallowWidth:  defaultBlzShallowWidth,
		reportError:      func(string, error) {},
		metrics:          &noopMetrics{},
		waitBeforeDropDB: defaultWaitBeforeDropDB,
	}
}

func WithLogger(l *logger.Logger) Option {
	return func(c *cfg) {
		c.log = l
		c.blzOpts = append(c.blzOpts, blobovnicza.WithLogger(l))
	}
}

func WithPermissions(perm fs.FileMode) Option {
	return func(c *cfg) {
		c.perm = perm
	}
}

func WithBlobovniczaShallowWidth(width uint64) Option {
	return func(c *cfg) {
		c.blzShallowWidth = width
	}
}

func WithBlobovniczaLeafWidth(w uint64) Option {
	return func(c *cfg) {
		c.blzLeafWidth = w
	}
}

func WithBlobovniczaShallowDepth(depth uint64) Option {
	return func(c *cfg) {
		c.blzShallowDepth = depth
	}
}

func WithRootPath(p string) Option {
	return func(c *cfg) {
		c.rootPath = p
	}
}

func WithBlobovniczaSize(sz uint64) Option {
	return func(c *cfg) {
		c.blzOpts = append(c.blzOpts, blobovnicza.WithFullSizeLimit(sz))
	}
}

func WithOpenedCacheSize(sz int) Option {
	return func(c *cfg) {
		c.openedCacheSize = sz
	}
}

func WithObjectSizeLimit(sz uint64) Option {
	return func(c *cfg) {
		c.blzOpts = append(c.blzOpts, blobovnicza.WithObjectSizeLimit(sz))
	}
}

func WithMetrics(m Metrics) Option {
	return func(c *cfg) {
		c.metrics = m
	}
}

func WithWaitBeforeDropDB(t time.Duration) Option {
	return func(c *cfg) {
		c.waitBeforeDropDB = t
	}
}