Santiago
562cd9e44e
* Alerting: (wip) add template funcs * Alerting: (wip) numeric template functions * Alerting: (wip) template functions * Test for the "args" function * Alerting: (wip) Documentation for template functions * Alerting: template functions - refactor * code review changes * disable linter error * Use Prometheus implementation of TemplateExpander * Update docs/sources/alerting/unified-alerting/alerting-rules/create-grafana-managed-rule.md Co-authored-by: achatterjee-grafana <70489351+achatterjee-grafana@users.noreply.github.com> * change templateCaptureValue to support using template functions * Update pkg/services/ngalert/state/template.go Co-authored-by: gotjosh <josue.abreu@gmail.com> * Test and documentation added for reReplaceAll template function * complete missing functions, documentation and tests * Use the alert instance's evaluation time for expanding the template * strvalue graphlink and tablelink functions * delete duplicate test * make strvalue return an empty string Co-authored-by: achatterjee-grafana <70489351+achatterjee-grafana@users.noreply.github.com> Co-authored-by: gotjosh <josue.abreu@gmail.com>
90 lines
2.2 KiB
Go
90 lines
2.2 KiB
Go
package state
|
|
|
|
import (
|
|
"context"
|
|
"math"
|
|
"net/url"
|
|
"strconv"
|
|
"time"
|
|
|
|
text_template "text/template"
|
|
|
|
"github.com/grafana/grafana/pkg/services/ngalert/eval"
|
|
"github.com/prometheus/common/model"
|
|
"github.com/prometheus/prometheus/pkg/timestamp"
|
|
"github.com/prometheus/prometheus/promql"
|
|
"github.com/prometheus/prometheus/template"
|
|
)
|
|
|
|
// templateCaptureValue represents each value in .Values in the annotations
|
|
// and labels template.
|
|
type templateCaptureValue struct {
|
|
Labels map[string]string
|
|
Value float64
|
|
}
|
|
|
|
// String implements the Stringer interface to print the value of each RefID
|
|
// in the template via {{ $values.A }} rather than {{ $values.A.Value }}.
|
|
func (v templateCaptureValue) String() string {
|
|
return strconv.FormatFloat(v.Value, 'f', -1, 64)
|
|
}
|
|
|
|
func expandTemplate(name, text string, labels map[string]string, alertInstance eval.Result, externalURL *url.URL) (result string, resultErr error) {
|
|
name = "__alert_" + name
|
|
text = "{{- $labels := .Labels -}}{{- $values := .Values -}}{{- $value := .Value -}}" + text
|
|
data := struct {
|
|
Labels map[string]string
|
|
Values map[string]templateCaptureValue
|
|
Value string
|
|
}{
|
|
Labels: labels,
|
|
Values: newTemplateCaptureValues(alertInstance.Values),
|
|
Value: alertInstance.EvaluationString,
|
|
}
|
|
|
|
expander := template.NewTemplateExpander(
|
|
context.TODO(), // This context is only used with the `query()` function - which we don't support yet.
|
|
text,
|
|
name,
|
|
data,
|
|
model.Time(timestamp.FromTime(alertInstance.EvaluatedAt)),
|
|
func(context.Context, string, time.Time) (promql.Vector, error) {
|
|
return nil, nil
|
|
},
|
|
externalURL,
|
|
[]string{"missingkey=error"},
|
|
)
|
|
|
|
expander.Funcs(text_template.FuncMap{
|
|
// These three functions are no-ops for now.
|
|
"strvalue": func(value templateCaptureValue) string {
|
|
return ""
|
|
},
|
|
"graphLink": func() string {
|
|
return ""
|
|
},
|
|
"tableLink": func() string {
|
|
return ""
|
|
},
|
|
})
|
|
|
|
return expander.Expand()
|
|
}
|
|
|
|
func newTemplateCaptureValues(values map[string]eval.NumberValueCapture) map[string]templateCaptureValue {
|
|
m := make(map[string]templateCaptureValue)
|
|
for k, v := range values {
|
|
var f float64
|
|
if v.Value != nil {
|
|
f = *v.Value
|
|
} else {
|
|
f = math.NaN()
|
|
}
|
|
m[k] = templateCaptureValue{
|
|
Labels: v.Labels,
|
|
Value: f,
|
|
}
|
|
}
|
|
return m
|
|
}
|