frostfs-node/cmd/neofs-node/attributes.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/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
}

func addWellKnownAttributes(attrs []*netmap.NodeAttribute) []*netmap.NodeAttribute {
	var hasCapacity, hasPrice bool

	// check if user defined capacity and price attributes
	for i := range attrs {
		if !hasPrice && attrs[i].Key() == netmap.PriceAttr {
			hasPrice = true
		} else if !hasCapacity && attrs[i].Key() == netmap.CapacityAttr {
			hasCapacity = true
		}
	}

	// do not override user defined capacity and price attributes

	if !hasCapacity {
		capacity := netmap.NewNodeAttribute()
		capacity.SetKey(netmap.CapacityAttr)
		capacity.SetValue(strconv.FormatUint(defaultCapacity, 10))
		attrs = append(attrs, capacity)
	}

	if !hasPrice {
		price := netmap.NewNodeAttribute()
		price.SetKey(netmap.PriceAttr)
		price.SetValue(strconv.FormatUint(defaultPrice, 10))
		attrs = append(attrs, price)
	}

	return attrs
}