(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)
+}