feat(metrics): Prometheus registry (checks/drift/notifications) + /metrics handler

This commit is contained in:
2026-07-04 13:18:58 +07:00
parent 6fd847a909
commit 98d8dee413
4 changed files with 139 additions and 0 deletions
+74
View File
@@ -0,0 +1,74 @@
// Package metrics предоставляет Prometheus-метрики DNS Autoresolver.
package metrics
import (
"net/http"
"time"
"github.com/prometheus/client_golang/prometheus"
"github.com/prometheus/client_golang/prometheus/collectors"
"github.com/prometheus/client_golang/prometheus/promauto"
"github.com/prometheus/client_golang/prometheus/promhttp"
)
// Metrics агрегирует Prometheus-метрики приложения на собственном реестре.
type Metrics struct {
Registry *prometheus.Registry
ChecksTotal *prometheus.CounterVec
CheckDuration prometheus.Histogram
DriftDomains prometheus.Gauge
NotificationsTotal *prometheus.CounterVec
}
// New создаёт реестр метрик, регистрирует стандартные Go/Process-коллекторы
// и все метрики приложения.
func New() *Metrics {
reg := prometheus.NewRegistry()
reg.MustRegister(
collectors.NewGoCollector(),
collectors.NewProcessCollector(collectors.ProcessCollectorOpts{}),
)
f := promauto.With(reg)
return &Metrics{
Registry: reg,
ChecksTotal: f.NewCounterVec(prometheus.CounterOpts{
Name: "dns_ar_checks_total",
Help: "Общее количество выполненных проверок доменов по статусу.",
}, []string{"status"}),
CheckDuration: f.NewHistogram(prometheus.HistogramOpts{
Name: "dns_ar_check_duration_seconds",
Help: "Длительность выполнения проверки домена в секундах.",
}),
DriftDomains: f.NewGauge(prometheus.GaugeOpts{
Name: "dns_ar_drift_domains",
Help: "Текущее количество доменов в состоянии drift.",
}),
NotificationsTotal: f.NewCounterVec(prometheus.CounterOpts{
Name: "dns_ar_notifications_total",
Help: "Общее количество отправленных уведомлений по каналу и статусу.",
}, []string{"channel", "status"}),
}
}
// Handler возвращает HTTP-обработчик для отдачи метрик реестра.
func (m *Metrics) Handler() http.Handler {
return promhttp.HandlerFor(m.Registry, promhttp.HandlerOpts{})
}
// ObserveCheck фиксирует результат проверки: статус и длительность.
func (m *Metrics) ObserveCheck(status string, dur time.Duration) {
m.ChecksTotal.WithLabelValues(status).Inc()
m.CheckDuration.Observe(dur.Seconds())
}
// SetDrift устанавливает текущее количество доменов в состоянии drift.
func (m *Metrics) SetDrift(n int) {
m.DriftDomains.Set(float64(n))
}
// IncNotification фиксирует отправку уведомления по каналу и статусу.
func (m *Metrics) IncNotification(channel, status string) {
m.NotificationsTotal.WithLabelValues(channel, status).Inc()
}
+38
View File
@@ -0,0 +1,38 @@
package metrics
import (
"net/http/httptest"
"strings"
"testing"
"time"
"github.com/prometheus/client_golang/prometheus/testutil"
)
func TestMetricsRecord(t *testing.T) {
m := New()
m.ObserveCheck("drift", 100*time.Millisecond)
m.ObserveCheck("in_sync", 50*time.Millisecond)
m.IncNotification("telegram", "ok")
m.SetDrift(3)
if got := testutil.ToFloat64(m.ChecksTotal.WithLabelValues("drift")); got != 1 {
t.Fatalf("checks drift = %v", got)
}
if got := testutil.ToFloat64(m.DriftDomains); got != 3 {
t.Fatalf("drift gauge = %v", got)
}
if got := testutil.ToFloat64(m.NotificationsTotal.WithLabelValues("telegram", "ok")); got != 1 {
t.Fatalf("notif = %v", got)
}
}
func TestHandlerExposesMetrics(t *testing.T) {
m := New()
m.ObserveCheck("in_sync", time.Millisecond)
rec := httptest.NewRecorder()
m.Handler().ServeHTTP(rec, httptest.NewRequest("GET", "/metrics", nil))
if rec.Code != 200 || !strings.Contains(rec.Body.String(), "dns_ar_checks_total") {
t.Fatalf("metrics not exposed: %d", rec.Code)
}
}