SQL Expressions: Change metric conversion to full long (#102728)
When querying metric data (non-table data) with SQL Expressions, we need to convert the data to table format. This is alternative format which does not have the same issues with sparse data. There is now a __metric_name__ column and one __value__ column. Also a __display_name__ column if there is DisplayNameFromDS metadata. --------- Co-authored-by: Adam Simpson <adam@adamsimpson.net>
This commit is contained in:
Kyle Brandt
co-authored by
Adam Simpson
Adam Simpson
parent
5dd0aa2c73
commit
f4849eabc7
@@ -0,0 +1,507 @@
|
||||
package expr
|
||||
|
||||
import (
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"github.com/google/go-cmp/cmp"
|
||||
"github.com/grafana/grafana-plugin-sdk-go/data"
|
||||
"github.com/stretchr/testify/require"
|
||||
)
|
||||
|
||||
func TestConvertNumericWideToFullLong(t *testing.T) {
|
||||
t.Run("SingleItemNoLabels", func(t *testing.T) {
|
||||
input := data.Frames{
|
||||
data.NewFrame("numeric",
|
||||
data.NewField("cpu", nil, []float64{3.14}),
|
||||
),
|
||||
}
|
||||
input[0].Meta = &data.FrameMeta{Type: data.FrameTypeNumericWide}
|
||||
|
||||
expected := data.NewFrame("",
|
||||
data.NewField(SQLMetricFieldName, nil, []string{"cpu"}),
|
||||
data.NewField(SQLValueFieldName, nil, []*float64{fp(3.14)}),
|
||||
)
|
||||
expected.Meta = &data.FrameMeta{Type: numericFullLongType}
|
||||
|
||||
output, err := ConvertToFullLong(input)
|
||||
require.NoError(t, err)
|
||||
require.Len(t, output, 1)
|
||||
if diff := cmp.Diff(expected, output[0], data.FrameTestCompareOptions()...); diff != "" {
|
||||
require.FailNowf(t, "Result mismatch (-want +got):%s", diff)
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("MultiRowShouldError", func(t *testing.T) {
|
||||
input := data.Frames{
|
||||
data.NewFrame("numeric",
|
||||
data.NewField("cpu", nil, []float64{1.0, 2.0}),
|
||||
),
|
||||
}
|
||||
input[0].Meta = &data.FrameMeta{Type: data.FrameTypeNumericWide}
|
||||
|
||||
_, err := ConvertToFullLong(input)
|
||||
require.Error(t, err)
|
||||
require.Contains(t, err.Error(), "no more than one row")
|
||||
})
|
||||
|
||||
t.Run("TwoItemsWithSingleLabel", func(t *testing.T) {
|
||||
input := data.Frames{
|
||||
data.NewFrame("numeric",
|
||||
data.NewField("cpu", data.Labels{"host": "a"}, []float64{1.0}),
|
||||
data.NewField("cpu", data.Labels{"host": "b"}, []float64{2.0}),
|
||||
),
|
||||
}
|
||||
input[0].Meta = &data.FrameMeta{Type: data.FrameTypeNumericWide}
|
||||
|
||||
expected := data.NewFrame("",
|
||||
data.NewField(SQLMetricFieldName, nil, []string{"cpu", "cpu"}),
|
||||
data.NewField(SQLValueFieldName, nil, []*float64{fp(1.0), fp(2.0)}),
|
||||
data.NewField("host", nil, []*string{sp("a"), sp("b")}),
|
||||
)
|
||||
expected.Meta = &data.FrameMeta{Type: numericFullLongType}
|
||||
|
||||
output, err := ConvertToFullLong(input)
|
||||
require.NoError(t, err)
|
||||
require.Len(t, output, 1)
|
||||
if diff := cmp.Diff(expected, output[0], data.FrameTestCompareOptions()...); diff != "" {
|
||||
require.FailNowf(t, "Result mismatch (-want +got):%s", diff)
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("TwoItemsWithSparseLabels", func(t *testing.T) {
|
||||
input := data.Frames{
|
||||
data.NewFrame("numeric",
|
||||
data.NewField("cpu", data.Labels{"host": "a"}, []float64{1.0}),
|
||||
data.NewField("cpu", data.Labels{"host": "b", "env": "prod"}, []float64{2.0}),
|
||||
),
|
||||
}
|
||||
input[0].Meta = &data.FrameMeta{Type: data.FrameTypeNumericWide}
|
||||
|
||||
expected := data.NewFrame("",
|
||||
data.NewField(SQLMetricFieldName, nil, []string{"cpu", "cpu"}),
|
||||
data.NewField(SQLValueFieldName, nil, []*float64{fp(1.0), fp(2.0)}),
|
||||
data.NewField("env", nil, []*string{nil, sp("prod")}),
|
||||
data.NewField("host", nil, []*string{sp("a"), sp("b")}),
|
||||
)
|
||||
expected.Meta = &data.FrameMeta{Type: numericFullLongType}
|
||||
|
||||
output, err := ConvertToFullLong(input)
|
||||
require.NoError(t, err)
|
||||
require.Len(t, output, 1)
|
||||
if diff := cmp.Diff(expected, output[0], data.FrameTestCompareOptions()...); diff != "" {
|
||||
require.FailNowf(t, "Result mismatch (-want +got):%s", diff)
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("TwoDifferentMetricsWithSharedLabels", func(t *testing.T) {
|
||||
input := data.Frames{
|
||||
data.NewFrame("numeric",
|
||||
data.NewField("cpu", data.Labels{"host": "a"}, []float64{1.0}),
|
||||
data.NewField("mem", data.Labels{"host": "a"}, []float64{4.0}),
|
||||
),
|
||||
}
|
||||
input[0].Meta = &data.FrameMeta{Type: data.FrameTypeNumericWide}
|
||||
|
||||
expected := data.NewFrame("",
|
||||
data.NewField(SQLMetricFieldName, nil, []string{"cpu", "mem"}),
|
||||
data.NewField(SQLValueFieldName, nil, []*float64{fp(1.0), fp(4.0)}),
|
||||
data.NewField("host", nil, []*string{sp("a"), sp("a")}),
|
||||
)
|
||||
expected.Meta = &data.FrameMeta{Type: numericFullLongType}
|
||||
|
||||
output, err := ConvertToFullLong(input)
|
||||
require.NoError(t, err)
|
||||
require.Len(t, output, 1)
|
||||
if diff := cmp.Diff(expected, output[0], data.FrameTestCompareOptions()...); diff != "" {
|
||||
require.FailNowf(t, "Result mismatch (-want +got):%s", diff)
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("TwoSparseMetricsAndLabels", func(t *testing.T) {
|
||||
input := data.Frames{
|
||||
data.NewFrame("numeric",
|
||||
data.NewField("cpu", data.Labels{"host": "a"}, []float64{1.0}),
|
||||
data.NewField("mem", data.Labels{"env": "prod"}, []float64{4.0}),
|
||||
),
|
||||
}
|
||||
input[0].Meta = &data.FrameMeta{Type: data.FrameTypeNumericWide}
|
||||
|
||||
expected := data.NewFrame("",
|
||||
data.NewField(SQLMetricFieldName, nil, []string{"cpu", "mem"}),
|
||||
data.NewField(SQLValueFieldName, nil, []*float64{fp(1.0), fp(4.0)}),
|
||||
data.NewField("env", nil, []*string{nil, sp("prod")}),
|
||||
data.NewField("host", nil, []*string{sp("a"), nil}),
|
||||
)
|
||||
expected.Meta = &data.FrameMeta{Type: numericFullLongType}
|
||||
|
||||
output, err := ConvertToFullLong(input)
|
||||
require.NoError(t, err)
|
||||
require.Len(t, output, 1)
|
||||
if diff := cmp.Diff(expected, output[0], data.FrameTestCompareOptions()...); diff != "" {
|
||||
require.FailNowf(t, "Result mismatch (-want +got):%s", diff)
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("ExtraTimeFieldIsDropped", func(t *testing.T) {
|
||||
// Note we may consider changing this behavior and looking into keeping
|
||||
// remainder fields in the future.
|
||||
input := data.Frames{
|
||||
data.NewFrame("numeric",
|
||||
data.NewField("timestamp", nil, []time.Time{time.Now()}), // extra time field
|
||||
data.NewField("cpu", nil, []float64{1.23}),
|
||||
),
|
||||
}
|
||||
input[0].Meta = &data.FrameMeta{Type: data.FrameTypeNumericWide}
|
||||
|
||||
expected := data.NewFrame("",
|
||||
data.NewField(SQLMetricFieldName, nil, []string{"cpu"}),
|
||||
data.NewField(SQLValueFieldName, nil, []*float64{fp(1.23)}),
|
||||
)
|
||||
expected.Meta = &data.FrameMeta{Type: numericFullLongType}
|
||||
|
||||
output, err := ConvertToFullLong(input)
|
||||
require.NoError(t, err)
|
||||
require.Len(t, output, 1)
|
||||
if diff := cmp.Diff(expected, output[0], data.FrameTestCompareOptions()...); diff != "" {
|
||||
require.FailNowf(t, "Result mismatch (-want +got):%s", diff)
|
||||
}
|
||||
})
|
||||
}
|
||||
|
||||
func TestConvertNumericWideToFullLongWithDisplayName(t *testing.T) {
|
||||
t.Run("SingleFieldWithDisplayName", func(t *testing.T) {
|
||||
input := data.Frames{
|
||||
data.NewFrame("numeric",
|
||||
func() *data.Field {
|
||||
f := data.NewField("cpu", nil, []float64{3.14})
|
||||
f.Config = &data.FieldConfig{DisplayNameFromDS: "CPU Display"}
|
||||
return f
|
||||
}(),
|
||||
),
|
||||
}
|
||||
input[0].Meta = &data.FrameMeta{Type: data.FrameTypeNumericWide}
|
||||
|
||||
expected := data.NewFrame("",
|
||||
data.NewField(SQLMetricFieldName, nil, []string{"cpu"}),
|
||||
data.NewField(SQLValueFieldName, nil, []*float64{fp(3.14)}),
|
||||
data.NewField(SQLDisplayFieldName, nil, []*string{sp("CPU Display")}),
|
||||
)
|
||||
expected.Meta = &data.FrameMeta{Type: numericFullLongType}
|
||||
|
||||
output, err := ConvertToFullLong(input)
|
||||
require.NoError(t, err)
|
||||
require.Len(t, output, 1)
|
||||
if diff := cmp.Diff(expected, output[0], data.FrameTestCompareOptions()...); diff != "" {
|
||||
require.FailNowf(t, "Result mismatch (-want +got):%s", diff)
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("MixedDisplayNames", func(t *testing.T) {
|
||||
input := data.Frames{
|
||||
data.NewFrame("numeric",
|
||||
func() *data.Field {
|
||||
f := data.NewField("cpu", data.Labels{"host": "a"}, []float64{1.0})
|
||||
f.Config = &data.FieldConfig{DisplayNameFromDS: "CPU A"}
|
||||
return f
|
||||
}(),
|
||||
data.NewField("cpu", data.Labels{"host": "b"}, []float64{2.0}),
|
||||
),
|
||||
}
|
||||
input[0].Meta = &data.FrameMeta{Type: data.FrameTypeNumericWide}
|
||||
|
||||
expected := data.NewFrame("",
|
||||
data.NewField(SQLMetricFieldName, nil, []string{"cpu", "cpu"}),
|
||||
data.NewField(SQLValueFieldName, nil, []*float64{fp(1.0), fp(2.0)}),
|
||||
data.NewField(SQLDisplayFieldName, nil, []*string{sp("CPU A"), nil}),
|
||||
data.NewField("host", nil, []*string{sp("a"), sp("b")}),
|
||||
)
|
||||
expected.Meta = &data.FrameMeta{Type: numericFullLongType}
|
||||
|
||||
output, err := ConvertToFullLong(input)
|
||||
require.NoError(t, err)
|
||||
require.Len(t, output, 1)
|
||||
if diff := cmp.Diff(expected, output[0], data.FrameTestCompareOptions()...); diff != "" {
|
||||
require.FailNowf(t, "Result mismatch (-want +got):%s", diff)
|
||||
}
|
||||
})
|
||||
}
|
||||
|
||||
func TestConvertNumericMultiToFullLong(t *testing.T) {
|
||||
t.Run("SingleItemNoLabels", func(t *testing.T) {
|
||||
input := data.Frames{
|
||||
data.NewFrame("",
|
||||
data.NewField("cpu", nil, []float64{3.14}),
|
||||
),
|
||||
}
|
||||
input[0].Meta = &data.FrameMeta{Type: data.FrameTypeNumericMulti}
|
||||
|
||||
expected := data.NewFrame("",
|
||||
data.NewField(SQLMetricFieldName, nil, []string{"cpu"}),
|
||||
data.NewField(SQLValueFieldName, nil, []*float64{fp(3.14)}),
|
||||
)
|
||||
expected.Meta = &data.FrameMeta{Type: numericFullLongType}
|
||||
|
||||
output, err := ConvertToFullLong(input)
|
||||
require.NoError(t, err)
|
||||
require.Len(t, output, 1)
|
||||
if diff := cmp.Diff(expected, output[0], data.FrameTestCompareOptions()...); diff != "" {
|
||||
require.FailNowf(t, "Result mismatch (-want +got):%s", diff)
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("TwoItemsWithSingleLabel", func(t *testing.T) {
|
||||
input := data.Frames{
|
||||
data.NewFrame("",
|
||||
data.NewField("cpu", data.Labels{"host": "a"}, []float64{1.0}),
|
||||
),
|
||||
data.NewFrame("",
|
||||
data.NewField("cpu", data.Labels{"host": "b"}, []float64{2.0}),
|
||||
),
|
||||
}
|
||||
for _, f := range input {
|
||||
f.Meta = &data.FrameMeta{Type: data.FrameTypeNumericMulti}
|
||||
}
|
||||
|
||||
expected := data.NewFrame("",
|
||||
data.NewField(SQLMetricFieldName, nil, []string{"cpu", "cpu"}),
|
||||
data.NewField(SQLValueFieldName, nil, []*float64{fp(1.0), fp(2.0)}),
|
||||
data.NewField("host", nil, []*string{sp("a"), sp("b")}),
|
||||
)
|
||||
expected.Meta = &data.FrameMeta{Type: numericFullLongType}
|
||||
|
||||
output, err := ConvertToFullLong(input)
|
||||
require.NoError(t, err)
|
||||
require.Len(t, output, 1)
|
||||
if diff := cmp.Diff(expected, output[0], data.FrameTestCompareOptions()...); diff != "" {
|
||||
require.FailNowf(t, "Result mismatch (-want +got):%s", diff)
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("TwoItemsWithSparseLabels", func(t *testing.T) {
|
||||
input := data.Frames{
|
||||
data.NewFrame("",
|
||||
data.NewField("cpu", data.Labels{"host": "a"}, []float64{1.0}),
|
||||
),
|
||||
data.NewFrame("",
|
||||
data.NewField("cpu", data.Labels{"host": "b", "env": "prod"}, []float64{2.0}),
|
||||
),
|
||||
}
|
||||
for _, f := range input {
|
||||
f.Meta = &data.FrameMeta{Type: data.FrameTypeNumericMulti}
|
||||
}
|
||||
|
||||
expected := data.NewFrame("",
|
||||
data.NewField(SQLMetricFieldName, nil, []string{"cpu", "cpu"}),
|
||||
data.NewField(SQLValueFieldName, nil, []*float64{fp(1.0), fp(2.0)}),
|
||||
data.NewField("env", nil, []*string{nil, sp("prod")}),
|
||||
data.NewField("host", nil, []*string{sp("a"), sp("b")}),
|
||||
)
|
||||
expected.Meta = &data.FrameMeta{Type: numericFullLongType}
|
||||
|
||||
output, err := ConvertToFullLong(input)
|
||||
require.NoError(t, err)
|
||||
require.Len(t, output, 1)
|
||||
if diff := cmp.Diff(expected, output[0], data.FrameTestCompareOptions()...); diff != "" {
|
||||
require.FailNowf(t, "Result mismatch (-want +got):%s", diff)
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("TwoDifferentMetricsWithSharedLabels", func(t *testing.T) {
|
||||
input := data.Frames{
|
||||
data.NewFrame("",
|
||||
data.NewField("cpu", data.Labels{"host": "a"}, []float64{1.0}),
|
||||
),
|
||||
data.NewFrame("",
|
||||
data.NewField("mem", data.Labels{"host": "a"}, []float64{4.0}),
|
||||
),
|
||||
}
|
||||
for _, f := range input {
|
||||
f.Meta = &data.FrameMeta{Type: data.FrameTypeNumericMulti}
|
||||
}
|
||||
|
||||
expected := data.NewFrame("",
|
||||
data.NewField(SQLMetricFieldName, nil, []string{"cpu", "mem"}),
|
||||
data.NewField(SQLValueFieldName, nil, []*float64{fp(1.0), fp(4.0)}),
|
||||
data.NewField("host", nil, []*string{sp("a"), sp("a")}),
|
||||
)
|
||||
expected.Meta = &data.FrameMeta{Type: numericFullLongType}
|
||||
|
||||
output, err := ConvertToFullLong(input)
|
||||
require.NoError(t, err)
|
||||
require.Len(t, output, 1)
|
||||
if diff := cmp.Diff(expected, output[0], data.FrameTestCompareOptions()...); diff != "" {
|
||||
require.FailNowf(t, "Result mismatch (-want +got):%s", diff)
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("TwoSparseMetricsAndLabels", func(t *testing.T) {
|
||||
input := data.Frames{
|
||||
data.NewFrame("",
|
||||
data.NewField("cpu", data.Labels{"host": "a"}, []float64{1.0}),
|
||||
),
|
||||
data.NewFrame("",
|
||||
data.NewField("mem", data.Labels{"env": "prod"}, []float64{4.0}),
|
||||
),
|
||||
}
|
||||
for _, f := range input {
|
||||
f.Meta = &data.FrameMeta{Type: data.FrameTypeNumericMulti}
|
||||
}
|
||||
|
||||
expected := data.NewFrame("",
|
||||
data.NewField(SQLMetricFieldName, nil, []string{"cpu", "mem"}),
|
||||
data.NewField(SQLValueFieldName, nil, []*float64{fp(1.0), fp(4.0)}),
|
||||
data.NewField("env", nil, []*string{nil, sp("prod")}),
|
||||
data.NewField("host", nil, []*string{sp("a"), nil}),
|
||||
)
|
||||
expected.Meta = &data.FrameMeta{Type: numericFullLongType}
|
||||
|
||||
output, err := ConvertToFullLong(input)
|
||||
require.NoError(t, err)
|
||||
require.Len(t, output, 1)
|
||||
if diff := cmp.Diff(expected, output[0], data.FrameTestCompareOptions()...); diff != "" {
|
||||
require.FailNowf(t, "Result mismatch (-want +got):%s", diff)
|
||||
}
|
||||
})
|
||||
}
|
||||
|
||||
func TestConvertTimeSeriesWideToFullLong(t *testing.T) {
|
||||
times := []time.Time{
|
||||
time.Unix(0, 0),
|
||||
time.Unix(10, 0),
|
||||
}
|
||||
|
||||
t.Run("SingleSeriesNoLabels", func(t *testing.T) {
|
||||
input := data.Frames{
|
||||
data.NewFrame("",
|
||||
data.NewField("time", nil, times),
|
||||
data.NewField("cpu", nil, []float64{1.0, 2.0}),
|
||||
),
|
||||
}
|
||||
input[0].Meta = &data.FrameMeta{Type: data.FrameTypeTimeSeriesWide}
|
||||
|
||||
expected := data.NewFrame("",
|
||||
data.NewField("time", nil, times),
|
||||
data.NewField(SQLValueFieldName, nil, []*float64{fp(1.0), fp(2.0)}),
|
||||
data.NewField(SQLMetricFieldName, nil, []string{"cpu", "cpu"}),
|
||||
)
|
||||
expected.Meta = &data.FrameMeta{Type: timeseriesFullLongType}
|
||||
|
||||
output, err := ConvertToFullLong(input)
|
||||
require.NoError(t, err)
|
||||
require.Len(t, output, 1)
|
||||
if diff := cmp.Diff(expected, output[0], data.FrameTestCompareOptions()...); diff != "" {
|
||||
require.FailNowf(t, "Mismatch (-want +got):\n%s", diff)
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("TwoSeriesOneLabel", func(t *testing.T) {
|
||||
input := data.Frames{
|
||||
data.NewFrame("",
|
||||
data.NewField("time", nil, times),
|
||||
data.NewField("cpu", data.Labels{"host": "a"}, []float64{1.0, 2.0}),
|
||||
data.NewField("cpu", data.Labels{"host": "b"}, []float64{3.0, 4.0}),
|
||||
),
|
||||
}
|
||||
input[0].Meta = &data.FrameMeta{Type: data.FrameTypeTimeSeriesWide}
|
||||
|
||||
expected := data.NewFrame("",
|
||||
data.NewField("time", nil, []time.Time{times[0], times[0], times[1], times[1]}),
|
||||
data.NewField(SQLValueFieldName, nil, []*float64{fp(1.0), fp(3.0), fp(2.0), fp(4.0)}),
|
||||
data.NewField(SQLMetricFieldName, nil, []string{"cpu", "cpu", "cpu", "cpu"}),
|
||||
data.NewField("host", nil, []*string{sp("a"), sp("b"), sp("a"), sp("b")}),
|
||||
)
|
||||
expected.Meta = &data.FrameMeta{Type: timeseriesFullLongType}
|
||||
|
||||
output, err := ConvertToFullLong(input)
|
||||
require.NoError(t, err)
|
||||
require.Len(t, output, 1)
|
||||
if diff := cmp.Diff(expected, output[0], data.FrameTestCompareOptions()...); diff != "" {
|
||||
require.FailNowf(t, "Mismatch (-want +got):\n%s", diff)
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("TwoMetricsWithSharedLabels", func(t *testing.T) {
|
||||
input := data.Frames{
|
||||
data.NewFrame("",
|
||||
data.NewField("time", nil, times),
|
||||
data.NewField("cpu", data.Labels{"host": "a"}, []float64{1.0, 2.0}),
|
||||
data.NewField("mem", data.Labels{"host": "a"}, []float64{3.0, 4.0}),
|
||||
),
|
||||
}
|
||||
input[0].Meta = &data.FrameMeta{Type: data.FrameTypeTimeSeriesWide}
|
||||
|
||||
expected := data.NewFrame("",
|
||||
data.NewField("time", nil, []time.Time{times[0], times[0], times[1], times[1]}),
|
||||
data.NewField(SQLValueFieldName, nil, []*float64{fp(1.0), fp(3.0), fp(2.0), fp(4.0)}),
|
||||
data.NewField(SQLMetricFieldName, nil, []string{"cpu", "mem", "cpu", "mem"}),
|
||||
data.NewField("host", nil, []*string{sp("a"), sp("a"), sp("a"), sp("a")}),
|
||||
)
|
||||
expected.Meta = &data.FrameMeta{Type: timeseriesFullLongType}
|
||||
|
||||
output, err := ConvertToFullLong(input)
|
||||
require.NoError(t, err)
|
||||
require.Len(t, output, 1)
|
||||
if diff := cmp.Diff(expected, output[0], data.FrameTestCompareOptions()...); diff != "" {
|
||||
require.FailNowf(t, "Mismatch (-want +got):\n%s", diff)
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("TwoSeriesSparseLabels", func(t *testing.T) {
|
||||
input := data.Frames{
|
||||
data.NewFrame("",
|
||||
data.NewField("time", nil, times),
|
||||
data.NewField("cpu", data.Labels{"host": "a"}, []float64{1.0, 2.0}),
|
||||
data.NewField("cpu", data.Labels{"host": "b", "env": "prod"}, []float64{3.0, 4.0}),
|
||||
),
|
||||
}
|
||||
input[0].Meta = &data.FrameMeta{Type: data.FrameTypeTimeSeriesWide}
|
||||
|
||||
expected := data.NewFrame("",
|
||||
data.NewField("time", nil, []time.Time{times[0], times[0], times[1], times[1]}),
|
||||
data.NewField(SQLValueFieldName, nil, []*float64{fp(1.0), fp(3.0), fp(2.0), fp(4.0)}),
|
||||
data.NewField(SQLMetricFieldName, nil, []string{"cpu", "cpu", "cpu", "cpu"}),
|
||||
data.NewField("env", nil, []*string{nil, sp("prod"), nil, sp("prod")}),
|
||||
data.NewField("host", nil, []*string{sp("a"), sp("b"), sp("a"), sp("b")}),
|
||||
)
|
||||
expected.Meta = &data.FrameMeta{Type: timeseriesFullLongType}
|
||||
|
||||
output, err := ConvertToFullLong(input)
|
||||
require.NoError(t, err)
|
||||
require.Len(t, output, 1)
|
||||
if diff := cmp.Diff(expected, output[0], data.FrameTestCompareOptions()...); diff != "" {
|
||||
require.FailNowf(t, "Mismatch (-want +got):\n%s", diff)
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("TwoSeriesSparseMetricsAndLabels", func(t *testing.T) {
|
||||
input := data.Frames{
|
||||
data.NewFrame("",
|
||||
data.NewField("time", nil, times),
|
||||
data.NewField("cpu", data.Labels{"host": "a"}, []float64{1.0, 2.0}),
|
||||
data.NewField("mem", data.Labels{"host": "b", "env": "prod"}, []float64{3.0, 4.0}),
|
||||
),
|
||||
}
|
||||
input[0].Meta = &data.FrameMeta{Type: data.FrameTypeTimeSeriesWide}
|
||||
|
||||
expected := data.NewFrame("",
|
||||
data.NewField("time", nil, []time.Time{times[0], times[0], times[1], times[1]}),
|
||||
data.NewField(SQLValueFieldName, nil, []*float64{fp(1.0), fp(3.0), fp(2.0), fp(4.0)}),
|
||||
data.NewField(SQLMetricFieldName, nil, []string{"cpu", "mem", "cpu", "mem"}),
|
||||
data.NewField("env", nil, []*string{nil, sp("prod"), nil, sp("prod")}),
|
||||
data.NewField("host", nil, []*string{sp("a"), sp("b"), sp("a"), sp("b")}),
|
||||
)
|
||||
expected.Meta = &data.FrameMeta{Type: timeseriesFullLongType}
|
||||
|
||||
output, err := ConvertToFullLong(input)
|
||||
require.NoError(t, err)
|
||||
require.Len(t, output, 1)
|
||||
if diff := cmp.Diff(expected, output[0], data.FrameTestCompareOptions()...); diff != "" {
|
||||
require.FailNowf(t, "Mismatch (-want +got):\n%s", diff)
|
||||
}
|
||||
})
|
||||
}
|
||||
|
||||
func sp(s string) *string {
|
||||
return &s
|
||||
}
|
||||
Reference in New Issue
Block a user