Skip to content

Commit 02cb1b8

Browse files
committed
start unifying encoding/decoding
1 parent 56c61b6 commit 02cb1b8

File tree

1 file changed

+31
-17
lines changed

1 file changed

+31
-17
lines changed

local/network.go

+31-17
Original file line numberDiff line numberDiff line change
@@ -336,15 +336,15 @@ func NewDefaultConfigNNodes(binaryPath string, numNodes uint32) (network.Config,
336336
netConfig.BinaryPath = binaryPath
337337
if int(numNodes) > constants.DefaultNumNodes {
338338
toAdd := int(numNodes) - constants.DefaultNumNodes
339-
keys, err := getNNodeKeys(toAdd)
339+
keys, err := generateNNodeKeys(toAdd)
340340
if err != nil {
341341
return netConfig, err
342342
}
343343
for i := 0; i < toAdd; i++ {
344344
netConfig.NodeConfigs = append(netConfig.NodeConfigs, node.Config{
345-
StakingKey: string(keys[i].stakingKey),
346-
StakingCert: string(keys[i].stakingCert),
347-
StakingSigningKey: base64.StdEncoding.EncodeToString(keys[i].blsKey),
345+
StakingKey: string(keys[i].StakingKey),
346+
StakingCert: string(keys[i].StakingCert),
347+
StakingSigningKey: base64.StdEncoding.EncodeToString(keys[i].BlsKey),
348348
IsBeacon: true,
349349
})
350350
}
@@ -376,35 +376,49 @@ func NewDefaultConfig(binaryPath string) (network.Config, error) {
376376
return NewDefaultConfigNNodes(binaryPath, constants.DefaultNumNodes)
377377
}
378378

379-
type nodeKeys struct {
380-
stakingKey []byte
381-
stakingCert []byte
382-
blsKey []byte
379+
type EncodedNodeKeys struct {
380+
StakingKey string
381+
StakingCert string
382+
BlsKey string
383383
}
384384

385-
func getNodeKeys() (*nodeKeys, error) {
386-
keys := nodeKeys{}
385+
type NodeKeys struct {
386+
StakingKey []byte
387+
StakingCert []byte
388+
BlsKey []byte
389+
}
390+
391+
func encodeNodeKeys(key *NodeKeys) *EncodedNodeKeys {
392+
return &EncodedNodeKeys{
393+
StakingKey: string(key.StakingKey),
394+
StakingCert: string(key.StakingCert),
395+
BlsKey: base64.StdEncoding.EncodeToString(key.BlsKey),
396+
}
397+
}
398+
399+
func generateNodeKeys() (*NodeKeys, error) {
387400
stakingCert, stakingKey, err := staking.NewCertAndKeyBytes()
388401
if err != nil {
389402
return nil, fmt.Errorf("couldn't generate staking Cert/Key: %w", err)
390403
}
391-
keys.stakingKey = stakingKey
392-
keys.stakingCert = stakingCert
393404
key, err := bls.NewSecretKey()
394405
if err != nil {
395406
return nil, fmt.Errorf("couldn't generate new signing key: %w", err)
396407
}
397-
keys.blsKey = bls.SecretKeyToBytes(key)
398-
return &keys, nil
408+
return &NodeKeys{
409+
StakingKey: stakingKey,
410+
StakingCert: stakingCert,
411+
BlsKey: bls.SecretKeyToBytes(key),
412+
}, nil
399413
}
400414

401-
func getNNodeKeys(num int) ([]*nodeKeys, error) {
402-
nodesKeys := []*nodeKeys{}
415+
func generateNNodeKeys(num int) ([]*NodeKeys, error) {
416+
nodesKeys := []*NodeKeys{}
403417
lock := sync.Mutex{}
404418
eg := errgroup.Group{}
405419
for i := 0; i < num; i++ {
406420
eg.Go(func() error {
407-
keys, err := getNodeKeys()
421+
keys, err := generateNodeKeys()
408422
if err != nil {
409423
return err
410424
}

0 commit comments

Comments
 (0)