Skip to content
Snippets Groups Projects
Commit b0b4312d authored by ensiouel's avatar ensiouel
Browse files

feature: функция argsToLabels была вынесена в пакет metrics и переименована в...

feature: функция argsToLabels была вынесена в пакет metrics и переименована в GetLabelsFromKV. Теперь функция игнорирует значение без ключа
parent 25361978
No related branches found
No related tags found
No related merge requests found
package cache
import (
"git.perx.ru/perxis/perxis-go/pkg/metrics"
"github.com/prometheus/client_golang/prometheus"
)
const badKey = "BADKEY"
type metricsMiddleware struct {
cache Cache
hitsTotal prometheus.Counter
......@@ -44,7 +43,7 @@ func MetricsMiddleware(cache Cache, subsystem string, labels ...string) Cache {
Help: "Количество инвалидаций кэша.",
}),
}
prometheus.WrapRegistererWith(argsToLabels(labels), prometheus.DefaultRegisterer).MustRegister(
prometheus.WrapRegistererWith(metrics.GetLabelsFromKV(labels), prometheus.DefaultRegisterer).MustRegister(
middleware.hitsTotal,
middleware.missesTotal,
middleware.invalidatesTotal,
......@@ -70,22 +69,3 @@ func (c *metricsMiddleware) Remove(key any) error {
c.invalidatesTotal.Inc()
return c.cache.Remove(key)
}
// argsToLabels преобразует массив строк args в метки типа prometheus.Labels.
//
// Функция ожидает, что каждое значение будет следовать за соответствующим ключом в массиве args,
// и возвращает метки, соответствующие парам ключ-значение.
func argsToLabels(args []string) prometheus.Labels {
labels := make(prometheus.Labels, len(args)/2)
for len(args) > 0 {
// если в массиве args остался только один элемент, он будет рассмотрен как значение с недопустимым ключом.
if len(args) == 1 {
labels[badKey] = args[0]
break
}
labels[args[0]] = args[1]
args = args[2:]
}
return labels
}
package metrics
import "github.com/prometheus/client_golang/prometheus"
// GetLabelsFromKV преобразует массив строк args в метки типа prometheus.Labels.
//
// Функция ожидает, что каждое значение будет следовать за соответствующим ключом в массиве args,
// и возвращает метки, соответствующие парам ключ-значение.
func GetLabelsFromKV(args []string) prometheus.Labels {
labels := make(prometheus.Labels, len(args)/2)
for len(args) > 0 {
// если в массиве args остался только один элемент, он будет проигнорирован
if len(args) == 1 {
break
}
labels[args[0]] = args[1]
args = args[2:]
}
return labels
}
package cache
package metrics
import (
"reflect"
......@@ -8,7 +8,7 @@ import (
"github.com/stretchr/testify/require"
)
func TestMetricsMiddleware_argsToLabels(t *testing.T) {
func TestArgsToLabels(t *testing.T) {
testcases := []struct {
name string
input []string
......@@ -37,18 +37,18 @@ func TestMetricsMiddleware_argsToLabels(t *testing.T) {
{
name: "bad key",
input: []string{"value"},
want: prometheus.Labels{badKey: "value"},
want: prometheus.Labels{},
},
{
name: "multi bad key",
input: []string{"key", "value", "value1"},
want: prometheus.Labels{"key": "value", badKey: "value1"},
want: prometheus.Labels{"key": "value"},
},
}
for _, tc := range testcases {
t.Run(tc.name, func(t *testing.T) {
got := argsToLabels(tc.input)
got := GetLabelsFromKV(tc.input)
require.True(t, reflect.DeepEqual(tc.want, got))
})
}
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment