mirror of
https://github.com/slackhq/nebula.git
synced 2025-11-08 21:33:58 +01:00
a71541fb0b
This is how Prometheus recommends you do it, and how they do it themselves in their client. This makes it easy to see which versions you have deployed in your fleet, and query over it too.
111 lines
3.0 KiB
Go
111 lines
3.0 KiB
Go
package nebula
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"log"
|
|
"net"
|
|
"net/http"
|
|
"runtime"
|
|
"time"
|
|
|
|
graphite "github.com/cyberdelia/go-metrics-graphite"
|
|
mp "github.com/nbrownus/go-metrics-prometheus"
|
|
"github.com/prometheus/client_golang/prometheus"
|
|
"github.com/prometheus/client_golang/prometheus/promhttp"
|
|
"github.com/rcrowley/go-metrics"
|
|
"github.com/sirupsen/logrus"
|
|
)
|
|
|
|
func startStats(l *logrus.Logger, c *Config, buildVersion string, configTest bool) error {
|
|
mType := c.GetString("stats.type", "")
|
|
if mType == "" || mType == "none" {
|
|
return nil
|
|
}
|
|
|
|
interval := c.GetDuration("stats.interval", 0)
|
|
if interval == 0 {
|
|
return fmt.Errorf("stats.interval was an invalid duration: %s", c.GetString("stats.interval", ""))
|
|
}
|
|
|
|
switch mType {
|
|
case "graphite":
|
|
startGraphiteStats(l, interval, c, configTest)
|
|
case "prometheus":
|
|
startPrometheusStats(l, interval, c, buildVersion, configTest)
|
|
default:
|
|
return fmt.Errorf("stats.type was not understood: %s", mType)
|
|
}
|
|
|
|
metrics.RegisterDebugGCStats(metrics.DefaultRegistry)
|
|
metrics.RegisterRuntimeMemStats(metrics.DefaultRegistry)
|
|
|
|
go metrics.CaptureDebugGCStats(metrics.DefaultRegistry, interval)
|
|
go metrics.CaptureRuntimeMemStats(metrics.DefaultRegistry, interval)
|
|
|
|
return nil
|
|
}
|
|
|
|
func startGraphiteStats(l *logrus.Logger, i time.Duration, c *Config, configTest bool) error {
|
|
proto := c.GetString("stats.protocol", "tcp")
|
|
host := c.GetString("stats.host", "")
|
|
if host == "" {
|
|
return errors.New("stats.host can not be empty")
|
|
}
|
|
|
|
prefix := c.GetString("stats.prefix", "nebula")
|
|
addr, err := net.ResolveTCPAddr(proto, host)
|
|
if err != nil {
|
|
return fmt.Errorf("error while setting up graphite sink: %s", err)
|
|
}
|
|
|
|
l.Infof("Starting graphite. Interval: %s, prefix: %s, addr: %s", i, prefix, addr)
|
|
if !configTest {
|
|
go graphite.Graphite(metrics.DefaultRegistry, i, prefix, addr)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func startPrometheusStats(l *logrus.Logger, i time.Duration, c *Config, buildVersion string, configTest bool) error {
|
|
namespace := c.GetString("stats.namespace", "")
|
|
subsystem := c.GetString("stats.subsystem", "")
|
|
|
|
listen := c.GetString("stats.listen", "")
|
|
if listen == "" {
|
|
return fmt.Errorf("stats.listen should not be empty")
|
|
}
|
|
|
|
path := c.GetString("stats.path", "")
|
|
if path == "" {
|
|
return fmt.Errorf("stats.path should not be empty")
|
|
}
|
|
|
|
pr := prometheus.NewRegistry()
|
|
pClient := mp.NewPrometheusProvider(metrics.DefaultRegistry, namespace, subsystem, pr, i)
|
|
go pClient.UpdatePrometheusMetrics()
|
|
|
|
// Export our version information as labels on a static gauge
|
|
g := prometheus.NewGauge(prometheus.GaugeOpts{
|
|
Namespace: namespace,
|
|
Subsystem: subsystem,
|
|
Name: "info",
|
|
Help: "Version information for the Nebula binary",
|
|
ConstLabels: prometheus.Labels{
|
|
"version": buildVersion,
|
|
"goversion": runtime.Version(),
|
|
},
|
|
})
|
|
pr.MustRegister(g)
|
|
g.Set(1)
|
|
|
|
if !configTest {
|
|
go func() {
|
|
l.Infof("Prometheus stats listening on %s at %s", listen, path)
|
|
http.Handle(path, promhttp.HandlerFor(pr, promhttp.HandlerOpts{ErrorLog: l}))
|
|
log.Fatal(http.ListenAndServe(listen, nil))
|
|
}()
|
|
}
|
|
|
|
return nil
|
|
}
|