forked from TrueCloudLab/frostfs-node
51 lines
1.5 KiB
Go
51 lines
1.5 KiB
Go
package placementrouter
|
|
|
|
import (
|
|
"bytes"
|
|
"errors"
|
|
"fmt"
|
|
|
|
netmapcore "github.com/nspcc-dev/neofs-node/pkg/core/netmap"
|
|
loadroute "github.com/nspcc-dev/neofs-node/pkg/services/container/announcement/load/route"
|
|
"github.com/nspcc-dev/neofs-sdk-go/container"
|
|
)
|
|
|
|
// NextStage composes container nodes for the container and epoch from a,
|
|
// and returns the list of nodes with maximum weight (one from each vector).
|
|
//
|
|
// If passed route has more than one point, then endpoint of the route is reached.
|
|
//
|
|
// The traversed route is not checked, it is assumed to be correct.
|
|
func (b *Builder) NextStage(a container.UsedSpaceAnnouncement, passed []loadroute.ServerInfo) ([]loadroute.ServerInfo, error) {
|
|
if len(passed) > 1 {
|
|
return nil, nil
|
|
}
|
|
|
|
cnr, ok := a.ContainerID()
|
|
if !ok {
|
|
return nil, errors.New("missing container in load announcement")
|
|
}
|
|
|
|
placement, err := b.placementBuilder.BuildPlacement(a.Epoch(), cnr)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("could not build placement %s: %w", cnr, err)
|
|
}
|
|
|
|
res := make([]loadroute.ServerInfo, 0, len(placement))
|
|
|
|
for i := range placement {
|
|
if len(placement[i]) == 0 {
|
|
continue
|
|
}
|
|
|
|
if len(passed) == 1 && bytes.Equal(passed[0].PublicKey(), placement[i][0].PublicKey()) {
|
|
// add nil element so the announcement will be saved in local memory
|
|
res = append(res, nil)
|
|
} else {
|
|
// add element with remote node to send announcement to
|
|
res = append(res, netmapcore.Node(placement[i][0]))
|
|
}
|
|
}
|
|
|
|
return res, nil
|
|
}
|