forked from TrueCloudLab/frostfs-node
[#488] reputation/intermediate: Add implementation of
`DaughterTrustIteratorProvider` interface Signed-off-by: Pavel Karpy <carpawell@nspcc.ru>
This commit is contained in:
parent
e2a1b0e0ee
commit
b293ea3ba1
1 changed files with 92 additions and 0 deletions
92
cmd/neofs-node/reputation/intermediate/storage.go
Normal file
92
cmd/neofs-node/reputation/intermediate/storage.go
Normal file
|
@ -0,0 +1,92 @@
|
|||
package intermediate
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
|
||||
"github.com/nspcc-dev/neofs-node/pkg/services/reputation"
|
||||
eigentrustcalc "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/services/reputation/eigentrust/storage/daughters"
|
||||
)
|
||||
|
||||
// InitialTrustSource is implementation of the
|
||||
// reputation/eigentrust/calculator's InitialTrustSource interface.
|
||||
type InitialTrustSource struct{}
|
||||
|
||||
// InitialTrust returns `1` as initial trust value.
|
||||
func (i InitialTrustSource) InitialTrust(reputation.PeerID) (reputation.TrustValue, error) {
|
||||
return reputation.TrustOne, nil
|
||||
}
|
||||
|
||||
// DaughterTrustIteratorProvider is implementation of the
|
||||
// reputation/eigentrust/calculator's DaughterTrustIteratorProvider interface.
|
||||
type DaughterTrustIteratorProvider struct {
|
||||
daughterStorage *daughters.Storage
|
||||
consumerStorage *consumerstorage.Storage
|
||||
}
|
||||
|
||||
type ErrNoData struct {
|
||||
hasDaughter bool
|
||||
daughter reputation.PeerID
|
||||
epoch uint64
|
||||
}
|
||||
|
||||
func (e *ErrNoData) Error() string {
|
||||
if e.hasDaughter {
|
||||
return fmt.Sprintf("no data in %d epoch for peer: %s", e.epoch, e.daughter)
|
||||
}
|
||||
|
||||
return fmt.Sprintf("no daughter data in %d epoch", e.epoch)
|
||||
}
|
||||
|
||||
// InitDaughterIterator returns iterator over received
|
||||
// local trusts for ctx.Epoch() epoch from daughter p.
|
||||
//
|
||||
// Returns ErrNoData if there is no trust data for
|
||||
// specified epoch and daughter's PeerId.
|
||||
func (ip *DaughterTrustIteratorProvider) InitDaughterIterator(ctx eigentrustcalc.Context,
|
||||
p reputation.PeerID) (eigentrustcalc.TrustIterator, error) {
|
||||
daughterIterator, ok := ip.daughterStorage.DaughterTrusts(ctx.Epoch(), p)
|
||||
if !ok {
|
||||
return nil, &ErrNoData{
|
||||
daughter: p,
|
||||
hasDaughter: true,
|
||||
epoch: ctx.Epoch(),
|
||||
}
|
||||
}
|
||||
|
||||
return daughterIterator, nil
|
||||
}
|
||||
|
||||
// InitAllDaughtersIterator returns iterator over all
|
||||
// daughters of the current node(manager) and all local
|
||||
// trusts received from them for ctx.Epoch() epoch.
|
||||
//
|
||||
// Returns ErrNoData if there is no trust data for
|
||||
// specified epoch.
|
||||
func (ip *DaughterTrustIteratorProvider) InitAllDaughtersIterator(
|
||||
ctx eigentrustcalc.Context) (eigentrustcalc.PeerTrustsIterator, error) {
|
||||
iter, ok := ip.daughterStorage.AllDaughterTrusts(ctx.Epoch())
|
||||
if !ok {
|
||||
return nil, &ErrNoData{epoch: ctx.Epoch()}
|
||||
}
|
||||
|
||||
return iter, nil
|
||||
|
||||
}
|
||||
|
||||
// InitConsumersIterator returns iterator over all daughters
|
||||
// of the current node(manager) and all their consumers' local
|
||||
// trusts for ctx.Epoch() epoch and ctx.I() iteration.
|
||||
|
||||
// Returns ErrNoData if there is no trust data for
|
||||
// specified epoch and iteration.
|
||||
func (ip *DaughterTrustIteratorProvider) InitConsumersIterator(
|
||||
ctx eigentrustcalc.Context) (eigentrustcalc.PeerTrustsIterator, error) {
|
||||
consumerIterator, ok := ip.consumerStorage.Consumers(ctx.Epoch(), ctx.I())
|
||||
if !ok {
|
||||
return nil, &ErrNoData{epoch: ctx.Epoch()}
|
||||
}
|
||||
|
||||
return consumerIterator, nil
|
||||
}
|
Loading…
Reference in a new issue