82 lines
1.9 KiB
Go
82 lines
1.9 KiB
Go
package netmap
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/nspcc-dev/hrw"
|
|
)
|
|
|
|
// Netmap represents netmap which contains preprocessed nodes.
|
|
type Netmap struct {
|
|
Nodes Nodes
|
|
}
|
|
|
|
// NewNetmap constructs netmap from the list of raw nodes.
|
|
func NewNetmap(nodes Nodes) (*Netmap, error) {
|
|
return &Netmap{
|
|
Nodes: nodes,
|
|
}, nil
|
|
}
|
|
|
|
func flattenNodes(ns []Nodes) Nodes {
|
|
result := make(Nodes, 0, len(ns))
|
|
for i := range ns {
|
|
result = append(result, ns[i]...)
|
|
}
|
|
|
|
return result
|
|
}
|
|
|
|
// GetPlacementVectors returns placement vectors for an object given containerNodes cnt.
|
|
func (m *Netmap) GetPlacementVectors(cnt ContainerNodes, pivot []byte) ([]Nodes, error) {
|
|
h := hrw.Hash(pivot)
|
|
wf := GetDefaultWeightFunc(m.Nodes)
|
|
result := make([]Nodes, len(cnt.Replicas()))
|
|
|
|
for i, rep := range cnt.Replicas() {
|
|
result[i] = make(Nodes, len(rep))
|
|
copy(result[i], rep)
|
|
hrw.SortSliceByWeightValue(result[i], result[i].Weights(wf), h)
|
|
}
|
|
|
|
return result, nil
|
|
}
|
|
|
|
// GetContainerNodes returns nodes corresponding to each replica.
|
|
// Order of returned nodes corresponds to order of replicas in p.
|
|
// pivot is a seed for HRW sorting.
|
|
func (m *Netmap) GetContainerNodes(p *PlacementPolicy, pivot []byte) (ContainerNodes, error) {
|
|
c := NewContext(m)
|
|
c.setPivot(pivot)
|
|
|
|
if err := c.processFilters(p); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if err := c.processSelectors(p); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
result := make([]Nodes, len(p.Replicas()))
|
|
|
|
for i, r := range p.Replicas() {
|
|
if r == nil {
|
|
return nil, fmt.Errorf("%w: REPLICA", ErrMissingField)
|
|
}
|
|
|
|
if r.Selector() == "" {
|
|
for _, s := range p.Selectors() {
|
|
result[i] = append(result[i], flattenNodes(c.Selections[s.Name()])...)
|
|
}
|
|
}
|
|
|
|
nodes, ok := c.Selections[r.Selector()]
|
|
if !ok {
|
|
return nil, fmt.Errorf("%w: REPLICA '%s'", ErrSelectorNotFound, r.Selector())
|
|
}
|
|
|
|
result[i] = append(result[i], flattenNodes(nodes)...)
|
|
}
|
|
|
|
return containerNodes(result), nil
|
|
}
|