github.com/go-graphite/carbonapi@v0.17.0/expr/functions/aliasSub/function_test.go (about)

     1  package aliasSub
     2  
     3  import (
     4  	"context"
     5  	"testing"
     6  	"time"
     7  
     8  	"github.com/go-graphite/carbonapi/expr/interfaces"
     9  	"github.com/go-graphite/carbonapi/expr/metadata"
    10  	"github.com/go-graphite/carbonapi/expr/types"
    11  	"github.com/go-graphite/carbonapi/pkg/parser"
    12  	th "github.com/go-graphite/carbonapi/tests"
    13  )
    14  
    15  var (
    16  	md []interfaces.FunctionMetadata = New("")
    17  )
    18  
    19  func init() {
    20  	for _, m := range md {
    21  		metadata.RegisterFunction(m.Name, m.F)
    22  	}
    23  }
    24  
    25  func TestAliasSub(t *testing.T) {
    26  	now32 := int64(time.Now().Unix())
    27  
    28  	tests := []th.EvalTestItem{
    29  		{
    30  			"aliasSub(metric1.foo.bar.baz, \"foo\", \"replaced\")",
    31  			map[parser.MetricRequest][]*types.MetricData{
    32  				{Metric: "metric1.foo.bar.baz", From: 0, Until: 1}: {types.MakeMetricData("metric1.foo.bar.baz", []float64{1, 2, 3, 4, 5}, 1, now32)},
    33  			},
    34  			[]*types.MetricData{types.MakeMetricData("metric1.replaced.bar.baz",
    35  				[]float64{1, 2, 3, 4, 5}, 1, now32).SetNameTag("metric1.foo.bar.baz")},
    36  		},
    37  		{
    38  			"aliasSub(metric1.TCP100,\"^.*TCP(\\d+)\",\"$1\")",
    39  			map[parser.MetricRequest][]*types.MetricData{
    40  				{Metric: "metric1.TCP100", From: 0, Until: 1}: {types.MakeMetricData("metric1.TCP100", []float64{1, 2, 3, 4, 5}, 1, now32)},
    41  			},
    42  			[]*types.MetricData{types.MakeMetricData("100",
    43  				[]float64{1, 2, 3, 4, 5}, 1, now32).SetNameTag("metric1.TCP100")},
    44  		},
    45  		{
    46  			"aliasSub(metric1.TCP100,\"^.*TCP(\\d+)\", \"\\1\")",
    47  			map[parser.MetricRequest][]*types.MetricData{
    48  				{Metric: "metric1.TCP100", From: 0, Until: 1}: {types.MakeMetricData("metric1.TCP100", []float64{1, 2, 3, 4, 5}, 1, now32)},
    49  			},
    50  			[]*types.MetricData{types.MakeMetricData("100",
    51  				[]float64{1, 2, 3, 4, 5}, 1, now32).SetNameTag("metric1.TCP100")},
    52  		},
    53  		{
    54  			"aliasSub(metric1.foo.bar.baz, \"foo\", \"replaced\")",
    55  			map[parser.MetricRequest][]*types.MetricData{
    56  				{Metric: "metric1.foo.bar.baz", From: 0, Until: 1}: {types.MakeMetricData("metric1.foo.bar.baz", []float64{1, 2, 3, 4, 5}, 1, now32)},
    57  			},
    58  			[]*types.MetricData{types.MakeMetricData("metric1.replaced.bar.baz",
    59  				[]float64{1, 2, 3, 4, 5}, 1, now32).SetNameTag("metric1.foo.bar.baz")},
    60  		},
    61  		// #290
    62  		{
    63  			//"aliasSub(*, '.dns.([^.]+).zone.', '\\1 diff to sql')",
    64  			"aliasSub(*, 'dns.([^.]*).zone.', '\\1 diff to sql ')",
    65  			map[parser.MetricRequest][]*types.MetricData{
    66  				{Metric: "*", From: 0, Until: 1}: {types.MakeMetricData("diffSeries(dns.snake.sql_updated, dns.snake.zone_updated)", []float64{1, 2, 3, 4, 5}, 1, now32)},
    67  			},
    68  			[]*types.MetricData{types.MakeMetricData("diffSeries(dns.snake.sql_updated, snake diff to sql updated)",
    69  				[]float64{1, 2, 3, 4, 5}, 1, now32).SetNameTag("dns.snake.sql_updated")},
    70  		},
    71  	}
    72  
    73  	for _, tt := range tests {
    74  		testName := tt.Target
    75  		t.Run(testName, func(t *testing.T) {
    76  			eval := th.EvaluatorFromFunc(md[0].F)
    77  			th.TestEvalExpr(t, eval, &tt)
    78  		})
    79  	}
    80  
    81  }
    82  
    83  func BenchmarkAverageAlias(b *testing.B) {
    84  	target := `aliasSub(metric1.TCP100,"^.*TCP(\\d+)","$1")`
    85  	metrics := map[parser.MetricRequest][]*types.MetricData{
    86  		{Metric: "metric1.TCP100", From: 0, Until: 1}:  {types.MakeMetricData("metric1.TCP100", []float64{1, 2, 3, 4, 5}, 1, 1)},
    87  		{Metric: "metric1.TCP1024", From: 0, Until: 1}: {types.MakeMetricData("metric1.TCP1024", []float64{1, 2, 3, 4, 5}, 1, 1)},
    88  	}
    89  
    90  	eval := th.EvaluatorFromFunc(md[0].F)
    91  	exp, _, err := parser.ParseExpr(target)
    92  	if err != nil {
    93  		b.Fatalf("failed to parse %s: %+v", target, err)
    94  	}
    95  
    96  	b.ResetTimer()
    97  	for n := 0; n < b.N; n++ {
    98  		g, err := eval.Eval(context.Background(), exp, 0, 1, metrics)
    99  		if err != nil {
   100  			b.Fatalf("failed to eval %s: %+v", target, err)
   101  		}
   102  		_ = g
   103  	}
   104  }