diff --git a/cmd/neofs-node/reputation/intermediate/consumers.go b/cmd/neofs-node/reputation/intermediate/consumers.go new file mode 100644 index 0000000000..9977454f3d --- /dev/null +++ b/cmd/neofs-node/reputation/intermediate/consumers.go @@ -0,0 +1,53 @@ +package intermediate + +import ( + "errors" + "fmt" + + "github.com/nspcc-dev/neofs-node/pkg/services/reputation" + "github.com/nspcc-dev/neofs-node/pkg/services/reputation/common" + reputationcommon "github.com/nspcc-dev/neofs-node/pkg/services/reputation/common" + "github.com/nspcc-dev/neofs-node/pkg/services/reputation/eigentrust" + eigencalc "github.com/nspcc-dev/neofs-node/pkg/services/reputation/eigentrust/calculator" + consumerstorage "github.com/nspcc-dev/neofs-node/pkg/services/reputation/eigentrust/storage/consumers" + "github.com/nspcc-dev/neofs-node/pkg/util/logger" +) + +var ErrIncorrectContext = errors.New("could not write intermediate trust: passed context incorrect") + +type ConsumerStorageWriterProvider struct { + Log *logger.Logger + Storage *consumerstorage.Storage +} + +type ConsumerTrustWriter struct { + log *logger.Logger + storage *consumerstorage.Storage +} + +func (w *ConsumerTrustWriter) Write(ctx common.Context, t reputation.Trust) error { + eiCtx, ok := ctx.(eigencalc.Context) + if !ok { + return ErrIncorrectContext + } + + trust := eigentrust.IterationTrust{Trust: t} + + trust.SetEpoch(eiCtx.Epoch()) + trust.SetI(eiCtx.I()) + + fmt.Println("decided to save consumers trusts to storage for epoch and iteration: ", eiCtx.Epoch(), eiCtx.I()) + w.storage.Put(trust) + return nil +} + +func (w *ConsumerTrustWriter) Close() error { + return nil +} + +func (s *ConsumerStorageWriterProvider) InitWriter(_ reputationcommon.Context) (reputationcommon.Writer, error) { + return &ConsumerTrustWriter{ + log: s.Log, + storage: s.Storage, + }, nil +} diff --git a/cmd/neofs-node/reputation/intermediate/daughters.go b/cmd/neofs-node/reputation/intermediate/daughters.go index 32018ce97f..271d31d3d9 100644 --- a/cmd/neofs-node/reputation/intermediate/daughters.go +++ b/cmd/neofs-node/reputation/intermediate/daughters.go @@ -8,7 +8,7 @@ import ( "github.com/nspcc-dev/neofs-node/pkg/util/logger" ) -type DaughterStorage struct { +type DaughterStorageWriterProvider struct { Log *logger.Logger Storage *daughters.Storage } @@ -27,7 +27,7 @@ func (w *DaughterTrustWriter) Close() error { return nil } -func (s *DaughterStorage) InitWriter(_ reputationcommon.Context) (reputationcommon.Writer, error) { +func (s *DaughterStorageWriterProvider) InitWriter(_ reputationcommon.Context) (reputationcommon.Writer, error) { return &DaughterTrustWriter{ log: s.Log, storage: s.Storage,