forked from TrueCloudLab/frostfs-node
d48fb81193
Implement a component for transmitting the value of the used container space along a route defined in the system. Implement WriterProvider interface on it. By implementation, it is the link between the route planner and the point-to-point transmitter, and abstracts from the implementation of both of them. In the future, this implementation will be used as a transmitter of local estimates of storage nodes among themselves. Signed-off-by: Leonard Lyubich <leonard@nspcc.ru>
39 lines
799 B
Go
39 lines
799 B
Go
package loadroute
|
|
|
|
import (
|
|
"bytes"
|
|
"errors"
|
|
|
|
"github.com/nspcc-dev/neofs-api-go/pkg/container"
|
|
)
|
|
|
|
var errWrongRoute = errors.New("wrong route")
|
|
|
|
// CheckRoute checks if the route is a route correctly constructed by the builder for value a.
|
|
//
|
|
// Returns nil if route is correct, otherwise an error clarifying the inconsistency.
|
|
func CheckRoute(builder Builder, a container.UsedSpaceAnnouncement, route []ServerInfo) error {
|
|
for i := 1; i < len(route); i++ {
|
|
servers, err := builder.NextStage(a, route[:i])
|
|
if err != nil {
|
|
return err
|
|
} else if len(servers) == 0 {
|
|
break
|
|
}
|
|
|
|
found := false
|
|
|
|
for j := range servers {
|
|
if bytes.Equal(servers[j].PublicKey(), route[i].PublicKey()) {
|
|
found = true
|
|
break
|
|
}
|
|
}
|
|
|
|
if !found {
|
|
return errWrongRoute
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|