migrating p2p to its own package.
This commit is contained in:
@@ -1,133 +0,0 @@
|
||||
package seed
|
||||
|
||||
import (
|
||||
"context"
|
||||
"time"
|
||||
|
||||
"github.com/libp2p/go-libp2p"
|
||||
"github.com/libp2p/go-libp2p/core/host"
|
||||
"github.com/multiformats/go-multiaddr"
|
||||
|
||||
"git-indra.lan/indra-labs/indra"
|
||||
"git-indra.lan/indra-labs/indra/pkg/interrupt"
|
||||
"git-indra.lan/indra-labs/indra/pkg/p2p/introducer"
|
||||
log2 "git-indra.lan/indra-labs/indra/pkg/proc/log"
|
||||
"git-indra.lan/indra-labs/indra/pkg/seed/metrics"
|
||||
)
|
||||
|
||||
var (
|
||||
log = log2.GetLogger(indra.PathBase)
|
||||
check = log.E.Chk
|
||||
)
|
||||
|
||||
var (
|
||||
userAgent = "/indra:" + indra.SemVer + "/"
|
||||
)
|
||||
|
||||
type Server struct {
|
||||
context.Context
|
||||
|
||||
config *Config
|
||||
|
||||
host host.Host
|
||||
}
|
||||
|
||||
func (srv *Server) Restart() (err error) {
|
||||
|
||||
log.I.Ln("restarting the server.")
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (srv *Server) Shutdown() (err error) {
|
||||
|
||||
if err = srv.host.Close(); check(err) {
|
||||
// continue
|
||||
}
|
||||
|
||||
log.I.Ln("shutdown complete")
|
||||
|
||||
return
|
||||
}
|
||||
|
||||
func (srv *Server) Serve() (err error) {
|
||||
|
||||
log.I.Ln("starting the p2p server")
|
||||
|
||||
// Here we create a context with cancel and add it to the interrupt handler
|
||||
var ctx context.Context
|
||||
var cancel context.CancelFunc
|
||||
|
||||
ctx, cancel = context.WithCancel(context.Background())
|
||||
|
||||
interrupt.AddHandler(cancel)
|
||||
|
||||
// Introduce your node to the network
|
||||
go introducer.Bootstrap(ctx, srv.host, srv.config.SeedAddresses)
|
||||
|
||||
// Get some basic metrics for the host
|
||||
// metrics.Init()
|
||||
// metrics.Set('indra.host.status.reporting.interval', 30 * time.Second)
|
||||
// metrics.Enable('indra.host.status')
|
||||
metrics.SetInterval(30 * time.Second)
|
||||
|
||||
go metrics.HostStatus(ctx, srv.host)
|
||||
|
||||
//var client *rpc.RPCClient
|
||||
//
|
||||
//if client, err = rpc.NewClient(rpc.DefaultClientConfig); check(err) {
|
||||
// return err
|
||||
//}
|
||||
//
|
||||
//client.Start()
|
||||
|
||||
select {
|
||||
|
||||
case <-ctx.Done():
|
||||
|
||||
log.I.Ln("shutting down p2p server")
|
||||
|
||||
srv.Shutdown()
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func New(config *Config) (*Server, error) {
|
||||
|
||||
log.I.Ln("initializing the p2p server")
|
||||
|
||||
var err error
|
||||
var s Server
|
||||
|
||||
s.config = config
|
||||
|
||||
if s.host, err = libp2p.New(libp2p.Identity(config.PrivKey), libp2p.UserAgent(userAgent), libp2p.ListenAddrs(config.ListenAddresses...)); check(err) {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
log.I.Ln("host id:")
|
||||
log.I.Ln("-", s.host.ID())
|
||||
|
||||
log.I.Ln("p2p listeners:")
|
||||
log.I.Ln("-", s.host.Addrs())
|
||||
|
||||
if len(config.ConnectAddresses) > 0 {
|
||||
|
||||
log.I.Ln("connect detected, using only the connect seed addresses")
|
||||
|
||||
config.SeedAddresses = config.ConnectAddresses
|
||||
|
||||
return &s, nil
|
||||
}
|
||||
|
||||
var seedAddresses []multiaddr.Multiaddr
|
||||
|
||||
if seedAddresses, err = config.Params.ParseSeedMultiAddresses(); check(err) {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
config.SeedAddresses = append(config.SeedAddresses, seedAddresses...)
|
||||
|
||||
return &s, err
|
||||
}
|
||||
Reference in New Issue
Block a user