frostfs-node/pkg/services/object_manager/tombstone/constructor.go

package tombstone

import (
	"fmt"

	"github.com/TrueCloudLab/frostfs-node/pkg/util/logger"
	lru "github.com/hashicorp/golang-lru/v2"
	"go.uber.org/zap"
)

const defaultLRUCacheSize = 100

type cfg struct {
	log *logger.Logger

	cacheSize int

	tsSource Source
}

// Option is an option of ExpirationChecker's constructor.
type Option func(*cfg)

func defaultCfg() *cfg {
	return &cfg{
		log:       &logger.Logger{Logger: zap.NewNop()},
		cacheSize: defaultLRUCacheSize,
	}
}

// NewChecker creates, initializes and returns tombstone ExpirationChecker.
// The returned structure is ready to use.
//
// Panics if any of the provided options does not allow
// constructing a valid tombstone ExpirationChecker.
func NewChecker(oo ...Option) *ExpirationChecker {
	cfg := defaultCfg()

	for _, o := range oo {
		o(cfg)
	}

	panicOnNil := func(v any, name string) {
		if v == nil {
			panic(fmt.Sprintf("tombstone getter constructor: %s is nil", name))
		}
	}

	panicOnNil(cfg.tsSource, "Tombstone source")

	cache, err := lru.New[string, uint64](cfg.cacheSize)
	if err != nil {
		panic(fmt.Errorf("could not create LRU cache with %d size: %w", cfg.cacheSize, err))
	}

	return &ExpirationChecker{
		cache:    cache,
		log:      cfg.log,
		tsSource: cfg.tsSource,
	}
}

// WithLogger returns an option to specify
// logger.
func WithLogger(v *logger.Logger) Option {
	return func(c *cfg) {
		c.log = v
	}
}

// WithCacheSize returns an option to specify
// LRU cache size.
func WithCacheSize(v int) Option {
	return func(c *cfg) {
		c.cacheSize = v
	}
}

// WithTombstoneSource returns an option
// to specify tombstone source.
func WithTombstoneSource(v Source) Option {
	return func(c *cfg) {
		c.tsSource = v
	}
}