forked from TrueCloudLab/frostfs-node
83 lines
1.8 KiB
Go
83 lines
1.8 KiB
Go
package main
|
|
|
|
import (
|
|
"strconv"
|
|
|
|
"github.com/nspcc-dev/neofs-api-go/pkg/netmap"
|
|
"github.com/nspcc-dev/neofs-node/pkg/util/attributes"
|
|
"github.com/pkg/errors"
|
|
"github.com/spf13/viper"
|
|
)
|
|
|
|
const (
|
|
// list of default values for well-known attributes
|
|
defaultCapacity = 0
|
|
defaultPrice = 0
|
|
)
|
|
|
|
func parseAttributes(v *viper.Viper) []*netmap.NodeAttribute {
|
|
stringAttributes := readAttributes(v)
|
|
|
|
attrs, err := attributes.ParseV2Attributes(stringAttributes, nil)
|
|
if err != nil {
|
|
fatalOnErr(err)
|
|
}
|
|
|
|
return addWellKnownAttributes(attrs)
|
|
}
|
|
|
|
func readAttributes(v *viper.Viper) (attrs []string) {
|
|
const maxAttributes = 100
|
|
|
|
for i := 0; i < maxAttributes; i++ {
|
|
attr := v.GetString(cfgNodeAttributePrefix + "_" + strconv.Itoa(i))
|
|
if attr == "" {
|
|
return
|
|
}
|
|
|
|
attrs = append(attrs, attr)
|
|
}
|
|
|
|
return attrs
|
|
}
|
|
|
|
type wellKnownNodeAttrDesc struct {
|
|
explicit bool
|
|
defaultVal string
|
|
}
|
|
|
|
// TODO: define attribute keys in API lib.
|
|
const attrKeyLocode = "UN-LOCODE"
|
|
|
|
func listWellKnownAttrDesc() map[string]wellKnownNodeAttrDesc {
|
|
return map[string]wellKnownNodeAttrDesc{
|
|
netmap.PriceAttr: {defaultVal: strconv.FormatUint(defaultPrice, 10)},
|
|
netmap.CapacityAttr: {defaultVal: strconv.FormatUint(defaultCapacity, 10)},
|
|
attrKeyLocode: {explicit: true},
|
|
}
|
|
}
|
|
|
|
func addWellKnownAttributes(attrs []*netmap.NodeAttribute) []*netmap.NodeAttribute {
|
|
mWellKnown := listWellKnownAttrDesc()
|
|
|
|
// check how user defined well-known attributes
|
|
for i := range attrs {
|
|
delete(mWellKnown, attrs[i].Key())
|
|
}
|
|
|
|
for key, desc := range mWellKnown {
|
|
// check if required attribute is set
|
|
if desc.explicit {
|
|
fatalOnErr(errors.Errorf("missing explicit value of required node attribute %s", key))
|
|
}
|
|
|
|
// set default value of the attribute
|
|
a := netmap.NewNodeAttribute()
|
|
a.SetKey(key)
|
|
a.SetValue(desc.defaultVal)
|
|
|
|
attrs = append(attrs, a)
|
|
}
|
|
|
|
return attrs
|
|
}
|