From 16a5107ef1007ac2bdf24b7fe8314ee0bd0fef74 Mon Sep 17 00:00:00 2001 From: Leonard Lyubich Date: Wed, 21 Oct 2020 18:11:50 +0300 Subject: [PATCH] [#60] object/put: Provide network State interface to formatter Signed-off-by: Leonard Lyubich --- pkg/services/object/put/service.go | 8 ++++++++ pkg/services/object/put/streamer.go | 1 + 2 files changed, 9 insertions(+) diff --git a/pkg/services/object/put/service.go b/pkg/services/object/put/service.go index 74b89b79..bc8b6945 100644 --- a/pkg/services/object/put/service.go +++ b/pkg/services/object/put/service.go @@ -40,6 +40,8 @@ type cfg struct { fmtValidator *object.FormatValidator fmtValidatorOpts []object.FormatValidatorOption + + networkState netmap.State } func defaultCfg() *cfg { @@ -116,3 +118,9 @@ func WithFormatValidatorOpts(v ...object.FormatValidatorOption) Option { c.fmtValidatorOpts = v } } + +func WithNetworkState(v netmap.State) Option { + return func(c *cfg) { + c.networkState = v + } +} diff --git a/pkg/services/object/put/streamer.go b/pkg/services/object/put/streamer.go index 3fe7c946..7b798642 100644 --- a/pkg/services/object/put/streamer.go +++ b/pkg/services/object/put/streamer.go @@ -73,6 +73,7 @@ func (p *Streamer) initTarget(prm *PutInitPrm) error { Key: sessionKey, NextTarget: p.newCommonTarget(prm), SessionToken: sToken, + NetworkState: p.networkState, }) }, )