Merge pull request #501 from shutter-network/feat/libp2p_seed_node_metrics

Add libp2p seed node metrics

Author: ulope

rolling-shutter/cmd/p2pnode/p2pnode.go

@@ -3,7 +3,6 @@ package p2pnode
 import (
 	"context"
 
-	"github.com/pkg/errors"
 	"github.com/rs/zerolog/log"
 	"github.com/spf13/cobra"
 
@@ -31,9 +30,6 @@ func main(config *p2pnode.Config) error {
 	log.Info().
 		Str("version", shversion.Version()).
 		Msg("starting p2pnode")
-	p2pService, err := p2pnode.New(config)
-	if err != nil {
-		return errors.Wrapf(err, "failed to instantiate p2pnode")
-	}
-	return service.RunWithSighandler(context.Background(), p2pService)
+	p2pNode := p2pnode.New(config)
+	return service.RunWithSighandler(context.Background(), p2pNode)
 }

rolling-shutter/p2pnode/config.go

@@ -4,6 +4,7 @@ import (
 	"io"
 
 	"github.com/shutter-network/rolling-shutter/rolling-shutter/medley/configuration"
+	"github.com/shutter-network/rolling-shutter/rolling-shutter/medley/metricsserver"
 	"github.com/shutter-network/rolling-shutter/rolling-shutter/p2p"
 )
 
@@ -18,11 +19,13 @@ func NewConfig() *Config {
 type Config struct {
 	ListenMessages bool `comment:"whether to register handlers on the messages and log them"`
 
-	P2P *p2p.Config
+	P2P     *p2p.Config
+	Metrics *metricsserver.MetricsConfig
 }
 
 func (c *Config) Init() {
 	c.P2P = p2p.NewConfig()
+	c.Metrics = metricsserver.NewConfig()
 }
 
 func (c *Config) Name() string {

rolling-shutter/p2pnode/host.go

@@ -4,8 +4,10 @@ import (
 	"context"
 
 	pubsub "github.com/libp2p/go-libp2p-pubsub"
+	"github.com/pkg/errors"
 	"github.com/rs/zerolog/log"
 
+	"github.com/shutter-network/rolling-shutter/rolling-shutter/medley/metricsserver"
 	"github.com/shutter-network/rolling-shutter/rolling-shutter/medley/service"
 	"github.com/shutter-network/rolling-shutter/rolling-shutter/p2p"
 	"github.com/shutter-network/rolling-shutter/rolling-shutter/p2pmsg"
@@ -35,13 +37,33 @@ func (dummyMessageHandler) MessagePrototypes() []p2pmsg.Message {
 	}
 }
 
-func New(config *Config) (service.Service, error) {
-	p2pHandler, err := p2p.New(config.P2P)
+type P2PNode struct {
+	config *Config
+}
+
+func New(config *Config) *P2PNode {
+	return &P2PNode{
+		config: config,
+	}
+}
+
+func (node *P2PNode) Start(ctx context.Context, runner service.Runner) error {
+	services := []service.Service{}
+
+	p2pHandler, err := p2p.New(node.config.P2P)
 	if err != nil {
-		return nil, err
+		if err != nil {
+			return errors.Wrap(err, "failed to initialize p2p messaging")
+		}
 	}
-	if config.ListenMessages {
+	if node.config.ListenMessages {
 		p2pHandler.AddMessageHandler(dummyMessageHandler{})
 	}
-	return p2pHandler, nil
+	services = append(services, p2pHandler)
+
+	if node.config.Metrics.Enabled {
+		metricsServer := metricsserver.New(node.config.Metrics)
+		services = append(services, metricsServer)
+	}
+	return runner.StartService(services...)
 }