frostfs-node/pkg/services/replicator/replicator.go
Dmitrii Stepanov cbe9757490 [#1129] policer: Pull required EC chunks
Signed-off-by: Dmitrii Stepanov <d.stepanov@yadro.com>
2024-05-16 16:28:49 +03:00

93 lines
1.9 KiB
Go

package replicator
import (
"time"
"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/local_object_storage/engine"
getsvc "git.frostfs.info/TrueCloudLab/frostfs-node/pkg/services/object/get"
putsvc "git.frostfs.info/TrueCloudLab/frostfs-node/pkg/services/object/put"
"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/util/logger"
"go.uber.org/zap"
)
// Replicator represents the utility that replicates
// local objects to remote nodes.
type Replicator struct {
*cfg
}
// Option is an option for Policer constructor.
type Option func(*cfg)
type cfg struct {
putTimeout time.Duration
log *logger.Logger
remoteSender *putsvc.RemoteSender
remoteGetter *getsvc.RemoteGetter
localStorage *engine.StorageEngine
metrics MetricsRegister
}
func defaultCfg() *cfg {
return &cfg{}
}
// New creates, initializes and returns Replicator instance.
func New(opts ...Option) *Replicator {
c := defaultCfg()
for i := range opts {
opts[i](c)
}
c.log = &logger.Logger{Logger: c.log.With(zap.String("component", "Object Replicator"))}
return &Replicator{
cfg: c,
}
}
// WithPutTimeout returns option to set Put timeout of Replicator.
func WithPutTimeout(v time.Duration) Option {
return func(c *cfg) {
c.putTimeout = v
}
}
// WithLogger returns option to set Logger of Replicator.
func WithLogger(v *logger.Logger) Option {
return func(c *cfg) {
c.log = v
}
}
// WithRemoteSender returns option to set remote object sender of Replicator.
func WithRemoteSender(v *putsvc.RemoteSender) Option {
return func(c *cfg) {
c.remoteSender = v
}
}
func WithRemoteGetter(v *getsvc.RemoteGetter) Option {
return func(c *cfg) {
c.remoteGetter = v
}
}
// WithLocalStorage returns option to set local object storage of Replicator.
func WithLocalStorage(v *engine.StorageEngine) Option {
return func(c *cfg) {
c.localStorage = v
}
}
func WithMetrics(v MetricsRegister) Option {
return func(c *cfg) {
c.metrics = v
}
}