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

82
internal/metricAggregator/metricAggregator.go
View File

@ -66,6 +66,7 @@ var language gval.Language = gval.NewLanguage(
gval.Full(), gval.Full(),
metricCacheLanguage, metricCacheLanguage,
) )
var evaluables map[string]gval.Evaluable = make(map[string]gval.Evaluable)
func (c *metricAggregator) Init(output chan lp.CCMetric) error { func (c *metricAggregator) Init(output chan lp.CCMetric) error {
c.output = output 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) { func EvalBoolCondition(condition string, params map[string]interface{}) (bool, error) {
newcond := var evaluable gval.Evaluable
strings.ReplaceAll( var ok bool
if evaluable, ok = evaluables[condition]; !ok {
newcond :=
strings.ReplaceAll( strings.ReplaceAll(
condition, "'", "\""), "%", "\\") strings.ReplaceAll(
value, err := gval.Evaluate(newcond, params, language) condition, "'", "\""), "%", "\\")
if err != nil { var err error
return false, err evaluable, err = language.NewEvaluable(newcond)
if err != nil {
return false, err
}
evaluables[condition] = evaluable
} }
endResult := false value, err := evaluable.EvalBool(context.Background(), params)
err = nil return value, err
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)
}
return endResult, err
} }
func EvalFloat64Condition(condition string, params map[string]interface{}) (float64, error) { func EvalFloat64Condition(condition string, params map[string]interface{}) (float64, error) {
var endResult float64 = math.NaN() var evaluable gval.Evaluable
newcond := strings.ReplaceAll(condition, "'", "\"") var ok bool
newcond = strings.ReplaceAll(newcond, "%", "\\") if evaluable, ok = evaluables[condition]; !ok {
value, err := gval.Evaluate(newcond, params, language) newcond :=
if err != nil { strings.ReplaceAll(
cclog.ComponentDebug("MetricRouter", condition, " = ", err.Error()) strings.ReplaceAll(
return endResult, err condition, "'", "\""), "%", "\\")
} var err error
err = nil evaluable, err = language.NewEvaluable(newcond)
switch r := value.(type) { if err != nil {
case bool: return math.NaN(), err
if r {
endResult = 1.0
} else {
endResult = 0.0
} }
case float64: evaluables[condition] = evaluable
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) { func NewAggregator(output chan lp.CCMetric) (MetricAggregator, error) {