ClusterCockpit/cc-metric-collector
1
0
Fork 0
mirror of https://github.com/ClusterCockpit/cc-metric-collector.git synced 2024-12-25 15:09:05 +01:00
Code Issues Packages Projects Releases Wiki Activity

Add caching for condition evaluation

Browse Source
This commit is contained in:
Holger Obermaier 2022-02-16 13:59:32 +01:00
parent 248c815a1c
commit b44e226496
1 changed files with 29 additions and 53 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

68
internal/metricAggregator/metricAggregator.go
View File

@ -66,6 +66,7 @@ var language gval.Language = gval.NewLanguage(
gval.Full(),
metricCacheLanguage,
)
var evaluables map[string]gval.Evaluable = make(map[string]gval.Evaluable)
func (c *metricAggregator) Init(output chan lp.CCMetric) error {
c.output = output
@ -288,66 +289,41 @@ func (c *metricAggregator) AddFunction(name string, function func(args ...interf
}
func EvalBoolCondition(condition string, params map[string]interface{}) (bool, error) {
var evaluable gval.Evaluable
var ok bool
if evaluable, ok = evaluables[condition]; !ok {
newcond :=
strings.ReplaceAll(
strings.ReplaceAll(
condition, "'", "\""), "%", "\\")
value, err := gval.Evaluate(newcond, params, language)
var err error
evaluable, err = language.NewEvaluable(newcond)
if err != nil {
return false, err
}
endResult := false
err = nil
switch r := value.(type) {
case bool:
endResult = r
case float64:
endResult = r != 0.0
case float32:
endResult = r != 0.0
case int:
endResult = r != 0
case int64:
endResult = r != 0
case int32:
endResult = r != 0
default:
err = fmt.Errorf("cannot evaluate '%s' to bool", newcond)
evaluables[condition] = evaluable
}
return endResult, err
value, err := evaluable.EvalBool(context.Background(), params)
return value, err
}
func EvalFloat64Condition(condition string, params map[string]interface{}) (float64, error) {
var endResult float64 = math.NaN()
newcond := strings.ReplaceAll(condition, "'", "\"")
newcond = strings.ReplaceAll(newcond, "%", "\\")
value, err := gval.Evaluate(newcond, params, language)
var evaluable gval.Evaluable
var ok bool
if evaluable, ok = evaluables[condition]; !ok {
newcond :=
strings.ReplaceAll(
strings.ReplaceAll(
condition, "'", "\""), "%", "\\")
var err error
evaluable, err = language.NewEvaluable(newcond)
if err != nil {
cclog.ComponentDebug("MetricRouter", condition, " = ", err.Error())
return endResult, err
return math.NaN(), err
}
err = nil
switch r := value.(type) {
case bool:
if r {
endResult = 1.0
} else {
endResult = 0.0
evaluables[condition] = evaluable
}
case float64:
endResult = r
case float32:
endResult = float64(r)
case int:
endResult = float64(r)
case int64:
endResult = float64(r)
case int32:
endResult = float64(r)
default:
err = fmt.Errorf("cannot evaluate '%s' to float64", newcond)
}
return endResult, err
value, err := evaluable.EvalFloat64(context.Background(), params)
return value, err
}
func NewAggregator(output chan lp.CCMetric) (MetricAggregator, error) {