125 lines
2.7 KiB
Go
125 lines
2.7 KiB
Go
package nodeconf
|
|
|
|
import (
|
|
"github.com/anytypeio/go-anytype-infrastructure-experiments/app"
|
|
"github.com/anytypeio/go-anytype-infrastructure-experiments/app/logger"
|
|
"github.com/anytypeio/go-anytype-infrastructure-experiments/common/net/pool"
|
|
"github.com/anytypeio/go-anytype-infrastructure-experiments/config"
|
|
"github.com/anytypeio/go-anytype-infrastructure-experiments/util/keys"
|
|
"github.com/anytypeio/go-anytype-infrastructure-experiments/util/keys/asymmetric/encryptionkey"
|
|
"github.com/anytypeio/go-anytype-infrastructure-experiments/util/keys/asymmetric/signingkey"
|
|
"github.com/anytypeio/go-chash"
|
|
)
|
|
|
|
const CName = "common.nodeconf"
|
|
|
|
const (
|
|
partitionCount = 3000
|
|
replicationFactor = 3
|
|
)
|
|
|
|
var log = logger.NewNamed(CName)
|
|
|
|
type Service interface {
|
|
GetLast() Configuration
|
|
GetById(id string) Configuration
|
|
app.Component
|
|
}
|
|
|
|
type service struct {
|
|
accountId string
|
|
pool pool.Pool
|
|
|
|
last Configuration
|
|
}
|
|
|
|
type Node struct {
|
|
Address string
|
|
PeerId string
|
|
SigningKey signingkey.PubKey
|
|
EncryptionKey encryptionkey.PubKey
|
|
}
|
|
|
|
func (n *Node) Id() string {
|
|
return n.PeerId
|
|
}
|
|
|
|
func (n *Node) Capacity() float64 {
|
|
return 1
|
|
}
|
|
|
|
func (s *service) Init(a *app.App) (err error) {
|
|
conf := a.MustComponent(config.CName).(*config.Config)
|
|
s.accountId = conf.Account.PeerId
|
|
s.pool = a.MustComponent(pool.CName).(pool.Pool)
|
|
|
|
config := &configuration{
|
|
id: "config",
|
|
accountId: s.accountId,
|
|
pool: s.pool,
|
|
}
|
|
if config.chash, err = chash.New(chash.Config{
|
|
PartitionCount: partitionCount,
|
|
ReplicationFactor: replicationFactor,
|
|
}); err != nil {
|
|
return
|
|
}
|
|
members := make([]chash.Member, 0, len(conf.Nodes)-1)
|
|
for _, n := range conf.Nodes {
|
|
if n.PeerId == conf.Account.PeerId {
|
|
continue
|
|
}
|
|
var member *Node
|
|
member, err = nodeFromConfigNode(n)
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
members = append(members, member)
|
|
}
|
|
if err = config.chash.AddMembers(members...); err != nil {
|
|
return
|
|
}
|
|
s.last = config
|
|
return
|
|
}
|
|
|
|
func (s *service) Name() (name string) {
|
|
return CName
|
|
}
|
|
|
|
func (s *service) GetLast() Configuration {
|
|
return s.last
|
|
}
|
|
|
|
func (s *service) GetById(id string) Configuration {
|
|
//TODO implement me
|
|
panic("implement me")
|
|
}
|
|
|
|
func nodeFromConfigNode(
|
|
n config.Node) (*Node, error) {
|
|
decodedSigningKey, err := keys.DecodeKeyFromString(
|
|
n.SigningKey,
|
|
signingkey.UnmarshalEd25519PrivateKey,
|
|
nil)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
decodedEncryptionKey, err := keys.DecodeKeyFromString(
|
|
n.EncryptionKey,
|
|
encryptionkey.NewEncryptionRsaPrivKeyFromBytes,
|
|
nil)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &Node{
|
|
Address: n.Address,
|
|
PeerId: n.PeerId,
|
|
SigningKey: decodedSigningKey.GetPublic(),
|
|
EncryptionKey: decodedEncryptionKey.GetPublic(),
|
|
}, nil
|
|
}
|