Merge pull request #1 from eumel8/feat/metrics

WIP: init prometheus

api/api.go

@@ -39,6 +39,7 @@ import (
 	connect "github.com/bufbuild/connect-go"
 	grpc_zap "github.com/grpc-ecosystem/go-grpc-middleware/logging/zap"
 	"github.com/grpc-ecosystem/grpc-gateway/v2/runtime"
+	"github.com/prometheus/client_golang/prometheus/promhttp"
 	"github.com/telekom/canary-bot/data"
 	h "github.com/telekom/canary-bot/helper"
 	"github.com/telekom/canary-bot/proto/api/third_party"
@@ -133,6 +134,7 @@ func StartApi(data data.Database, config *Configuration, log *zap.SugaredLogger)
 
 	mux.Handle(apiv1connect.NewApiServiceHandler(a, interceptors))
 	mux.Handle("/api/v1/", gwmux)
+	mux.Handle("/metrics", promhttp.Handler())
 	server := &http.Server{
 		Addr:              addr,
 		Handler:           h2c.NewHandler(mux, &http2.Server{}),

main.go

@@ -33,12 +33,28 @@ import (
 	"github.com/spf13/cobra"
 	"github.com/spf13/pflag"
 	"github.com/spf13/viper"
+
+	"github.com/prometheus/client_golang/prometheus"
+	"github.com/prometheus/client_golang/prometheus/promauto"
 )
 
 const (
 	envPrefix = "MESH"
 )
 
+// prometheus metrics for nodes and samples and cluster
+
+var (
+	nodecounts = promauto.NewGauge(prometheus.GaugeOpts{
+		Name: "mesh_nodes",
+		Help: "The total number of nodes",
+	})
+	samplecounts = promauto.NewGauge(prometheus.GaugeOpts{
+		Name: "mesh_samples",
+		Help: "The total number of samples",
+	})
+)
+
 var cmd = &cobra.Command{
 	Use:   "cbot",
 	Short: "Canary Bot collecting & distributing sample data in a mesh",

mesh/mesh.go

@@ -145,6 +145,8 @@ func CreateCanaryMesh(routineConfig *RoutineConfiguration, setupConfig *SetupCon
 		CaCert:         setupConfig.CaCert,
 	}
 
+	// init prometheus metrics to fetch data from database
+
 	// start the mesh API
 	if err = api.StartApi(database, apiConfig, logger.Named("api")); err != nil {
 		logger.Fatal("Could not start API - Error: %+v", err)