forked from TrueCloudLab/frostfs-node
[#482] reputation/router: Make route pkg independent
Make route package independent from controller package. Add common interfaces to `./common` directory. Signed-off-by: Pavel Karpy <carpawell@nspcc.ru>
This commit is contained in:
parent
0a16aaacb1
commit
ac8441b718
8 changed files with 87 additions and 75 deletions
|
@ -20,6 +20,7 @@ import (
|
||||||
"github.com/nspcc-dev/neofs-node/pkg/network/cache"
|
"github.com/nspcc-dev/neofs-node/pkg/network/cache"
|
||||||
grpcreputation "github.com/nspcc-dev/neofs-node/pkg/network/transport/reputation/grpc"
|
grpcreputation "github.com/nspcc-dev/neofs-node/pkg/network/transport/reputation/grpc"
|
||||||
"github.com/nspcc-dev/neofs-node/pkg/services/reputation"
|
"github.com/nspcc-dev/neofs-node/pkg/services/reputation"
|
||||||
|
reputationcommon "github.com/nspcc-dev/neofs-node/pkg/services/reputation/common"
|
||||||
trustcontroller "github.com/nspcc-dev/neofs-node/pkg/services/reputation/local/controller"
|
trustcontroller "github.com/nspcc-dev/neofs-node/pkg/services/reputation/local/controller"
|
||||||
reputationroute "github.com/nspcc-dev/neofs-node/pkg/services/reputation/local/route"
|
reputationroute "github.com/nspcc-dev/neofs-node/pkg/services/reputation/local/route"
|
||||||
"github.com/nspcc-dev/neofs-node/pkg/services/reputation/local/route/managers"
|
"github.com/nspcc-dev/neofs-node/pkg/services/reputation/local/route/managers"
|
||||||
|
@ -41,7 +42,7 @@ type localTrustStorage struct {
|
||||||
}
|
}
|
||||||
|
|
||||||
type localTrustIterator struct {
|
type localTrustIterator struct {
|
||||||
ctx trustcontroller.Context
|
ctx reputationcommon.Context
|
||||||
|
|
||||||
storage *localTrustStorage
|
storage *localTrustStorage
|
||||||
|
|
||||||
|
@ -55,7 +56,7 @@ type managerBuilder struct {
|
||||||
|
|
||||||
type remoteLocalTrustProvider struct {
|
type remoteLocalTrustProvider struct {
|
||||||
localAddrSrc network.LocalAddressSource
|
localAddrSrc network.LocalAddressSource
|
||||||
deadEndProvider trustcontroller.WriterProvider
|
deadEndProvider reputationcommon.WriterProvider
|
||||||
key *ecdsa.PrivateKey
|
key *ecdsa.PrivateKey
|
||||||
|
|
||||||
clientCache interface {
|
clientCache interface {
|
||||||
|
@ -74,7 +75,7 @@ func (nopReputationWriter) Close() error {
|
||||||
}
|
}
|
||||||
|
|
||||||
type remoteLocalTrustWriter struct {
|
type remoteLocalTrustWriter struct {
|
||||||
ctx trustcontroller.Context
|
ctx reputationcommon.Context
|
||||||
client apiClient.Client
|
client apiClient.Client
|
||||||
key *ecdsa.PrivateKey
|
key *ecdsa.PrivateKey
|
||||||
|
|
||||||
|
@ -116,7 +117,7 @@ type remoteLocalTrustWriterProvider struct {
|
||||||
}
|
}
|
||||||
|
|
||||||
type localTrustLogger struct {
|
type localTrustLogger struct {
|
||||||
ctx trustcontroller.Context
|
ctx reputationcommon.Context
|
||||||
|
|
||||||
log *logger.Logger
|
log *logger.Logger
|
||||||
}
|
}
|
||||||
|
@ -135,7 +136,7 @@ func (*localTrustLogger) Close() error {
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (rtwp *remoteLocalTrustWriterProvider) InitWriter(ctx trustcontroller.Context) (trustcontroller.Writer, error) {
|
func (rtwp *remoteLocalTrustWriterProvider) InitWriter(ctx reputationcommon.Context) (reputationcommon.Writer, error) {
|
||||||
return &remoteLocalTrustWriter{
|
return &remoteLocalTrustWriter{
|
||||||
ctx: ctx,
|
ctx: ctx,
|
||||||
client: rtwp.client,
|
client: rtwp.client,
|
||||||
|
@ -143,7 +144,7 @@ func (rtwp *remoteLocalTrustWriterProvider) InitWriter(ctx trustcontroller.Conte
|
||||||
}, nil
|
}, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (rtp *remoteLocalTrustProvider) InitRemote(srv reputationroute.ServerInfo) (trustcontroller.WriterProvider, error) {
|
func (rtp *remoteLocalTrustProvider) InitRemote(srv reputationroute.ServerInfo) (reputationcommon.WriterProvider, error) {
|
||||||
if srv == nil {
|
if srv == nil {
|
||||||
return rtp.deadEndProvider, nil
|
return rtp.deadEndProvider, nil
|
||||||
}
|
}
|
||||||
|
@ -201,7 +202,7 @@ func (mb *managerBuilder) BuildManagers(epoch uint64, p reputation.PeerID) ([]re
|
||||||
return nil, nil
|
return nil, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *localTrustStorage) InitIterator(ctx trustcontroller.Context) (trustcontroller.Iterator, error) {
|
func (s *localTrustStorage) InitIterator(ctx reputationcommon.Context) (trustcontroller.Iterator, error) {
|
||||||
epochStorage, err := s.storage.DataForEpoch(ctx.Epoch())
|
epochStorage, err := s.storage.DataForEpoch(ctx.Epoch())
|
||||||
if err != nil && !errors.Is(err, truststorage.ErrNoPositiveTrust) {
|
if err != nil && !errors.Is(err, truststorage.ErrNoPositiveTrust) {
|
||||||
return nil, err
|
return nil, err
|
||||||
|
@ -214,7 +215,7 @@ func (s *localTrustStorage) InitIterator(ctx trustcontroller.Context) (trustcont
|
||||||
}, nil
|
}, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *localTrustStorage) InitWriter(ctx trustcontroller.Context) (trustcontroller.Writer, error) {
|
func (s *localTrustStorage) InitWriter(ctx reputationcommon.Context) (reputationcommon.Writer, error) {
|
||||||
return &localTrustLogger{
|
return &localTrustLogger{
|
||||||
ctx: ctx,
|
ctx: ctx,
|
||||||
log: s.log,
|
log: s.log,
|
||||||
|
@ -362,7 +363,7 @@ func initReputationService(c *cfg) {
|
||||||
type reputationServer struct {
|
type reputationServer struct {
|
||||||
*cfg
|
*cfg
|
||||||
log *logger.Logger
|
log *logger.Logger
|
||||||
router trustcontroller.WriterProvider
|
router reputationcommon.WriterProvider
|
||||||
routeBuilder reputationroute.Builder
|
routeBuilder reputationroute.Builder
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -447,7 +448,7 @@ func apiToLocalTrust(t *v2reputation.Trust, trustingPeer []byte) reputation.Trus
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *reputationServer) processTrust(epoch uint64, t reputation.Trust,
|
func (s *reputationServer) processTrust(epoch uint64, t reputation.Trust,
|
||||||
passedRoute []reputationroute.ServerInfo, w trustcontroller.Writer) error {
|
passedRoute []reputationroute.ServerInfo, w reputationcommon.Writer) error {
|
||||||
err := reputationroute.CheckRoute(s.routeBuilder, epoch, t, passedRoute)
|
err := reputationroute.CheckRoute(s.routeBuilder, epoch, t, passedRoute)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return errors.Wrap(err, "wrong route of reputation trust value")
|
return errors.Wrap(err, "wrong route of reputation trust value")
|
||||||
|
|
55
pkg/services/reputation/common/deps.go
Normal file
55
pkg/services/reputation/common/deps.go
Normal file
|
@ -0,0 +1,55 @@
|
||||||
|
package common
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"io"
|
||||||
|
|
||||||
|
"github.com/nspcc-dev/neofs-node/pkg/services/reputation"
|
||||||
|
)
|
||||||
|
|
||||||
|
// Context wraps stdlib context
|
||||||
|
// with accompanying meta values.
|
||||||
|
type Context interface {
|
||||||
|
context.Context
|
||||||
|
|
||||||
|
// Must return epoch number to select the values.
|
||||||
|
Epoch() uint64
|
||||||
|
}
|
||||||
|
|
||||||
|
// Writer describes the interface for storing reputation.Trust values.
|
||||||
|
//
|
||||||
|
// This interface is provided by both local storage
|
||||||
|
// of values and remote (wrappers over the RPC).
|
||||||
|
type Writer interface {
|
||||||
|
// Write performs a write operation of reputation.Trust value
|
||||||
|
// and returns any error encountered.
|
||||||
|
//
|
||||||
|
// All values after the Close call must be flushed to the
|
||||||
|
// physical target. Implementations can cache values before
|
||||||
|
// Close operation.
|
||||||
|
//
|
||||||
|
// Write must not be called after Close.
|
||||||
|
Write(reputation.Trust) error
|
||||||
|
|
||||||
|
// Close exits with method-providing Writer.
|
||||||
|
//
|
||||||
|
// All cached values must be flushed before
|
||||||
|
// the Close's return.
|
||||||
|
//
|
||||||
|
// Methods must not be called after Close.
|
||||||
|
io.Closer
|
||||||
|
}
|
||||||
|
|
||||||
|
// WriterProvider is a group of methods provided
|
||||||
|
// by entity which generates keepers of
|
||||||
|
// reputation.Trust values.
|
||||||
|
type WriterProvider interface {
|
||||||
|
// InitWriter should return an initialized Writer.
|
||||||
|
//
|
||||||
|
// Initialization problems are reported via error.
|
||||||
|
// If no error was returned, then the Writer must not be nil.
|
||||||
|
//
|
||||||
|
// Implementations can have different logic for different
|
||||||
|
// contexts, so specific ones may document their own behavior.
|
||||||
|
InitWriter(Context) (Writer, error)
|
||||||
|
}
|
|
@ -4,6 +4,7 @@ import (
|
||||||
"context"
|
"context"
|
||||||
|
|
||||||
"github.com/nspcc-dev/neofs-node/pkg/services/reputation"
|
"github.com/nspcc-dev/neofs-node/pkg/services/reputation"
|
||||||
|
"github.com/nspcc-dev/neofs-node/pkg/services/reputation/common"
|
||||||
"github.com/nspcc-dev/neofs-node/pkg/util/logger"
|
"github.com/nspcc-dev/neofs-node/pkg/util/logger"
|
||||||
"github.com/pkg/errors"
|
"github.com/pkg/errors"
|
||||||
"go.uber.org/zap"
|
"go.uber.org/zap"
|
||||||
|
@ -47,7 +48,7 @@ type reportContext struct {
|
||||||
|
|
||||||
log *logger.Logger
|
log *logger.Logger
|
||||||
|
|
||||||
ctx Context
|
ctx common.Context
|
||||||
}
|
}
|
||||||
|
|
||||||
type iteratorContext struct {
|
type iteratorContext struct {
|
||||||
|
|
|
@ -4,6 +4,8 @@ import (
|
||||||
"context"
|
"context"
|
||||||
"fmt"
|
"fmt"
|
||||||
"sync"
|
"sync"
|
||||||
|
|
||||||
|
"github.com/nspcc-dev/neofs-node/pkg/services/reputation/common"
|
||||||
)
|
)
|
||||||
|
|
||||||
// Prm groups the required parameters of the Controller's constructor.
|
// Prm groups the required parameters of the Controller's constructor.
|
||||||
|
@ -22,7 +24,7 @@ type Prm struct {
|
||||||
// trust to other nodes.
|
// trust to other nodes.
|
||||||
//
|
//
|
||||||
// Must not be nil.
|
// Must not be nil.
|
||||||
LocalTrustTarget WriterProvider
|
LocalTrustTarget common.WriterProvider
|
||||||
}
|
}
|
||||||
|
|
||||||
// Controller represents main handler for starting
|
// Controller represents main handler for starting
|
||||||
|
|
|
@ -1,59 +1,10 @@
|
||||||
package trustcontroller
|
package trustcontroller
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"context"
|
|
||||||
"io"
|
|
||||||
|
|
||||||
"github.com/nspcc-dev/neofs-node/pkg/services/reputation"
|
"github.com/nspcc-dev/neofs-node/pkg/services/reputation"
|
||||||
|
"github.com/nspcc-dev/neofs-node/pkg/services/reputation/common"
|
||||||
)
|
)
|
||||||
|
|
||||||
// Context wraps stdlib context
|
|
||||||
// with accompanying meta values.
|
|
||||||
type Context interface {
|
|
||||||
context.Context
|
|
||||||
|
|
||||||
// Must return epoch number to select the values.
|
|
||||||
Epoch() uint64
|
|
||||||
}
|
|
||||||
|
|
||||||
// Writer describes the interface for storing reputation.Trust values.
|
|
||||||
//
|
|
||||||
// This interface is provided by both local storage
|
|
||||||
// of values and remote (wrappers over the RPC).
|
|
||||||
type Writer interface {
|
|
||||||
// Write performs a write operation of reputation.Trust value
|
|
||||||
// and returns any error encountered.
|
|
||||||
//
|
|
||||||
// All values after the Close call must be flushed to the
|
|
||||||
// physical target. Implementations can cache values before
|
|
||||||
// Close operation.
|
|
||||||
//
|
|
||||||
// Write must not be called after Close.
|
|
||||||
Write(reputation.Trust) error
|
|
||||||
|
|
||||||
// Close exits with method-providing Writer.
|
|
||||||
//
|
|
||||||
// All cached values must be flushed before
|
|
||||||
// the Close's return.
|
|
||||||
//
|
|
||||||
// Methods must not be called after Close.
|
|
||||||
io.Closer
|
|
||||||
}
|
|
||||||
|
|
||||||
// WriterProvider is a group of methods provided
|
|
||||||
// by entity which generates keepers of
|
|
||||||
// reputation.Trust values.
|
|
||||||
type WriterProvider interface {
|
|
||||||
// InitWriter should return an initialized Writer.
|
|
||||||
//
|
|
||||||
// Initialization problems are reported via error.
|
|
||||||
// If no error was returned, then the Writer must not be nil.
|
|
||||||
//
|
|
||||||
// Implementations can have different logic for different
|
|
||||||
// contexts, so specific ones may document their own behavior.
|
|
||||||
InitWriter(Context) (Writer, error)
|
|
||||||
}
|
|
||||||
|
|
||||||
// Iterator is a group of methods provided by entity
|
// Iterator is a group of methods provided by entity
|
||||||
// which can iterate over a group of reputation.Trust values.
|
// which can iterate over a group of reputation.Trust values.
|
||||||
type Iterator interface {
|
type Iterator interface {
|
||||||
|
@ -79,5 +30,5 @@ type IteratorProvider interface {
|
||||||
//
|
//
|
||||||
// Implementations can have different logic for different
|
// Implementations can have different logic for different
|
||||||
// contexts, so specific ones may document their own behavior.
|
// contexts, so specific ones may document their own behavior.
|
||||||
InitIterator(Context) (Iterator, error)
|
InitIterator(common.Context) (Iterator, error)
|
||||||
}
|
}
|
||||||
|
|
|
@ -1,15 +1,17 @@
|
||||||
package trustcontroller
|
package trustcontroller
|
||||||
|
|
||||||
|
import "github.com/nspcc-dev/neofs-node/pkg/services/reputation/common"
|
||||||
|
|
||||||
type storageWrapper struct {
|
type storageWrapper struct {
|
||||||
w Writer
|
w common.Writer
|
||||||
i Iterator
|
i Iterator
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s storageWrapper) InitIterator(Context) (Iterator, error) {
|
func (s storageWrapper) InitIterator(common.Context) (Iterator, error) {
|
||||||
return s.i, nil
|
return s.i, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s storageWrapper) InitWriter(Context) (Writer, error) {
|
func (s storageWrapper) InitWriter(common.Context) (common.Writer, error) {
|
||||||
return s.w, nil
|
return s.w, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -23,7 +25,7 @@ func SimpleIteratorProvider(i Iterator) IteratorProvider {
|
||||||
|
|
||||||
// SimpleWriterProvider returns WriterProvider that provides
|
// SimpleWriterProvider returns WriterProvider that provides
|
||||||
// static context-independent Writer.
|
// static context-independent Writer.
|
||||||
func SimpleWriterProvider(w Writer) WriterProvider {
|
func SimpleWriterProvider(w common.Writer) common.WriterProvider {
|
||||||
return &storageWrapper{
|
return &storageWrapper{
|
||||||
w: w,
|
w: w,
|
||||||
}
|
}
|
||||||
|
|
|
@ -4,18 +4,18 @@ import (
|
||||||
"sync"
|
"sync"
|
||||||
|
|
||||||
"github.com/nspcc-dev/neofs-node/pkg/services/reputation"
|
"github.com/nspcc-dev/neofs-node/pkg/services/reputation"
|
||||||
reputationcontroller "github.com/nspcc-dev/neofs-node/pkg/services/reputation/local/controller"
|
"github.com/nspcc-dev/neofs-node/pkg/services/reputation/common"
|
||||||
"go.uber.org/zap"
|
"go.uber.org/zap"
|
||||||
)
|
)
|
||||||
|
|
||||||
type routeContext struct {
|
type routeContext struct {
|
||||||
reputationcontroller.Context
|
common.Context
|
||||||
|
|
||||||
passedRoute []ServerInfo
|
passedRoute []ServerInfo
|
||||||
}
|
}
|
||||||
|
|
||||||
// NewRouteContext wraps the main context of value passing with its traversal route and epoch.
|
// NewRouteContext wraps the main context of value passing with its traversal route and epoch.
|
||||||
func NewRouteContext(ctx reputationcontroller.Context, passed []ServerInfo) reputationcontroller.Context {
|
func NewRouteContext(ctx common.Context, passed []ServerInfo) common.Context {
|
||||||
return &routeContext{
|
return &routeContext{
|
||||||
Context: ctx,
|
Context: ctx,
|
||||||
passedRoute: passed,
|
passedRoute: passed,
|
||||||
|
@ -28,7 +28,7 @@ type trustWriter struct {
|
||||||
ctx *routeContext
|
ctx *routeContext
|
||||||
|
|
||||||
routeMtx sync.RWMutex
|
routeMtx sync.RWMutex
|
||||||
mServers map[string]reputationcontroller.Writer
|
mServers map[string]common.Writer
|
||||||
}
|
}
|
||||||
|
|
||||||
// InitWriter initializes and returns Writer that sends each value to its next route point.
|
// InitWriter initializes and returns Writer that sends each value to its next route point.
|
||||||
|
@ -45,7 +45,7 @@ type trustWriter struct {
|
||||||
// runtime and never returns an error.
|
// runtime and never returns an error.
|
||||||
//
|
//
|
||||||
// Always returns nil error.
|
// Always returns nil error.
|
||||||
func (r *Router) InitWriter(ctx reputationcontroller.Context) (reputationcontroller.Writer, error) {
|
func (r *Router) InitWriter(ctx common.Context) (common.Writer, error) {
|
||||||
var (
|
var (
|
||||||
routeCtx *routeContext
|
routeCtx *routeContext
|
||||||
ok bool
|
ok bool
|
||||||
|
@ -61,7 +61,7 @@ func (r *Router) InitWriter(ctx reputationcontroller.Context) (reputationcontrol
|
||||||
return &trustWriter{
|
return &trustWriter{
|
||||||
router: r,
|
router: r,
|
||||||
ctx: routeCtx,
|
ctx: routeCtx,
|
||||||
mServers: make(map[string]reputationcontroller.Writer),
|
mServers: make(map[string]common.Writer),
|
||||||
}, nil
|
}, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -2,7 +2,7 @@ package reputationroute
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"github.com/nspcc-dev/neofs-node/pkg/services/reputation"
|
"github.com/nspcc-dev/neofs-node/pkg/services/reputation"
|
||||||
reputationcontroller "github.com/nspcc-dev/neofs-node/pkg/services/reputation/local/controller"
|
"github.com/nspcc-dev/neofs-node/pkg/services/reputation/common"
|
||||||
)
|
)
|
||||||
|
|
||||||
// ServerInfo describes a set of
|
// ServerInfo describes a set of
|
||||||
|
@ -38,5 +38,5 @@ type RemoteWriterProvider interface {
|
||||||
// corresponding to info.
|
// corresponding to info.
|
||||||
//
|
//
|
||||||
// Nil info matches the end of the route.
|
// Nil info matches the end of the route.
|
||||||
InitRemote(info ServerInfo) (reputationcontroller.WriterProvider, error)
|
InitRemote(info ServerInfo) (common.WriterProvider, error)
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in a new issue