package putsvc import ( "context" "git.frostfs.info/TrueCloudLab/frostfs-node/pkg/core/client" "git.frostfs.info/TrueCloudLab/frostfs-node/pkg/core/container" "git.frostfs.info/TrueCloudLab/frostfs-node/pkg/core/netmap" "git.frostfs.info/TrueCloudLab/frostfs-node/pkg/core/object" objutil "git.frostfs.info/TrueCloudLab/frostfs-node/pkg/services/object/util" "git.frostfs.info/TrueCloudLab/frostfs-node/pkg/util" "git.frostfs.info/TrueCloudLab/frostfs-node/pkg/util/logger" "go.uber.org/zap" ) type MaxSizeSource interface { // MaxObjectSize returns maximum payload size // of physically stored object in system. // // Must return 0 if value can not be obtained. MaxObjectSize() uint64 } type Service struct { *cfg } type Option func(*cfg) type ClientConstructor interface { Get(client.NodeInfo) (client.MultiAddressClient, error) } type cfg struct { keyStorage *objutil.KeyStorage maxSizeSrc MaxSizeSource localStore ObjectStorage cnrSrc container.Source netMapSrc netmap.Source remotePool, localPool util.WorkerPool netmapKeys netmap.AnnouncedKeys fmtValidator *object.FormatValidator fmtValidatorOpts []object.FormatValidatorOption networkState netmap.State clientConstructor ClientConstructor log *logger.Logger } func defaultCfg() *cfg { return &cfg{ remotePool: util.NewPseudoWorkerPool(), localPool: util.NewPseudoWorkerPool(), log: &logger.Logger{Logger: zap.L()}, } } func NewService(opts ...Option) *Service { c := defaultCfg() for i := range opts { opts[i](c) } c.fmtValidator = object.NewFormatValidator(c.fmtValidatorOpts...) return &Service{ cfg: c, } } func (p *Service) Put(ctx context.Context) (*Streamer, error) { return &Streamer{ cfg: p.cfg, ctx: ctx, }, nil } func WithKeyStorage(v *objutil.KeyStorage) Option { return func(c *cfg) { c.keyStorage = v } } func WithMaxSizeSource(v MaxSizeSource) Option { return func(c *cfg) { c.maxSizeSrc = v } } func WithObjectStorage(v ObjectStorage) Option { return func(c *cfg) { c.localStore = v c.fmtValidatorOpts = append(c.fmtValidatorOpts, object.WithLockSource(v)) } } func WithContainerSource(v container.Source) Option { return func(c *cfg) { c.cnrSrc = v } } func WithNetworkMapSource(v netmap.Source) Option { return func(c *cfg) { c.netMapSrc = v } } func WithWorkerPools(remote, local util.WorkerPool) Option { return func(c *cfg) { c.remotePool, c.localPool = remote, local } } func WithNetmapKeys(v netmap.AnnouncedKeys) Option { return func(c *cfg) { c.netmapKeys = v } } func WithNetworkState(v netmap.State) Option { return func(c *cfg) { c.networkState = v c.fmtValidatorOpts = append(c.fmtValidatorOpts, object.WithNetState(v)) } } func WithClientConstructor(v ClientConstructor) Option { return func(c *cfg) { c.clientConstructor = v } } func WithLogger(l *logger.Logger) Option { return func(c *cfg) { c.log = l } }