forked from TrueCloudLab/frostfs-node
93 lines
1.9 KiB
Go
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
|
|
}
|
|
}
|