feat(metrics): add Prometheus metrics package for Victoria Metrics

Metrics exposed:
- sqs_requests_total{tenant, operation} — request counter
- sqs_request_bytes_total{tenant, operation} — traffic counter
- sqs_request_duration_seconds{operation} — latency histogram
- sqs_errors_total{operation} — error counter
- sqs_queues_count{tenant} — gauge, updated every 15s
- sqs_messages_count{tenant} — gauge, updated every 15s
This commit is contained in:
Naeel
2026-04-12 12:03:16 +03:00
parent aade320400
commit 3d59b5500c
4 changed files with 182 additions and 16 deletions
+62
View File
@@ -0,0 +1,62 @@
// app/metrics/gauge_updater.go
// Периодическое обновление gauge-метрик (количество очередей и сообщений per tenant).
// Запускается как горутина из main, обновляет каждые 15 секунд.
// Не блокирует SQS-запросы — читает через RLock.
// Created: 2026-04-12
package metrics
import (
"strings"
"time"
"shared-sqs/app/models"
log "github.com/sirupsen/logrus"
)
// StartGaugeUpdater — запускает горутину, которая каждые interval секунд
// пересчитывает количество очередей и сообщений по тенантам для Prometheus gauge.
// Останавливается при закрытии канала quit.
func StartGaugeUpdater(interval time.Duration, quit <-chan bool) {
go func() {
ticker := time.NewTicker(interval)
defer ticker.Stop()
for {
select {
case <-ticker.C:
updateGauges()
case <-quit:
log.Debug("metrics: gauge updater stopped")
return
}
}
}()
log.Infof("metrics: gauge updater started, interval=%s", interval)
}
// updateGauges — пересчитывает sqs_queues_count и sqs_messages_count per tenant.
// Формат ключа очереди: "{accessKey}:{queueName}" — берём часть до первого ":".
func updateGauges() {
// Собираем данные под RLock — не блокируем запись
models.SyncQueues.RLock()
tenantQueues := make(map[string]int)
tenantMessages := make(map[string]int)
for key, queue := range models.SyncQueues.Queues {
// Ключ формата "accessKey:queueName" — tenant определяется по accessKey
parts := strings.SplitN(key, ":", 2)
tenantKey := parts[0]
tenantQueues[tenantKey]++
tenantMessages[tenantKey] += len(queue.Messages)
}
models.SyncQueues.RUnlock()
// Сбрасываем старые значения и записываем новые
QueuesGauge.Reset()
MessagesGauge.Reset()
for tenant, count := range tenantQueues {
QueuesGauge.WithLabelValues(tenant).Set(float64(count))
}
for tenant, count := range tenantMessages {
MessagesGauge.WithLabelValues(tenant).Set(float64(count))
}
}
+71
View File
@@ -0,0 +1,71 @@
// app/metrics/metrics.go
// Prometheus-совместимые метрики shared-sqs для Victoria Metrics / Grafana.
// Отдаёт метрики на /metrics в стандартном формате Prometheus.
// VMAgent в кластере скрейпит этот endpoint через VMServiceScrape.
// Created: 2026-04-12
package metrics
import (
"github.com/prometheus/client_golang/prometheus"
"github.com/prometheus/client_golang/prometheus/promauto"
)
// RequestsTotal — счётчик SQS-запросов по тенанту и операции.
// Пример: sqs_requests_total{tenant="t-abc123", operation="SendMessage"} = 42
var RequestsTotal = promauto.NewCounterVec(
prometheus.CounterOpts{
Name: "sqs_requests_total",
Help: "Total number of SQS API requests by tenant and operation",
},
[]string{"tenant", "operation"},
)
// RequestBytesTotal — суммарный объём тел запросов (Content-Length) по тенанту и операции.
// Для биллинга по трафику.
var RequestBytesTotal = promauto.NewCounterVec(
prometheus.CounterOpts{
Name: "sqs_request_bytes_total",
Help: "Total request body bytes by tenant and operation",
},
[]string{"tenant", "operation"},
)
// RequestDuration — гистограмма latency SQS-операций в секундах.
// Бакеты подобраны для типичного SQS: от 1ms до 30s (long polling).
var RequestDuration = promauto.NewHistogramVec(
prometheus.HistogramOpts{
Name: "sqs_request_duration_seconds",
Help: "SQS API request duration in seconds",
Buckets: []float64{0.001, 0.005, 0.01, 0.025, 0.05, 0.1, 0.25, 0.5, 1, 2.5, 5, 10, 20, 30},
},
[]string{"operation"},
)
// ErrorsTotal — счётчик ошибочных SQS-ответов (HTTP >= 400) по операции.
var ErrorsTotal = promauto.NewCounterVec(
prometheus.CounterOpts{
Name: "sqs_errors_total",
Help: "Total number of SQS API error responses by operation",
},
[]string{"operation"},
)
// QueuesGauge — текущее количество очередей по тенанту.
// Обновляется периодически, не на каждый запрос.
var QueuesGauge = promauto.NewGaugeVec(
prometheus.GaugeOpts{
Name: "sqs_queues_count",
Help: "Current number of queues per tenant",
},
[]string{"tenant"},
)
// MessagesGauge — текущее количество сообщений по тенанту.
// Обновляется периодически, не на каждый запрос.
var MessagesGauge = promauto.NewGaugeVec(
prometheus.GaugeOpts{
Name: "sqs_messages_count",
Help: "Current number of messages per tenant",
},
[]string{"tenant"},
)