Simplify cache, calculate with expr

internal/metricRouter/metricCache_test.go

@@ -0,0 +1,69 @@
+// Copyright (C) NHR@FAU, University Erlangen-Nuremberg.
+// All rights reserved. This file is part of cc-lib.
+// Use of this source code is governed by a MIT-style
+// license that can be found in the LICENSE file.
+// additional authors:
+// Holger Obermaier (NHR@KIT)
+
+package metricRouter
+
+import (
+	"fmt"
+	"strings"
+	"sync"
+	"testing"
+	"time"
+
+	lp "github.com/ClusterCockpit/cc-lib/v2/ccMessage"
+	mct "github.com/ClusterCockpit/cc-metric-collector/pkg/multiChanTicker"
+)
+
+func TestCache(t *testing.T) {
+	output := make(chan lp.CCMessage, 2000)
+
+	var wg sync.WaitGroup
+	tickTime := time.Second
+	testChan := make(chan time.Time)
+	ticker := mct.NewTicker(tickTime)
+	ticker.AddChannel(testChan)
+	maxIntervals := 100
+
+	c, err := NewCache(output, ticker, &wg, 4)
+	if err != nil {
+		t.Errorf("failed to create new cache: %s", err.Error())
+		return
+	}
+	c.Start()
+	defer c.Close()
+	err = c.AddAggregation("ps_input_power", "Avg(values)", "name == 'ps1_input_power' || name == 'ps2_input_power' || name == 'ps3_input_power'", map[string]string{"hostname": "<copy>", "type": "<copy>"}, map[string]string{"unit": "<copy>"})
+	if err != nil {
+		t.Errorf("failed to add aggregation for %s", "ps_input_power")
+		return
+	}
+
+	for range maxIntervals {
+		timestamp := <-testChan
+		raw_metrics := []string{
+			fmt.Sprintf("ps1_input_power,type=node,hostname=myhost,unit=W value=696.0 %d", timestamp.UnixNano()),
+			fmt.Sprintf("ps2_input_power,type=node,hostname=myhost,unit=W value=732.0 %d", timestamp.UnixNano()),
+			fmt.Sprintf("ps3_input_power,type=node,hostname=myhost,unit=W value=720.0 %d", timestamp.UnixNano()),
+			fmt.Sprintf("cpu_load,type=hwthread,type-id=0,hostname=myhost value=45 %d", timestamp.UnixNano()),
+		}
+		metrics, err := lp.FromBytes([]byte(strings.Join(raw_metrics, "\n")))
+		if err != nil {
+			t.Errorf("failed to generate metrics: %s", err.Error())
+		}
+		for _, m := range metrics {
+			c.Add(m)
+		}
+
+		if len(output) > 0 {
+			for i := 0; i < len(output); i++ {
+				m := <-output
+				t.Log(m.ToLineProtocol(nil))
+			}
+		}
+
+	}
+
+}