feat(metrics): Prometheus registry (checks/drift/notifications) + /metrics handler
This commit is contained in:
@@ -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()
|
||||
}
|
||||
@@ -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)
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user