package placementrouter import ( "bytes" "fmt" netmapcore "git.frostfs.info/TrueCloudLab/frostfs-node/pkg/core/netmap" loadcontroller "git.frostfs.info/TrueCloudLab/frostfs-node/pkg/services/container/announcement/load/controller" "git.frostfs.info/TrueCloudLab/frostfs-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.SizeEstimation, passed []loadcontroller.ServerInfo) ([]loadcontroller.ServerInfo, error) { if len(passed) > 1 { return nil, nil } cnr := a.Container() 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([]loadcontroller.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 }