frostfs-node/pkg/services/object_manager/tombstone/constructor.go
Pavel Karpy 923f84722a Move to frostfs-node
Signed-off-by: Pavel Karpy <p.karpy@yadro.com>
2022-12-28 15:04:29 +03:00

85 lines
1.7 KiB
Go

package tombstone
import (
"fmt"
"github.com/TrueCloudLab/frostfs-node/pkg/util/logger"
lru "github.com/hashicorp/golang-lru"
"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 interface{}, name string) {
if v == nil {
panic(fmt.Sprintf("tombstone getter constructor: %s is nil", name))
}
}
panicOnNil(cfg.tsSource, "Tombstone source")
cache, err := lru.New(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
}
}