diff --git a/pkg/cache/metrics_middleware.go b/pkg/cache/metrics_middleware.go index a9104490c170a6bd608be4548d6dc7316590ef0a..2f3f6fe57836bf1b7430cfe7fef800344666a49e 100644 --- a/pkg/cache/metrics_middleware.go +++ b/pkg/cache/metrics_middleware.go @@ -8,26 +8,31 @@ import ( otelmetric "go.opentelemetry.io/otel/metric" ) -type metricsMiddleware struct { +type telemetryMiddleware struct { next Cache cacheMetrics *metrics.CacheMetrics attributes otelmetric.MeasurementOption } -// MetricsMiddleware возвращает обертку над кэшем, которая используется для отслеживания количества хитов и промахов в кэше. -func MetricsMiddleware(next Cache, cacheMetrics *metrics.CacheMetrics, keyValues ...attribute.KeyValue) Cache { - return &metricsMiddleware{ +// TelemetryMiddleware возвращает обертку над кэшем, которая используется для отслеживания количества хитов и промахов в кэше. +func TelemetryMiddleware(next Cache, scope string, keyValues ...attribute.KeyValue) Cache { + cacheMetrics, err := metrics.NewCacheMetrics(scope) + if err != nil { + panic(err) + } + + return &telemetryMiddleware{ next: next, cacheMetrics: cacheMetrics, attributes: otelmetric.WithAttributes(keyValues...), } } -func (c *metricsMiddleware) Set(key, value any) error { +func (c *telemetryMiddleware) Set(key, value any) error { return c.next.Set(key, value) } -func (c *metricsMiddleware) Get(key any) (any, error) { +func (c *telemetryMiddleware) Get(key any) (any, error) { value, err := c.next.Get(key) if err != nil { c.cacheMetrics.MissesTotal.Add(context.TODO(), 1, c.attributes) @@ -37,7 +42,7 @@ func (c *metricsMiddleware) Get(key any) (any, error) { return value, nil } -func (c *metricsMiddleware) Remove(key any) error { +func (c *telemetryMiddleware) Remove(key any) error { c.cacheMetrics.InvalidatesTotal.Add(context.TODO(), 1, c.attributes) return c.next.Remove(key) } diff --git a/pkg/metrics/cache.go b/pkg/metrics/cache.go index 11cec0bd479a8fde36f7662b5eaa025edeec57a6..b7e923826b1b8555a5b173b72990d3bbcbce5395 100644 --- a/pkg/metrics/cache.go +++ b/pkg/metrics/cache.go @@ -12,32 +12,29 @@ type CacheMetrics struct { } func NewCacheMetrics(scope string) (*CacheMetrics, error) { - meter := otel.Meter(scope) + var ( + meter = otel.Meter(scope) + cacheMetrics *CacheMetrics + err error + ) - hitsTotal, err := meter.Int64Counter("cache_hits", + if cacheMetrics.HitsTotal, err = meter.Int64Counter("cache_hits", otelmetric.WithDescription("Количество найденных в кэше значений"), - ) - if err != nil { + ); err != nil { return nil, err } - missesTotal, err := meter.Int64Counter("cache_misses", + if cacheMetrics.MissesTotal, err = meter.Int64Counter("cache_misses", otelmetric.WithDescription("Количество не найденных в кэше значений"), - ) - if err != nil { + ); err != nil { return nil, err } - invalidatesTotal, err := meter.Int64Counter("cache_invalidates", + if cacheMetrics.InvalidatesTotal, err = meter.Int64Counter("cache_invalidates", otelmetric.WithDescription("Количество инвалидаций кэша"), - ) - if err != nil { + ); err != nil { return nil, err } - return &CacheMetrics{ - HitsTotal: hitsTotal, - MissesTotal: missesTotal, - InvalidatesTotal: invalidatesTotal, - }, nil + return cacheMetrics, nil }