forked from TrueCloudLab/frostfs-node
[#36] Parse attributes from node config
Well-known attributes set up with explicit configuration values, other attributes set up from chain of attributes. Chain of attributes is a string, that contains keys-value pairs, divided by semicolon. Pairs itself divided by slash. E.g. "StorageType:HDD/RPM:7200" Signed-off-by: Alex Vanin <alexey@nspcc.ru>
This commit is contained in:
parent
28534a509d
commit
1759ff57f1
2 changed files with 40 additions and 6 deletions
|
@ -4,6 +4,7 @@ import (
|
|||
"context"
|
||||
"crypto/ecdsa"
|
||||
"net"
|
||||
"strconv"
|
||||
"strings"
|
||||
"sync"
|
||||
|
||||
|
@ -20,6 +21,9 @@ const (
|
|||
// config keys for cfgNodeInfo
|
||||
cfgNodeKey = "node.key"
|
||||
cfgBootstrapAddress = "node.address"
|
||||
cfgNodeAttributePrefix = "node.attribute"
|
||||
cfgNodeCapacity = "node.capacity"
|
||||
cfgNodePrice = "node.price"
|
||||
|
||||
// config keys for cfgGRPC
|
||||
cfgListenAddress = "grpc.endpoint"
|
||||
|
@ -96,6 +100,9 @@ type BootstrapType uint32
|
|||
|
||||
type cfgNodeInfo struct {
|
||||
bootType BootstrapType
|
||||
attributes []string
|
||||
capacity uint64
|
||||
price uint64
|
||||
}
|
||||
|
||||
const (
|
||||
|
@ -141,6 +148,9 @@ func initCfg(path string) *cfg {
|
|||
},
|
||||
cfgNodeInfo: cfgNodeInfo{
|
||||
bootType: StorageNode,
|
||||
attributes: readAttributes(viperCfg),
|
||||
capacity: viperCfg.GetUint64(cfgNodeCapacity),
|
||||
price: viperCfg.GetUint64(cfgNodePrice),
|
||||
},
|
||||
}
|
||||
}
|
||||
|
@ -169,6 +179,8 @@ func initViper(path string) *viper.Viper {
|
|||
func defaultConfiguration(v *viper.Viper) {
|
||||
// fixme: all hardcoded private keys must be removed
|
||||
v.SetDefault(cfgNodeKey, "Kwk6k2eC3L3QuPvD8aiaNyoSXgQ2YL1bwS5CP1oKoA9waeAze97s")
|
||||
v.SetDefault(cfgNodeCapacity, "10")
|
||||
v.SetDefault(cfgNodePrice, "10")
|
||||
v.SetDefault(cfgBootstrapAddress, "") // address to bootstrap with
|
||||
|
||||
v.SetDefault(cfgMorphRPCAddress, "http://morph_chain.localtest.nspcc.ru:30333/")
|
||||
|
@ -183,3 +195,18 @@ func defaultConfiguration(v *viper.Viper) {
|
|||
v.SetDefault(cfgNetmapContract, "75194459637323ea8837d2afe8225ec74a5658c3")
|
||||
v.SetDefault(cfgNetmapFee, "1")
|
||||
}
|
||||
|
||||
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
|
||||
} else {
|
||||
attrs = append(attrs, attr)
|
||||
}
|
||||
}
|
||||
|
||||
return attrs
|
||||
}
|
||||
|
|
|
@ -1,10 +1,12 @@
|
|||
package main
|
||||
|
||||
import (
|
||||
v2netmap "github.com/nspcc-dev/neofs-api-go/v2/netmap"
|
||||
crypto "github.com/nspcc-dev/neofs-crypto"
|
||||
"github.com/nspcc-dev/neofs-node/pkg/morph/client"
|
||||
"github.com/nspcc-dev/neofs-node/pkg/morph/client/netmap"
|
||||
"github.com/nspcc-dev/neofs-node/pkg/morph/client/netmap/wrapper"
|
||||
"github.com/nspcc-dev/neofs-node/pkg/util/attributes"
|
||||
"github.com/pkg/errors"
|
||||
)
|
||||
|
||||
|
@ -30,10 +32,15 @@ func bootstrapNode(c *cfg) {
|
|||
cliWrapper, err := wrapper.New(cli)
|
||||
fatalOnErr(errors.Wrap(err, "bootstrap error"))
|
||||
|
||||
peerInfo := new(netmap.NodeInfo)
|
||||
attrs, err := attributes.ParseV2Attributes(c.cfgNodeInfo.attributes, nil)
|
||||
if err != nil {
|
||||
fatalOnErr(errors.Wrap(err, "bootstrap attribute error"))
|
||||
}
|
||||
|
||||
peerInfo := new(v2netmap.NodeInfo)
|
||||
peerInfo.SetAddress(c.viper.GetString(cfgBootstrapAddress))
|
||||
peerInfo.SetPublicKey(crypto.MarshalPublicKey(&c.key.PublicKey))
|
||||
// todo: add attributes as opts
|
||||
peerInfo.SetAttributes(attrs)
|
||||
|
||||
err = cliWrapper.AddPeer(peerInfo)
|
||||
fatalOnErr(errors.Wrap(err, "bootstrap error"))
|
||||
|
|
Loading…
Reference in a new issue