Export metrics for prometheus on localhost:9990/metrics

Co-authored-by: Jose Luis Lucas <[email protected]>
BBVA · Feb 19, 2019 · c74224e · c74224e
1 parent 3b4f382
commit c74224e
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 0 deletions.
diff --git a/server/config.go b/server/config.go
@@ -65,6 +65,9 @@ type Config struct {
 	// Enables tampering endpoint.
 	EnableTampering bool
 
+	// Enables metrics endpoint.
+	EnableMetrics bool
+
 	// Enable TLS service
 	EnableTLS bool
 
@@ -92,6 +95,7 @@ func DefaultConfig() *Config {
 		RaftPath:          currentDir + "/wal",
 		EnableProfiling:   false,
 		EnableTampering:   false,
+		EnableMetrics:     true,
 		EnableTLS:         false,
 		SSLCertificate:    "",
 		SSLCertificateKey: "",

diff --git a/server/server.go b/server/server.go
@@ -43,6 +43,7 @@ import (
 	"github.com/bbva/qed/sign"
 	"github.com/bbva/qed/storage/badger"
 	"github.com/bbva/qed/util"
+	"github.com/prometheus/client_golang/prometheus/promhttp"
 )
 
 // Server encapsulates the data and login to start/stop a QED server
@@ -164,6 +165,9 @@ func NewServer(conf *Config) (*Server, error) {
 	if conf.EnableProfiling {
 		server.profilingServer = newHTTPServer("localhost:6060", nil)
 	}
+	if conf.EnableMetrics {
+		http.Handle("/metrics", promhttp.Handler())
+	}
 
 	return server, nil
 }
@@ -194,6 +198,15 @@ func (s *Server) Start() error {
 		return err
 	}
 
+	if s.conf.EnableMetrics {
+		go func() {
+			log.Debugf("	* Starting metrics HTTP server in addr: localhost:9990")
+			if err := http.ListenAndServe("localhost:9990", nil); err != http.ErrServerClosed {
+				log.Errorf("Can't start metrics HTTP server: %s", err)
+			}
+		}()
+	}
+
 	if s.profilingServer != nil {
 		go func() {
 			log.Debugf("	* Starting profiling HTTP server in addr: localhost:6060")