(Feat): Initial Commit

internal/metrics/users/collector.go

@@ -0,0 +1,29 @@
+package users
+
+import (
+	"context"
+	"database/sql"
+
+	"allaboutapps.dev/aw/go-starter/internal/models"
+	"allaboutapps.dev/aw/go-starter/internal/util"
+)
+
+type DatabaseMetricsCollector struct {
+	db *sql.DB
+}
+
+func NewDatabaseMetricsCollector(db *sql.DB) *DatabaseMetricsCollector {
+	return &DatabaseMetricsCollector{db: db}
+}
+
+func (c DatabaseMetricsCollector) GetTotalUsersCount(ctx context.Context) float64 {
+	log := util.LogFromContext(ctx)
+
+	count, err := models.Users().Count(ctx, c.db)
+	if err != nil {
+		log.Error().Err(err).Msg("Failed to total user count")
+		return 0
+	}
+
+	return float64(count)
+}

internal/metrics/users/metrics.go

@@ -0,0 +1,27 @@
+package users
+
+import (
+	"context"
+
+	"github.com/prometheus/client_golang/prometheus"
+)
+
+type MetricsCollector interface {
+	GetTotalUsersCount(ctx context.Context) float64
+}
+
+const (
+	MetricNameTotalUsers = "total_users"
+)
+
+func Metrics(ctx context.Context, collector MetricsCollector) []prometheus.Collector {
+	return []prometheus.Collector{
+		prometheus.NewGaugeFunc(
+			prometheus.GaugeOpts{
+				Name: MetricNameTotalUsers,
+				Help: "Total users",
+			},
+			func() float64 { return collector.GetTotalUsersCount(ctx) },
+		),
+	}
+}