kubernetes-sigs/prometheus-adapter
1
0
Fork 1
mirror of https://github.com/kubernetes-sigs/prometheus-adapter.git synced 2026-04-06 17:57:51 +00:00
Code Issues Projects Releases Packages Activity Actions

Add label value replacers to the config and query

Browse source
This commit is contained in:
Carson Anderson 2021-03-05 16:37:40 -07:00
parent 7e11fe30ee
commit 0536a5d481
5 changed files with 35 additions and 15 deletions
Download patch file Download diff file Expand all files Collapse all files

19
pkg/naming/metrics_query.go
View file

@ -28,6 +28,7 @@ import (
"k8s.io/apimachinery/pkg/selection"
prom "github.com/kubernetes-sigs/prometheus-adapter/pkg/client"
"github.com/kubernetes-sigs/prometheus-adapter/pkg/config"
)
// MetricsQuery represents a compiled metrics query for some set of
@ -50,15 +51,16 @@ type MetricsQuery interface {
// - LabelMatchersByName: the raw map-form of the above matchers
// - GroupBy: the group-by clause to use for the resources in the query (stringified)
// - GroupBySlice: the raw slice form of the above group-by clause
func NewMetricsQuery(queryTemplate string, resourceConverter ResourceConverter) (MetricsQuery, error) {
func NewMetricsQuery(queryTemplate string, resourceConverter ResourceConverter, labelValueReplacements []config.LabelValueReplacement) (MetricsQuery, error) {
templ, err := template.New("metrics-query").Delims("<<", ">>").Parse(queryTemplate)
if err != nil {
return nil, fmt.Errorf("unable to parse metrics query template %q: %v", queryTemplate, err)
}
return &metricsQuery{
resConverter: resourceConverter,
template: templ,
resConverter: resourceConverter,
template: templ,
labelValueReplacements: labelValueReplacements,
}, nil
}
@ -66,8 +68,9 @@ func NewMetricsQuery(queryTemplate string, resourceConverter ResourceConverter)
// with the delimiters as `<<` and `>>`, and the arguments found in
// queryTemplateArgs.
type metricsQuery struct {
resConverter ResourceConverter
template *template.Template
resConverter ResourceConverter
template *template.Template
labelValueReplacements []config.LabelValueReplacement
}
// queryTemplateArgs contains the arguments for the template used in metricsQuery.
@ -170,6 +173,12 @@ func (q *metricsQuery) BuildExternal(seriesName string, namespace string, groupB
return "", err
}
for _, repl := range q.labelValueReplacements {
for key, value := range valuesByName {
valuesByName[key] = strings.ReplaceAll(value, repl.Find, repl.Replace)
}
}
args := queryTemplateArgs{
Series: seriesName,
LabelMatchers: strings.Join(exprs, ","),