Merge pull request #67 from joostjager/disable-graph

config: add --disablegraph

Author: carlaKC

collectors/prometheus.go

@@ -64,6 +64,9 @@ type MonitoringConfig struct {
 	// PrimaryNode is the pubkey of the primary node in primary-gateway
 	// setups.
 	PrimaryNode *route.Vertex
+
+	// DisableGraph disables collection of graph metrics
+	DisableGraph bool
 }
 
 func DefaultConfig() *PrometheusConfig {
@@ -88,25 +91,30 @@ func NewPrometheusExporter(cfg *PrometheusConfig, lnd *lndclient.LndServices,
 
 	htlcMonitor := newHtlcMonitor(lnd.Router, errChan)
 
+	collectors := append(
+		[]prometheus.Collector{
+			NewChainCollector(lnd.Client, errChan),
+			NewChannelsCollector(
+				lnd.Client, errChan, monitoringCfg,
+			),
+			NewWalletCollector(lnd, errChan),
+			NewPeerCollector(lnd.Client, errChan),
+			NewInfoCollector(lnd.Client, errChan),
+		},
+		htlcMonitor.collectors()...,
+	)
+
+	if !monitoringCfg.DisableGraph {
+		collectors = append(collectors, NewGraphCollector(lnd.Client, errChan))
+	}
+
 	return &PrometheusExporter{
 		cfg:           cfg,
 		lnd:           lnd,
 		monitoringCfg: monitoringCfg,
-		collectors: append(
-			[]prometheus.Collector{
-				NewChainCollector(lnd.Client, errChan),
-				NewChannelsCollector(
-					lnd.Client, errChan, monitoringCfg,
-				),
-				NewWalletCollector(lnd, errChan),
-				NewGraphCollector(lnd.Client, errChan),
-				NewPeerCollector(lnd.Client, errChan),
-				NewInfoCollector(lnd.Client, errChan),
-			},
-			htlcMonitor.collectors()...,
-		),
-		htlcMonitor: htlcMonitor,
-		errChan:     errChan,
+		collectors:    collectors,
+		htlcMonitor:   htlcMonitor,
+		errChan:       errChan,
 	}
 }
 

config.go

@@ -43,6 +43,9 @@ type config struct {
 
 	// PrimaryNode is the pubkey of the primary node in primary-gateway setups.
 	PrimaryNode string `long:"primarynode" description:"Public key of the primary node in a primary-gateway setup"`
+
+	// DisableGraph disables collection of graph metrics
+	DisableGraph bool `long:"disablegraph" description:"Do not collect graph metrics"`
 }
 
 var defaultConfig = config{

lndmon.go

@@ -56,7 +56,9 @@ func start() error {
 	}
 	defer lnd.Close()
 
-	monitoringCfg := collectors.MonitoringConfig{}
+	monitoringCfg := collectors.MonitoringConfig{
+		DisableGraph: cfg.DisableGraph,
+	}
 	if cfg.PrimaryNode != "" {
 		primaryNode, err := route.NewVertexFromStr(cfg.PrimaryNode)
 		if err != nil {