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 func(string, error) // reportError is the function called when encountering disk errors. metrics Metrics waitBeforeDropDB time.Duration blzInitWorkerCount int } type Option func(*cfg) const ( defaultPerm = 0o700 defaultOpenedCacheSize = 50 defaultBlzShallowDepth = 2 defaultBlzShallowWidth = 16 defaultWaitBeforeDropDB = 10 * time.Second defaultBlzInitWorkerCount = 5 ) 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, blzInitWorkerCount: defaultBlzInitWorkerCount, } } 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 } } // WithInitWorkerCount sets maximum workers count to init blobovnicza tree. // // Negative or zero value means no limit. func WithInitWorkerCount(v int) Option { if v <= 0 { v = -1 } return func(c *cfg) { c.blzInitWorkerCount = v } }