83 lines
1.9 KiB
Go
83 lines
1.9 KiB
Go
|
package node
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"crypto/rand"
|
||
|
|
"fmt"
|
||
|
|
|
||
|
|
"github.com/Secured-Finance/p2p-oracle-node/config"
|
||
|
|
"github.com/Secured-Finance/p2p-oracle-node/handler"
|
||
|
|
mapset "github.com/deckarep/golang-set"
|
||
|
|
"github.com/ipfs/go-log"
|
||
|
|
"github.com/libp2p/go-libp2p"
|
||
|
|
crypto "github.com/libp2p/go-libp2p-core/crypto"
|
||
|
|
"github.com/libp2p/go-libp2p-core/host"
|
||
|
|
pubsub "github.com/libp2p/go-libp2p-pubsub"
|
||
|
|
"github.com/multiformats/go-multiaddr"
|
||
|
|
)
|
||
|
|
|
||
|
|
type Node struct {
|
||
|
|
Host host.Host
|
||
|
|
PubSub *pubsub.PubSub
|
||
|
|
GlobalCtx context.Context
|
||
|
|
GlobalCtxCancel context.CancelFunc
|
||
|
|
OracleTopic string
|
||
|
|
networkTopics mapset.Set
|
||
|
|
handler *handler.Handler
|
||
|
|
Config *config.Config
|
||
|
|
Logger *log.ZapEventLogger
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewNode() *Node {
|
||
|
|
node := &Node{
|
||
|
|
Config: config.NewConfig(),
|
||
|
|
Logger: log.Logger("rendezvous"),
|
||
|
|
networkTopics: mapset.NewSet(),
|
||
|
|
}
|
||
|
|
|
||
|
|
return node
|
||
|
|
}
|
||
|
|
|
||
|
|
func (node *Node) setupNode(ctx context.Context, prvKey crypto.PrivKey) {
|
||
|
|
listenMultiAddr, err := multiaddr.NewMultiaddr(fmt.Sprintf("/ip4/%s/tcp/%s", node.Config.ListenAddr, node.Config.ListenPort))
|
||
|
|
if err != nil {
|
||
|
|
node.Logger.Fatal("Failed to generate new node multiaddress:", err)
|
||
|
|
}
|
||
|
|
host, err := libp2p.New(
|
||
|
|
ctx,
|
||
|
|
libp2p.ListenAddrs(listenMultiAddr),
|
||
|
|
libp2p.Identity(prvKey),
|
||
|
|
)
|
||
|
|
if err != nil {
|
||
|
|
node.Logger.Fatal("Failed to set a new libp2p node:", err)
|
||
|
|
}
|
||
|
|
node.Host = host
|
||
|
|
node.startPubSub(ctx, host)
|
||
|
|
}
|
||
|
|
|
||
|
|
func Start() {
|
||
|
|
node := NewNode()
|
||
|
|
log.SetAllLoggers(log.LevelInfo)
|
||
|
|
|
||
|
|
err := log.SetLogLevel("rendezvous", "info")
|
||
|
|
if err != nil {
|
||
|
|
node.Logger.Warn("Failed to set a rendezvous log level:", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
node.parseFlags()
|
||
|
|
|
||
|
|
r := rand.Reader
|
||
|
|
|
||
|
|
// Creates a new RSA key pair for this host.
|
||
|
|
prvKey, _, err := crypto.GenerateKeyPairWithReader(crypto.RSA, 2048, r)
|
||
|
|
if err != nil {
|
||
|
|
node.Logger.Fatal(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
ctx, ctxCancel := context.WithCancel(context.Background())
|
||
|
|
node.GlobalCtx = ctx
|
||
|
|
node.GlobalCtxCancel = ctxCancel
|
||
|
|
|
||
|
|
node.setupNode(ctx, prvKey)
|
||
|
|
}
|