Use db

mtojek · mtojek · commit a6c9c1a3bdad · 2023-10-13T15:36:51.000+02:00
diff --git a/cli/server.go b/cli/server.go
@@ -199,7 +199,10 @@ func enablePrometheus(
 	}
 	afterCtx(ctx, closeWorkspacesFunc)
 
-	insightsMetricsCollector := insights.NewMetricsCollector(options.Database, 0)
+	insightsMetricsCollector, err := insights.NewMetricsCollector(options.Database, options.Logger, 0)
+	if err != nil {
+		return nil, xerrors.Errorf("unable to initialize insights metrics collector: %w", err)
+	}
 	err = options.PrometheusRegistry.Register(insightsMetricsCollector)
 	if err != nil {
 		return nil, xerrors.Errorf("unable to register insights metrics collector: %w", err)
diff --git a/coderd/prometheusmetrics/insights/metricscollector.go b/coderd/prometheusmetrics/insights/metricscollector.go
@@ -4,7 +4,9 @@ import (
 	"context"
 	"time"
 
+	"cdr.dev/slog"
 	"github.com/prometheus/client_golang/prometheus"
+	"golang.org/x/xerrors"
 
 	"github.com/coder/coder/v2/coderd/database"
 )
@@ -17,20 +19,25 @@ var (
 
 type MetricsCollector struct {
 	database database.Store
+	logger   slog.Logger
 	duration time.Duration
 }
 
 var _ prometheus.Collector = new(MetricsCollector)
 
-func NewMetricsCollector(db database.Store, duration time.Duration) *MetricsCollector {
+func NewMetricsCollector(db database.Store, logger slog.Logger, duration time.Duration) (*MetricsCollector, error) {
 	if duration == 0 {
 		duration = 5 * time.Minute
 	}
+	if duration < 5*time.Minute {
+		return nil, xerrors.Errorf("refresh interval must be at least 5 mins")
+	}
 
 	return &MetricsCollector{
 		database: db,
+		logger:   logger.Named("insights_metrics_collector"),
 		duration: duration,
-	}
+	}, nil
 }
 
 func (mc *MetricsCollector) Run(ctx context.Context) (func(), error) {
@@ -42,6 +49,18 @@ func (mc *MetricsCollector) Run(ctx context.Context) (func(), error) {
 	ticker := time.NewTicker(time.Nanosecond)
 	doTick := func() {
 		defer ticker.Reset(mc.duration)
+
+		now := time.Now()
+
+		parameterRows, err := mc.database.GetTemplateInsights(ctx, database.GetTemplateInsightsParams{
+			StartTime: now.Add(-mc.duration),
+			EndTime:   now,
+		})
+		if err != nil {
+			mc.logger.Error(ctx, "unable to fetch template insights from database: %w", err)
+			return
+		}
+
 	}
 
 	go func() {
