mirror of https://github.com/grafana/grafana
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>pull/102751/head
Kyle Brandt
3 months ago
committed by
GitHub
parent
5dd0aa2c73
commit
f4849eabc7
@ -0,0 +1,128 @@ |
||||
package expr |
||||
|
||||
import ( |
||||
"fmt" |
||||
|
||||
"github.com/grafana/grafana-plugin-sdk-go/data" |
||||
) |
||||
|
||||
func ConvertFromFullLongToNumericMulti(frames data.Frames) (data.Frames, error) { |
||||
if len(frames) != 1 { |
||||
return nil, fmt.Errorf("expected exactly one frame, got %d", len(frames)) |
||||
} |
||||
frame := frames[0] |
||||
if frame.Meta == nil || frame.Meta.Type != numericFullLongType { |
||||
return nil, fmt.Errorf("expected frame of type %q", numericFullLongType) |
||||
} |
||||
|
||||
var ( |
||||
metricField *data.Field |
||||
valueField *data.Field |
||||
displayField *data.Field |
||||
labelFields []*data.Field |
||||
) |
||||
|
||||
// Identify key fields
|
||||
for _, f := range frame.Fields { |
||||
switch f.Name { |
||||
case SQLMetricFieldName: |
||||
metricField = f |
||||
case SQLValueFieldName: |
||||
valueField = f |
||||
case SQLDisplayFieldName: |
||||
displayField = f |
||||
default: |
||||
if f.Type() == data.FieldTypeNullableString { |
||||
labelFields = append(labelFields, f) |
||||
} |
||||
} |
||||
} |
||||
|
||||
if metricField == nil || valueField == nil { |
||||
return nil, fmt.Errorf("missing required fields: %q or %q", SQLMetricFieldName, SQLValueFieldName) |
||||
} |
||||
|
||||
type seriesKey struct { |
||||
metric string |
||||
labelFP data.Fingerprint |
||||
displayName string |
||||
} |
||||
|
||||
type seriesEntry struct { |
||||
indices []int |
||||
labels data.Labels |
||||
displayName *string |
||||
} |
||||
|
||||
grouped := make(map[seriesKey]*seriesEntry) |
||||
|
||||
for i := 0; i < frame.Rows(); i++ { |
||||
if valueField.NilAt(i) { |
||||
continue // skip null values
|
||||
} |
||||
|
||||
metric := metricField.At(i).(string) |
||||
|
||||
// collect labels
|
||||
labels := data.Labels{} |
||||
for _, f := range labelFields { |
||||
if f.NilAt(i) { |
||||
continue |
||||
} |
||||
val := f.At(i).(*string) |
||||
if val != nil { |
||||
labels[f.Name] = *val |
||||
} |
||||
} |
||||
fp := labels.Fingerprint() |
||||
|
||||
// handle optional display name
|
||||
var displayPtr *string |
||||
displayKey := "" |
||||
if displayField != nil && !displayField.NilAt(i) { |
||||
if raw := displayField.At(i).(*string); raw != nil { |
||||
displayPtr = raw |
||||
displayKey = *raw |
||||
} |
||||
} |
||||
|
||||
key := seriesKey{ |
||||
metric: metric, |
||||
labelFP: fp, |
||||
displayName: displayKey, |
||||
} |
||||
|
||||
entry, ok := grouped[key] |
||||
if !ok { |
||||
entry = &seriesEntry{ |
||||
labels: labels, |
||||
displayName: displayPtr, |
||||
} |
||||
grouped[key] = entry |
||||
} |
||||
entry.indices = append(entry.indices, i) |
||||
} |
||||
|
||||
var result data.Frames |
||||
for key, entry := range grouped { |
||||
values := make([]*float64, 0, len(entry.indices)) |
||||
for _, i := range entry.indices { |
||||
v, err := valueField.FloatAt(i) |
||||
if err != nil { |
||||
return nil, fmt.Errorf("failed to convert value at index %d to float: %w", i, err) |
||||
} |
||||
values = append(values, &v) |
||||
} |
||||
|
||||
field := data.NewField(key.metric, entry.labels, values) |
||||
if entry.displayName != nil { |
||||
field.Config = &data.FieldConfig{DisplayNameFromDS: *entry.displayName} |
||||
} |
||||
|
||||
frame := data.NewFrame("", field) |
||||
frame.Meta = &data.FrameMeta{Type: data.FrameTypeNumericMulti} |
||||
result = append(result, frame) |
||||
} |
||||
|
||||
return result, nil |
||||
} |
||||
@ -0,0 +1,192 @@ |
||||
package expr |
||||
|
||||
import ( |
||||
"sort" |
||||
"testing" |
||||
|
||||
"github.com/google/go-cmp/cmp" |
||||
"github.com/grafana/grafana-plugin-sdk-go/data" |
||||
"github.com/stretchr/testify/require" |
||||
) |
||||
|
||||
func TestConvertFromFullLongToNumericMulti(t *testing.T) { |
||||
t.Run("SingleRowNoLabels", func(t *testing.T) { |
||||
input := data.NewFrame("", |
||||
data.NewField(SQLMetricFieldName, nil, []string{"cpu"}), |
||||
data.NewField(SQLValueFieldName, nil, []*float64{fp(3.14)}), |
||||
) |
||||
input.Meta = &data.FrameMeta{Type: numericFullLongType} |
||||
|
||||
out, err := ConvertFromFullLongToNumericMulti(data.Frames{input}) |
||||
require.NoError(t, err) |
||||
require.Len(t, out, 1) |
||||
|
||||
expected := data.NewFrame("", |
||||
data.NewField("cpu", nil, []*float64{fp(3.14)}), |
||||
) |
||||
expected.Meta = &data.FrameMeta{Type: data.FrameTypeNumericMulti} |
||||
|
||||
if diff := cmp.Diff(expected, out[0], data.FrameTestCompareOptions()...); diff != "" { |
||||
require.FailNowf(t, "Mismatch (-want +got):\n%s", diff) |
||||
} |
||||
}) |
||||
|
||||
t.Run("TwoRowsWithLabelsAndDisplay", func(t *testing.T) { |
||||
input := 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"), sp("CPU A")}), |
||||
data.NewField("host", nil, []*string{sp("a"), sp("a")}), |
||||
) |
||||
input.Meta = &data.FrameMeta{Type: numericFullLongType} |
||||
|
||||
out, err := ConvertFromFullLongToNumericMulti(data.Frames{input}) |
||||
require.NoError(t, err) |
||||
require.Len(t, out, 1) |
||||
|
||||
expected := data.NewFrame("", |
||||
func() *data.Field { |
||||
f := data.NewField("cpu", data.Labels{"host": "a"}, []*float64{fp(1.0), fp(2.0)}) |
||||
f.Config = &data.FieldConfig{DisplayNameFromDS: "CPU A"} |
||||
return f |
||||
}(), |
||||
) |
||||
expected.Meta = &data.FrameMeta{Type: data.FrameTypeNumericMulti} |
||||
|
||||
if diff := cmp.Diff(expected, out[0], data.FrameTestCompareOptions()...); diff != "" { |
||||
require.FailNowf(t, "Mismatch (-want +got):\n%s", diff) |
||||
} |
||||
}) |
||||
|
||||
t.Run("SkipsNullValues", func(t *testing.T) { |
||||
input := data.NewFrame("", |
||||
data.NewField(SQLMetricFieldName, nil, []string{"cpu", "cpu"}), |
||||
data.NewField(SQLValueFieldName, nil, []*float64{fp(1.0), nil}), |
||||
) |
||||
input.Meta = &data.FrameMeta{Type: numericFullLongType} |
||||
|
||||
out, err := ConvertFromFullLongToNumericMulti(data.Frames{input}) |
||||
require.NoError(t, err) |
||||
require.Len(t, out, 1) |
||||
|
||||
expected := data.NewFrame("", |
||||
data.NewField("cpu", nil, []*float64{fp(1.0)}), |
||||
) |
||||
expected.Meta = &data.FrameMeta{Type: data.FrameTypeNumericMulti} |
||||
|
||||
if diff := cmp.Diff(expected, out[0], data.FrameTestCompareOptions()...); diff != "" { |
||||
require.FailNowf(t, "Mismatch (-want +got):\n%s", diff) |
||||
} |
||||
}) |
||||
} |
||||
|
||||
func TestConvertNumericMultiRoundTripToFullLongAndBack(t *testing.T) { |
||||
t.Run("TwoFieldsWithSparseLabels", func(t *testing.T) { |
||||
input := data.Frames{ |
||||
data.NewFrame("", |
||||
data.NewField("cpu", data.Labels{"host": "a"}, []*float64{fp(1.0)}), |
||||
), |
||||
data.NewFrame("", |
||||
data.NewField("cpu", data.Labels{"host": "b", "env": "prod"}, []*float64{fp(2.0)}), |
||||
), |
||||
} |
||||
for _, f := range input { |
||||
f.Meta = &data.FrameMeta{Type: data.FrameTypeNumericMulti} |
||||
} |
||||
|
||||
fullLong, err := ConvertToFullLong(input) |
||||
require.NoError(t, err) |
||||
require.Len(t, fullLong, 1) |
||||
|
||||
roundTrip, err := ConvertFromFullLongToNumericMulti(fullLong) |
||||
require.NoError(t, err) |
||||
|
||||
expected := data.Frames{ |
||||
data.NewFrame("", |
||||
data.NewField("cpu", data.Labels{"host": "a"}, []*float64{fp(1.0)}), |
||||
), |
||||
data.NewFrame("", |
||||
data.NewField("cpu", data.Labels{"host": "b", "env": "prod"}, []*float64{fp(2.0)}), |
||||
), |
||||
} |
||||
for _, f := range expected { |
||||
f.Meta = &data.FrameMeta{Type: data.FrameTypeNumericMulti} |
||||
} |
||||
|
||||
sortFramesByMetricDisplayAndLabels(expected) |
||||
sortFramesByMetricDisplayAndLabels(roundTrip) |
||||
|
||||
require.Len(t, roundTrip, len(expected)) |
||||
for i := range expected { |
||||
if diff := cmp.Diff(expected[i], roundTrip[i], data.FrameTestCompareOptions()...); diff != "" { |
||||
t.Errorf("Mismatch on frame %d (-want +got):\n%s", i, diff) |
||||
} |
||||
} |
||||
}) |
||||
|
||||
t.Run("PreservesDisplayName", func(t *testing.T) { |
||||
input := data.Frames{ |
||||
data.NewFrame("", |
||||
func() *data.Field { |
||||
f := data.NewField("cpu", data.Labels{"host": "a"}, []*float64{fp(1.0)}) |
||||
f.Config = &data.FieldConfig{DisplayNameFromDS: "CPU A"} |
||||
return f |
||||
}(), |
||||
), |
||||
} |
||||
input[0].Meta = &data.FrameMeta{Type: data.FrameTypeNumericMulti} |
||||
|
||||
fullLong, err := ConvertToFullLong(input) |
||||
require.NoError(t, err) |
||||
require.Len(t, fullLong, 1) |
||||
|
||||
roundTrip, err := ConvertFromFullLongToNumericMulti(fullLong) |
||||
require.NoError(t, err) |
||||
|
||||
expected := data.Frames{ |
||||
data.NewFrame("", |
||||
func() *data.Field { |
||||
f := data.NewField("cpu", data.Labels{"host": "a"}, []*float64{fp(1.0)}) |
||||
f.Config = &data.FieldConfig{DisplayNameFromDS: "CPU A"} |
||||
return f |
||||
}(), |
||||
), |
||||
} |
||||
expected[0].Meta = &data.FrameMeta{Type: data.FrameTypeNumericMulti} |
||||
|
||||
sortFramesByMetricDisplayAndLabels(expected) |
||||
sortFramesByMetricDisplayAndLabels(roundTrip) |
||||
|
||||
require.Len(t, roundTrip, 1) |
||||
if diff := cmp.Diff(expected[0], roundTrip[0], data.FrameTestCompareOptions()...); diff != "" { |
||||
t.Errorf("Mismatch (-want +got):\n%s", diff) |
||||
} |
||||
}) |
||||
} |
||||
|
||||
func sortFramesByMetricDisplayAndLabels(frames data.Frames) { |
||||
sort.Slice(frames, func(i, j int) bool { |
||||
fi := frames[i].Fields[0] |
||||
fj := frames[j].Fields[0] |
||||
|
||||
// 1. Metric name
|
||||
if fi.Name != fj.Name { |
||||
return fi.Name < fj.Name |
||||
} |
||||
|
||||
// 2. Display name (if set)
|
||||
var di, dj string |
||||
if fi.Config != nil { |
||||
di = fi.Config.DisplayNameFromDS |
||||
} |
||||
if fj.Config != nil { |
||||
dj = fj.Config.DisplayNameFromDS |
||||
} |
||||
if di != dj { |
||||
return di < dj |
||||
} |
||||
|
||||
// 3. Labels fingerprint
|
||||
return fi.Labels.Fingerprint() < fj.Labels.Fingerprint() |
||||
}) |
||||
} |
||||
@ -0,0 +1,400 @@ |
||||
package expr |
||||
|
||||
import ( |
||||
"fmt" |
||||
"sort" |
||||
"time" |
||||
|
||||
"github.com/grafana/grafana-plugin-sdk-go/data" |
||||
) |
||||
|
||||
const ( |
||||
SQLMetricFieldName = "__metric_name__" |
||||
SQLValueFieldName = "__value__" |
||||
SQLDisplayFieldName = "__display_name__" |
||||
|
||||
// These are not types in the SDK or dataplane contract yet.
|
||||
numericFullLongType = "numeric_full_long" |
||||
timeseriesFullLongType = "time_series_full_long" |
||||
) |
||||
|
||||
func ConvertToFullLong(frames data.Frames) (data.Frames, error) { |
||||
if len(frames) == 0 { |
||||
return frames, nil |
||||
} |
||||
|
||||
var inputType data.FrameType |
||||
if frames[0].Meta != nil && frames[0].Meta.Type != "" { |
||||
inputType = frames[0].Meta.Type |
||||
} else { |
||||
return nil, fmt.Errorf("input frame missing FrameMeta.Type") |
||||
} |
||||
|
||||
if !supportedToLongConversion(inputType) { |
||||
return nil, fmt.Errorf("unsupported input dataframe type %s for full long conversion", inputType) |
||||
} |
||||
|
||||
switch inputType { |
||||
case data.FrameTypeNumericMulti: |
||||
return convertNumericMultiToFullLong(frames) |
||||
case data.FrameTypeNumericWide: |
||||
return convertNumericWideToFullLong(frames) |
||||
case data.FrameTypeTimeSeriesMulti: |
||||
return convertTimeSeriesMultiToFullLong(frames) |
||||
case data.FrameTypeTimeSeriesWide: |
||||
return convertTimeSeriesWideToFullLong(frames) |
||||
default: |
||||
return nil, fmt.Errorf("unsupported input type %s for full long conversion", inputType) |
||||
} |
||||
} |
||||
|
||||
func convertNumericMultiToFullLong(frames data.Frames) (data.Frames, error) { |
||||
wide := convertNumericMultiToNumericWide(frames) |
||||
return convertNumericWideToFullLong(wide) |
||||
} |
||||
|
||||
func convertNumericWideToFullLong(frames data.Frames) (data.Frames, error) { |
||||
if len(frames) != 1 { |
||||
return nil, fmt.Errorf("expected exactly one frame for wide format, but got %d", len(frames)) |
||||
} |
||||
inputFrame := frames[0] |
||||
if inputFrame.Rows() > 1 { |
||||
return nil, fmt.Errorf("expected no more than one row in the frame, but got %d", inputFrame.Rows()) |
||||
} |
||||
|
||||
var ( |
||||
metricCol = make([]string, 0, len(inputFrame.Fields)) |
||||
valueCol = make([]*float64, 0, len(inputFrame.Fields)) |
||||
displayCol = make([]*string, 0, len(inputFrame.Fields)) |
||||
hasDisplayCol bool |
||||
) |
||||
|
||||
labelKeySet := map[string]struct{}{} |
||||
for _, field := range inputFrame.Fields { |
||||
if !field.Type().Numeric() { |
||||
continue |
||||
} |
||||
val, err := field.FloatAt(0) |
||||
if err != nil { |
||||
continue |
||||
} |
||||
v := val |
||||
valueCol = append(valueCol, &v) |
||||
metricCol = append(metricCol, field.Name) |
||||
|
||||
// Display name
|
||||
var d *string |
||||
if field.Config != nil && field.Config.DisplayNameFromDS != "" { |
||||
s := field.Config.DisplayNameFromDS |
||||
d = &s |
||||
hasDisplayCol = true |
||||
} |
||||
displayCol = append(displayCol, d) |
||||
|
||||
for k := range field.Labels { |
||||
labelKeySet[k] = struct{}{} |
||||
} |
||||
} |
||||
|
||||
labelKeys := make([]string, 0, len(labelKeySet)) |
||||
|
||||
labelValues := make(map[string][]*string) |
||||
for k := range labelKeySet { |
||||
labelKeys = append(labelKeys, k) |
||||
labelValues[k] = make([]*string, 0, len(valueCol)) |
||||
} |
||||
sort.Strings(labelKeys) |
||||
|
||||
for _, field := range inputFrame.Fields { |
||||
if !field.Type().Numeric() { |
||||
continue |
||||
} |
||||
for _, k := range labelKeys { |
||||
var val *string |
||||
if field.Labels != nil { |
||||
if v, ok := field.Labels[k]; ok { |
||||
val = &v |
||||
} |
||||
} |
||||
labelValues[k] = append(labelValues[k], val) |
||||
} |
||||
} |
||||
|
||||
fields := []*data.Field{ |
||||
data.NewField(SQLMetricFieldName, nil, metricCol), |
||||
data.NewField(SQLValueFieldName, nil, valueCol), |
||||
} |
||||
if hasDisplayCol { |
||||
fields = append(fields, data.NewField(SQLDisplayFieldName, nil, displayCol)) |
||||
} |
||||
for _, k := range labelKeys { |
||||
fields = append(fields, data.NewField(k, nil, labelValues[k])) |
||||
} |
||||
|
||||
out := data.NewFrame("", fields...) |
||||
out.Meta = &data.FrameMeta{Type: numericFullLongType} |
||||
return data.Frames{out}, nil |
||||
} |
||||
|
||||
func convertTimeSeriesMultiToFullLong(frames data.Frames) (data.Frames, error) { |
||||
type row struct { |
||||
t time.Time |
||||
value *float64 |
||||
metric string |
||||
display *string |
||||
labels data.Labels |
||||
} |
||||
|
||||
var rows []row |
||||
labelKeysSet := map[string]struct{}{} |
||||
hasDisplayCol := false |
||||
|
||||
for _, frame := range frames { |
||||
var timeField *data.Field |
||||
for _, f := range frame.Fields { |
||||
if f.Type() == data.FieldTypeTime { |
||||
timeField = f |
||||
break |
||||
} |
||||
} |
||||
if timeField == nil { |
||||
return nil, fmt.Errorf("missing time field") |
||||
} |
||||
for _, f := range frame.Fields { |
||||
if !f.Type().Numeric() { |
||||
continue |
||||
} |
||||
var display *string |
||||
if f.Config != nil && f.Config.DisplayNameFromDS != "" { |
||||
s := f.Config.DisplayNameFromDS |
||||
display = &s |
||||
hasDisplayCol = true |
||||
} |
||||
for i := 0; i < f.Len(); i++ { |
||||
t := timeField.At(i).(time.Time) |
||||
v, err := f.FloatAt(i) |
||||
if err != nil { |
||||
continue |
||||
} |
||||
val := v |
||||
rows = append(rows, row{ |
||||
t: t, |
||||
value: &val, |
||||
metric: f.Name, |
||||
display: display, |
||||
labels: f.Labels, |
||||
}) |
||||
for k := range f.Labels { |
||||
labelKeysSet[k] = struct{}{} |
||||
} |
||||
} |
||||
} |
||||
} |
||||
|
||||
labelKeys := make([]string, 0, len(labelKeysSet)) |
||||
for k := range labelKeysSet { |
||||
labelKeys = append(labelKeys, k) |
||||
} |
||||
sort.Strings(labelKeys) |
||||
sort.SliceStable(rows, func(i, j int) bool { |
||||
if rows[i].t.Equal(rows[j].t) { |
||||
return rows[i].metric < rows[j].metric |
||||
} |
||||
return rows[i].t.Before(rows[j].t) |
||||
}) |
||||
|
||||
times := make([]time.Time, len(rows)) |
||||
values := make([]*float64, len(rows)) |
||||
metrics := make([]string, len(rows)) |
||||
var displays []*string |
||||
if hasDisplayCol { |
||||
displays = make([]*string, len(rows)) |
||||
} |
||||
labels := make(map[string][]*string) |
||||
for _, k := range labelKeys { |
||||
labels[k] = make([]*string, len(rows)) |
||||
} |
||||
|
||||
for i, r := range rows { |
||||
times[i] = r.t |
||||
values[i] = r.value |
||||
metrics[i] = r.metric |
||||
if hasDisplayCol { |
||||
displays[i] = r.display |
||||
} |
||||
for _, k := range labelKeys { |
||||
if v, ok := r.labels[k]; ok { |
||||
labels[k][i] = &v |
||||
} |
||||
} |
||||
} |
||||
|
||||
fields := []*data.Field{ |
||||
data.NewField("time", nil, times), |
||||
data.NewField(SQLValueFieldName, nil, values), |
||||
data.NewField(SQLMetricFieldName, nil, metrics), |
||||
} |
||||
if hasDisplayCol { |
||||
fields = append(fields, data.NewField(SQLDisplayFieldName, nil, displays)) |
||||
} |
||||
for _, k := range labelKeys { |
||||
fields = append(fields, data.NewField(k, nil, labels[k])) |
||||
} |
||||
|
||||
out := data.NewFrame("", fields...) |
||||
out.Meta = &data.FrameMeta{Type: timeseriesFullLongType} |
||||
return data.Frames{out}, nil |
||||
} |
||||
|
||||
func convertTimeSeriesWideToFullLong(frames data.Frames) (data.Frames, error) { |
||||
if len(frames) != 1 { |
||||
return nil, fmt.Errorf("expected exactly one frame for wide format, but got %d", len(frames)) |
||||
} |
||||
frame := frames[0] |
||||
|
||||
var timeField *data.Field |
||||
for _, f := range frame.Fields { |
||||
if f.Type() == data.FieldTypeTime { |
||||
timeField = f |
||||
break |
||||
} |
||||
} |
||||
if timeField == nil { |
||||
return nil, fmt.Errorf("time field not found in TimeSeriesWide frame") |
||||
} |
||||
|
||||
type row struct { |
||||
t time.Time |
||||
value *float64 |
||||
metric string |
||||
display *string |
||||
labels data.Labels |
||||
} |
||||
|
||||
var ( |
||||
rows []row |
||||
labelKeysSet = map[string]struct{}{} |
||||
hasDisplayCol bool |
||||
) |
||||
|
||||
// Collect all label keys
|
||||
for _, f := range frame.Fields { |
||||
if !f.Type().Numeric() { |
||||
continue |
||||
} |
||||
for k := range f.Labels { |
||||
labelKeysSet[k] = struct{}{} |
||||
} |
||||
} |
||||
|
||||
labelKeys := make([]string, 0, len(labelKeysSet)) |
||||
for k := range labelKeysSet { |
||||
labelKeys = append(labelKeys, k) |
||||
} |
||||
sort.Strings(labelKeys) |
||||
|
||||
timeLen := timeField.Len() |
||||
for _, f := range frame.Fields { |
||||
if !f.Type().Numeric() { |
||||
continue |
||||
} |
||||
var display *string |
||||
if f.Config != nil && f.Config.DisplayNameFromDS != "" { |
||||
s := f.Config.DisplayNameFromDS |
||||
display = &s |
||||
hasDisplayCol = true |
||||
} |
||||
for i := 0; i < timeLen; i++ { |
||||
t := timeField.At(i).(time.Time) |
||||
v, err := f.FloatAt(i) |
||||
if err != nil { |
||||
continue |
||||
} |
||||
val := v |
||||
rows = append(rows, row{ |
||||
t: t, |
||||
value: &val, |
||||
metric: f.Name, |
||||
display: display, |
||||
labels: f.Labels, |
||||
}) |
||||
} |
||||
} |
||||
|
||||
sort.SliceStable(rows, func(i, j int) bool { |
||||
if rows[i].t.Equal(rows[j].t) { |
||||
return rows[i].metric < rows[j].metric |
||||
} |
||||
return rows[i].t.Before(rows[j].t) |
||||
}) |
||||
|
||||
times := make([]time.Time, len(rows)) |
||||
values := make([]*float64, len(rows)) |
||||
metrics := make([]string, len(rows)) |
||||
var displays []*string |
||||
if hasDisplayCol { |
||||
displays = make([]*string, len(rows)) |
||||
} |
||||
labels := make(map[string][]*string) |
||||
for _, k := range labelKeys { |
||||
labels[k] = make([]*string, len(rows)) |
||||
} |
||||
|
||||
for i, r := range rows { |
||||
times[i] = r.t |
||||
values[i] = r.value |
||||
metrics[i] = r.metric |
||||
if hasDisplayCol { |
||||
displays[i] = r.display |
||||
} |
||||
for _, k := range labelKeys { |
||||
if v, ok := r.labels[k]; ok { |
||||
labels[k][i] = &v |
||||
} |
||||
} |
||||
} |
||||
|
||||
fields := []*data.Field{ |
||||
data.NewField("time", nil, times), |
||||
data.NewField(SQLValueFieldName, nil, values), |
||||
data.NewField(SQLMetricFieldName, nil, metrics), |
||||
} |
||||
if hasDisplayCol { |
||||
fields = append(fields, data.NewField(SQLDisplayFieldName, nil, displays)) |
||||
} |
||||
for _, k := range labelKeys { |
||||
fields = append(fields, data.NewField(k, nil, labels[k])) |
||||
} |
||||
|
||||
out := data.NewFrame("", fields...) |
||||
out.Meta = &data.FrameMeta{Type: timeseriesFullLongType} |
||||
return data.Frames{out}, nil |
||||
} |
||||
|
||||
func supportedToLongConversion(inputType data.FrameType) bool { |
||||
switch inputType { |
||||
case data.FrameTypeNumericMulti, data.FrameTypeNumericWide: |
||||
return true |
||||
case data.FrameTypeTimeSeriesMulti, data.FrameTypeTimeSeriesWide: |
||||
return true |
||||
default: |
||||
return false |
||||
} |
||||
} |
||||
|
||||
func convertNumericMultiToNumericWide(frames data.Frames) data.Frames { |
||||
if len(frames) == 0 { |
||||
return nil |
||||
} |
||||
|
||||
out := data.NewFrame("") |
||||
for _, frame := range frames { |
||||
for _, field := range frame.Fields { |
||||
if field.Type().Numeric() { |
||||
out.Fields = append(out.Fields, field) |
||||
} |
||||
} |
||||
} |
||||
out.Meta = &data.FrameMeta{Type: data.FrameTypeNumericWide} |
||||
return data.Frames{out} |
||||
} |
||||
@ -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 |
||||
} |
||||
@ -0,0 +1,373 @@ |
||||
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 TestConvertTimeSeriesMultiToFullLong(t *testing.T) { |
||||
t.Run("SingleSeriesNoLabels", func(t *testing.T) { |
||||
times := []time.Time{ |
||||
time.Unix(0, 0), |
||||
time.Unix(10, 0), |
||||
time.Unix(20, 0), |
||||
} |
||||
values := []float64{1.0, 2.0, 3.0} |
||||
|
||||
input := data.Frames{ |
||||
data.NewFrame("cpu", |
||||
data.NewField("time", nil, times), |
||||
data.NewField("cpu", nil, values), |
||||
), |
||||
} |
||||
input[0].Meta = &data.FrameMeta{Type: data.FrameTypeTimeSeriesMulti} |
||||
|
||||
expected := data.NewFrame("", |
||||
data.NewField("time", nil, times), |
||||
data.NewField(SQLValueFieldName, nil, []*float64{fp(1.0), fp(2.0), fp(3.0)}), |
||||
data.NewField(SQLMetricFieldName, nil, []string{"cpu", "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, "Result mismatch (-want +got):%s", diff) |
||||
} |
||||
}) |
||||
|
||||
t.Run("TwoSeriesOneLabel", func(t *testing.T) { |
||||
times := []time.Time{ |
||||
time.Unix(0, 0), |
||||
time.Unix(10, 0), |
||||
} |
||||
|
||||
input := data.Frames{ |
||||
data.NewFrame("cpu", |
||||
data.NewField("time", nil, times), |
||||
data.NewField("cpu", data.Labels{"host": "a"}, []float64{1.0, 2.0}), |
||||
), |
||||
data.NewFrame("cpu", |
||||
data.NewField("time", nil, times), |
||||
data.NewField("cpu", data.Labels{"host": "b"}, []float64{3.0, 4.0}), |
||||
), |
||||
} |
||||
input[0].Meta = &data.FrameMeta{Type: data.FrameTypeTimeSeriesMulti} |
||||
input[1].Meta = &data.FrameMeta{Type: data.FrameTypeTimeSeriesMulti} |
||||
|
||||
expected := data.NewFrame("", |
||||
data.NewField("time", nil, []time.Time{ |
||||
time.Unix(0, 0), time.Unix(0, 0), time.Unix(10, 0), time.Unix(10, 0), |
||||
}), |
||||
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, "Result mismatch (-want +got):%s", diff) |
||||
} |
||||
}) |
||||
|
||||
t.Run("TwoMetricsWithSharedLabels", func(t *testing.T) { |
||||
times := []time.Time{ |
||||
time.Unix(0, 0), |
||||
time.Unix(10, 0), |
||||
} |
||||
|
||||
input := data.Frames{ |
||||
data.NewFrame("cpu", |
||||
data.NewField("time", nil, times), |
||||
data.NewField("cpu", data.Labels{"host": "a"}, []float64{1.0, 2.0}), |
||||
), |
||||
data.NewFrame("mem", |
||||
data.NewField("time", nil, times), |
||||
data.NewField("mem", data.Labels{"host": "a"}, []float64{3.0, 4.0}), |
||||
), |
||||
} |
||||
input[0].Meta = &data.FrameMeta{Type: data.FrameTypeTimeSeriesMulti} |
||||
input[1].Meta = &data.FrameMeta{Type: data.FrameTypeTimeSeriesMulti} |
||||
|
||||
expected := data.NewFrame("", |
||||
data.NewField("time", nil, []time.Time{ |
||||
time.Unix(0, 0), time.Unix(0, 0), time.Unix(10, 0), time.Unix(10, 0), |
||||
}), |
||||
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, "Result mismatch (-want +got):%s", diff) |
||||
} |
||||
}) |
||||
|
||||
t.Run("TwoSeriesSparseLabels", func(t *testing.T) { |
||||
times := []time.Time{ |
||||
time.Unix(0, 0), |
||||
time.Unix(10, 0), |
||||
} |
||||
|
||||
input := data.Frames{ |
||||
data.NewFrame("cpu", |
||||
data.NewField("time", nil, times), |
||||
data.NewField("cpu", data.Labels{"host": "a"}, []float64{1.0, 2.0}), |
||||
), |
||||
data.NewFrame("cpu", |
||||
data.NewField("time", nil, times), |
||||
data.NewField("cpu", data.Labels{"host": "b", "env": "prod"}, []float64{3.0, 4.0}), |
||||
), |
||||
} |
||||
input[0].Meta = &data.FrameMeta{Type: data.FrameTypeTimeSeriesMulti} |
||||
input[1].Meta = &data.FrameMeta{Type: data.FrameTypeTimeSeriesMulti} |
||||
|
||||
expected := data.NewFrame("", |
||||
data.NewField("time", nil, []time.Time{ |
||||
time.Unix(0, 0), time.Unix(0, 0), time.Unix(10, 0), time.Unix(10, 0), |
||||
}), |
||||
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, "Result mismatch (-want +got):%s", diff) |
||||
} |
||||
}) |
||||
|
||||
t.Run("TwoSeriesSparseMetrics", func(t *testing.T) { |
||||
times := []time.Time{ |
||||
time.Unix(0, 0), |
||||
time.Unix(10, 0), |
||||
} |
||||
|
||||
input := data.Frames{ |
||||
data.NewFrame("cpu", |
||||
data.NewField("time", nil, times), |
||||
data.NewField("cpu", data.Labels{"host": "a"}, []float64{1.0, 2.0}), |
||||
), |
||||
data.NewFrame("mem", |
||||
data.NewField("time", nil, times), |
||||
data.NewField("mem", data.Labels{"host": "b"}, []float64{3.0, 4.0}), |
||||
), |
||||
} |
||||
input[0].Meta = &data.FrameMeta{Type: data.FrameTypeTimeSeriesMulti} |
||||
input[1].Meta = &data.FrameMeta{Type: data.FrameTypeTimeSeriesMulti} |
||||
|
||||
expected := data.NewFrame("", |
||||
data.NewField("time", nil, []time.Time{ |
||||
time.Unix(0, 0), time.Unix(0, 0), time.Unix(10, 0), time.Unix(10, 0), |
||||
}), |
||||
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("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, "Result mismatch (-want +got):%s", diff) |
||||
} |
||||
}) |
||||
|
||||
t.Run("TwoSeriesSparseMetricsAndLabels", func(t *testing.T) { |
||||
times := []time.Time{ |
||||
time.Unix(0, 0), |
||||
time.Unix(10, 0), |
||||
} |
||||
|
||||
input := data.Frames{ |
||||
data.NewFrame("cpu", |
||||
data.NewField("time", nil, times), |
||||
data.NewField("cpu", data.Labels{"host": "a"}, []float64{1.0, 2.0}), |
||||
), |
||||
data.NewFrame("mem", |
||||
data.NewField("time", nil, times), |
||||
data.NewField("mem", data.Labels{"host": "b", "env": "prod"}, []float64{3.0, 4.0}), |
||||
), |
||||
} |
||||
input[0].Meta = &data.FrameMeta{Type: data.FrameTypeTimeSeriesMulti} |
||||
input[1].Meta = &data.FrameMeta{Type: data.FrameTypeTimeSeriesMulti} |
||||
|
||||
expected := data.NewFrame("", |
||||
data.NewField("time", nil, []time.Time{ |
||||
time.Unix(0, 0), time.Unix(0, 0), time.Unix(10, 0), time.Unix(10, 0), |
||||
}), |
||||
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, "Result mismatch (-want +got):%s", diff) |
||||
} |
||||
}) |
||||
|
||||
t.Run("ThreeSeriesSparseTimeLabelsMetrics", func(t *testing.T) { |
||||
timesA := []time.Time{ |
||||
time.Unix(0, 0), |
||||
time.Unix(10, 0), |
||||
} |
||||
timesB := []time.Time{ |
||||
time.Unix(5, 0), |
||||
time.Unix(15, 0), |
||||
} |
||||
timesMem := []time.Time{ |
||||
time.Unix(10, 0), |
||||
time.Unix(30, 0), |
||||
} |
||||
|
||||
input := data.Frames{ |
||||
data.NewFrame("cpu", |
||||
data.NewField("time", nil, timesA), |
||||
data.NewField("cpu", data.Labels{"host": "a"}, []float64{1.0, 2.0}), |
||||
), |
||||
data.NewFrame("cpu", |
||||
data.NewField("time", nil, timesB), |
||||
data.NewField("cpu", nil, []float64{9.0, 10.0}), // no labels
|
||||
), |
||||
data.NewFrame("mem", |
||||
data.NewField("time", nil, timesMem), |
||||
data.NewField("mem", data.Labels{"host": "b", "env": "prod"}, []float64{3.0, 4.0}), |
||||
), |
||||
} |
||||
for _, f := range input { |
||||
f.Meta = &data.FrameMeta{Type: data.FrameTypeTimeSeriesMulti} |
||||
} |
||||
|
||||
expected := data.NewFrame("", |
||||
data.NewField("time", nil, []time.Time{ |
||||
time.Unix(0, 0), // cpu a
|
||||
time.Unix(5, 0), // cpu no label
|
||||
time.Unix(10, 0), // cpu a
|
||||
time.Unix(10, 0), // mem
|
||||
time.Unix(15, 0), // cpu no label
|
||||
time.Unix(30, 0), // mem
|
||||
}), |
||||
data.NewField(SQLValueFieldName, nil, []*float64{ |
||||
fp(1.0), fp(9.0), fp(2.0), fp(3.0), fp(10.0), fp(4.0), |
||||
}), |
||||
data.NewField(SQLMetricFieldName, nil, []string{ |
||||
"cpu", "cpu", "cpu", "mem", "cpu", "mem", |
||||
}), |
||||
data.NewField("env", nil, []*string{ |
||||
nil, nil, nil, sp("prod"), nil, sp("prod"), |
||||
}), |
||||
data.NewField("host", nil, []*string{ |
||||
sp("a"), nil, sp("a"), sp("b"), nil, 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, "Result mismatch (-want +got):%s", diff) |
||||
} |
||||
}) |
||||
} |
||||
|
||||
func TestConvertTimeSeriesMultiToFullLongWithDisplayName(t *testing.T) { |
||||
t.Run("SingleSeriesWithDisplayName", func(t *testing.T) { |
||||
times := []time.Time{time.Unix(0, 0), time.Unix(10, 0)} |
||||
|
||||
input := data.Frames{ |
||||
data.NewFrame("cpu", |
||||
data.NewField("time", nil, times), |
||||
func() *data.Field { |
||||
f := data.NewField("cpu", nil, []float64{1.0, 2.0}) |
||||
f.Config = &data.FieldConfig{DisplayNameFromDS: "CPU Display"} |
||||
return f |
||||
}(), |
||||
), |
||||
} |
||||
input[0].Meta = &data.FrameMeta{Type: data.FrameTypeTimeSeriesMulti} |
||||
|
||||
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"}), |
||||
data.NewField(SQLDisplayFieldName, nil, []*string{sp("CPU Display"), sp("CPU Display")}), |
||||
) |
||||
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("TwoSeriesMixedDisplayNames", func(t *testing.T) { |
||||
times := []time.Time{time.Unix(0, 0), time.Unix(10, 0)} |
||||
|
||||
input := data.Frames{ |
||||
data.NewFrame("cpu", |
||||
data.NewField("time", nil, times), |
||||
func() *data.Field { |
||||
f := data.NewField("cpu", data.Labels{"host": "a"}, []float64{1.0, 2.0}) |
||||
f.Config = &data.FieldConfig{DisplayNameFromDS: "CPU A"} |
||||
return f |
||||
}(), |
||||
), |
||||
data.NewFrame("cpu", |
||||
data.NewField("time", nil, times), |
||||
data.NewField("cpu", data.Labels{"host": "b"}, []float64{3.0, 4.0}), |
||||
), |
||||
} |
||||
for _, f := range input { |
||||
f.Meta = &data.FrameMeta{Type: data.FrameTypeTimeSeriesMulti} |
||||
} |
||||
|
||||
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(SQLDisplayFieldName, nil, []*string{ |
||||
sp("CPU A"), nil, sp("CPU A"), nil, |
||||
}), |
||||
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) |
||||
} |
||||
}) |
||||
} |
||||
@ -1,311 +0,0 @@ |
||||
package expr |
||||
|
||||
import ( |
||||
"fmt" |
||||
"sort" |
||||
"time" |
||||
|
||||
"github.com/grafana/grafana-plugin-sdk-go/data" |
||||
) |
||||
|
||||
func ConvertToLong(frames data.Frames) (data.Frames, error) { |
||||
if len(frames) == 0 { |
||||
// general empty case for now
|
||||
return frames, nil |
||||
} |
||||
// Four Conversion Possible Cases
|
||||
// 1. NumericMulti -> NumericLong
|
||||
// 2. NumericWide -> NumericLong
|
||||
// 3. TimeSeriesMulti -> TimeSeriesLong
|
||||
// 4. TimeSeriesWide -> TimeSeriesLong
|
||||
|
||||
// Detect if input type is declared
|
||||
// First Check Frame Meta Type
|
||||
|
||||
var inputType data.FrameType |
||||
if frames[0].Meta != nil && frames[0].Meta.Type != "" { |
||||
inputType = frames[0].Meta.Type |
||||
} |
||||
|
||||
// TODO: Add some guessing of Type if not declared
|
||||
if inputType == "" { |
||||
return frames, fmt.Errorf("no input dataframe type set") |
||||
} |
||||
|
||||
if !supportedToLongConversion(inputType) { |
||||
return frames, fmt.Errorf("unsupported input dataframe type %s for SQL expression", inputType) |
||||
} |
||||
|
||||
toLong := getToLongConversionFunc(inputType) |
||||
if toLong == nil { |
||||
return frames, fmt.Errorf("could not get conversion function for input type %s", inputType) |
||||
} |
||||
|
||||
return toLong(frames) |
||||
} |
||||
|
||||
func convertNumericMultiToNumericLong(frames data.Frames) (data.Frames, error) { |
||||
// Apart from metadata, NumericMulti is basically NumericWide, except one frame per thing
|
||||
// so we collapse into wide and call the wide conversion
|
||||
wide := convertNumericMultiToNumericWide(frames) |
||||
return convertNumericWideToNumericLong(wide) |
||||
} |
||||
|
||||
func convertNumericMultiToNumericWide(frames data.Frames) data.Frames { |
||||
newFrame := data.NewFrame("") |
||||
for _, frame := range frames { |
||||
for _, field := range frame.Fields { |
||||
if !field.Type().Numeric() { |
||||
continue |
||||
} |
||||
newField := data.NewFieldFromFieldType(field.Type(), field.Len()) |
||||
newField.Name = field.Name |
||||
newField.Labels = field.Labels.Copy() |
||||
if field.Len() == 1 { |
||||
newField.Set(0, field.CopyAt(0)) |
||||
} |
||||
newFrame.Fields = append(newFrame.Fields, newField) |
||||
} |
||||
} |
||||
return data.Frames{newFrame} |
||||
} |
||||
|
||||
func convertNumericWideToNumericLong(frames data.Frames) (data.Frames, error) { |
||||
// Wide should only be one frame
|
||||
if len(frames) != 1 { |
||||
return nil, fmt.Errorf("expected exactly one frame for wide format, but got %d", len(frames)) |
||||
} |
||||
inputFrame := frames[0] |
||||
|
||||
// The Frame should have no more than one row
|
||||
if inputFrame.Rows() > 1 { |
||||
return nil, fmt.Errorf("expected no more than one row in the frame, but got %d", inputFrame.Rows()) |
||||
} |
||||
|
||||
// Gather:
|
||||
// - unique numeric Field Names, and
|
||||
// - unique Label Keys (from Numeric Fields only)
|
||||
// each one maps to a field in the output long Frame.
|
||||
uniqueNames := make([]string, 0) |
||||
uniqueKeys := make([]string, 0) |
||||
|
||||
uniqueNamesMap := make(map[string]data.FieldType) |
||||
uniqueKeysMap := make(map[string]struct{}) |
||||
|
||||
prints := make(map[string]int) |
||||
|
||||
registerPrint := func(labels data.Labels) { |
||||
fp := labels.Fingerprint().String() |
||||
if _, ok := prints[fp]; !ok { |
||||
prints[fp] = len(prints) |
||||
} |
||||
} |
||||
|
||||
for _, field := range inputFrame.Fields { |
||||
if field.Type().Numeric() { |
||||
if _, ok := uniqueNamesMap[field.Name]; !ok { |
||||
uniqueNames = append(uniqueNames, field.Name) |
||||
uniqueNamesMap[field.Name] = field.Type() |
||||
} |
||||
|
||||
if field.Labels != nil { |
||||
registerPrint(field.Labels) |
||||
for key := range field.Labels { |
||||
if _, ok := uniqueKeysMap[key]; !ok { |
||||
uniqueKeys = append(uniqueKeys, key) |
||||
} |
||||
uniqueKeysMap[key] = struct{}{} |
||||
} |
||||
} |
||||
} |
||||
} |
||||
|
||||
// Create new fields for output Long frame
|
||||
fields := make([]*data.Field, 0, len(uniqueNames)+len(uniqueKeys)) |
||||
|
||||
// Create the Numeric Fields, tracking the index of each field by name
|
||||
// Note: May want to use FloatAt and and prepopulate with NaN so missing
|
||||
// combinations of value can be NA instead of the zero value of 0.
|
||||
var nameIndexMap = make(map[string]int, len(uniqueNames)) |
||||
for i, name := range uniqueNames { |
||||
field := data.NewFieldFromFieldType(uniqueNamesMap[name], len(prints)) |
||||
field.Name = name |
||||
fields = append(fields, field) |
||||
nameIndexMap[name] = i |
||||
} |
||||
|
||||
// Create the String fields, tracking the index of each field by key
|
||||
var keyIndexMap = make(map[string]int, len(uniqueKeys)) |
||||
for i, k := range uniqueKeys { |
||||
fields = append(fields, data.NewField(k, nil, make([]string, len(prints)))) |
||||
keyIndexMap[k] = len(nameIndexMap) + i |
||||
} |
||||
|
||||
longFrame := data.NewFrame("", fields...) |
||||
|
||||
if inputFrame.Rows() == 0 { |
||||
return data.Frames{longFrame}, nil |
||||
} |
||||
|
||||
// Add Rows to the fields
|
||||
for _, field := range inputFrame.Fields { |
||||
if !field.Type().Numeric() { |
||||
continue |
||||
} |
||||
fieldIdx := prints[field.Labels.Fingerprint().String()] |
||||
longFrame.Fields[nameIndexMap[field.Name]].Set(fieldIdx, field.CopyAt(0)) |
||||
for key, value := range field.Labels { |
||||
longFrame.Fields[keyIndexMap[key]].Set(fieldIdx, value) |
||||
} |
||||
} |
||||
|
||||
return data.Frames{longFrame}, nil |
||||
} |
||||
|
||||
func convertTimeSeriesMultiToTimeSeriesLong(frames data.Frames) (data.Frames, error) { |
||||
// Collect all time values and ensure no duplicates
|
||||
timeSet := make(map[time.Time]struct{}) |
||||
labelKeys := make(map[string]struct{}) // Collect all unique label keys
|
||||
numericFields := make(map[string]struct{}) // Collect unique numeric field names
|
||||
|
||||
for _, frame := range frames { |
||||
for _, field := range frame.Fields { |
||||
if field.Type() == data.FieldTypeTime { |
||||
for i := 0; i < field.Len(); i++ { |
||||
t := field.At(i).(time.Time) |
||||
timeSet[t] = struct{}{} |
||||
} |
||||
} else if field.Type().Numeric() { |
||||
numericFields[field.Name] = struct{}{} |
||||
if field.Labels != nil { |
||||
for key := range field.Labels { |
||||
labelKeys[key] = struct{}{} |
||||
} |
||||
} |
||||
} |
||||
} |
||||
} |
||||
|
||||
// Create a sorted slice of unique time values
|
||||
times := make([]time.Time, 0, len(timeSet)) |
||||
for t := range timeSet { |
||||
times = append(times, t) |
||||
} |
||||
sort.Slice(times, func(i, j int) bool { return times[i].Before(times[j]) }) |
||||
|
||||
// Create output fields: Time, one numeric field per unique numeric name, and label fields
|
||||
timeField := data.NewField("Time", nil, times) |
||||
outputNumericFields := make(map[string]*data.Field) |
||||
for name := range numericFields { |
||||
outputNumericFields[name] = data.NewField(name, nil, make([]float64, len(times))) |
||||
} |
||||
outputLabelFields := make(map[string]*data.Field) |
||||
for key := range labelKeys { |
||||
outputLabelFields[key] = data.NewField(key, nil, make([]string, len(times))) |
||||
} |
||||
|
||||
// Map time to index for quick lookup
|
||||
timeIndexMap := make(map[time.Time]int, len(times)) |
||||
for i, t := range times { |
||||
timeIndexMap[t] = i |
||||
} |
||||
|
||||
// Populate output fields
|
||||
for _, frame := range frames { |
||||
var timeField *data.Field |
||||
for _, field := range frame.Fields { |
||||
if field.Type() == data.FieldTypeTime { |
||||
timeField = field |
||||
break |
||||
} |
||||
} |
||||
|
||||
if timeField == nil { |
||||
return nil, fmt.Errorf("no time field found in frame") |
||||
} |
||||
|
||||
for _, field := range frame.Fields { |
||||
if field.Type().Numeric() { |
||||
for i := 0; i < field.Len(); i++ { |
||||
t := timeField.At(i).(time.Time) |
||||
val, err := field.FloatAt(i) |
||||
if err != nil { |
||||
val = 0 // Default value for missing data
|
||||
} |
||||
idx := timeIndexMap[t] |
||||
if outputField, exists := outputNumericFields[field.Name]; exists { |
||||
outputField.Set(idx, val) |
||||
} |
||||
|
||||
// Add labels for the numeric field
|
||||
for key, value := range field.Labels { |
||||
if outputField, exists := outputLabelFields[key]; exists { |
||||
outputField.Set(idx, value) |
||||
} |
||||
} |
||||
} |
||||
} |
||||
} |
||||
} |
||||
|
||||
// Build the output frame
|
||||
outputFields := []*data.Field{timeField} |
||||
for _, field := range outputNumericFields { |
||||
outputFields = append(outputFields, field) |
||||
} |
||||
for _, field := range outputLabelFields { |
||||
outputFields = append(outputFields, field) |
||||
} |
||||
outputFrame := data.NewFrame("time_series_long", outputFields...) |
||||
|
||||
// Set metadata
|
||||
if outputFrame.Meta == nil { |
||||
outputFrame.Meta = &data.FrameMeta{} |
||||
} |
||||
outputFrame.Meta.Type = data.FrameTypeTimeSeriesLong |
||||
|
||||
return data.Frames{outputFrame}, nil |
||||
} |
||||
|
||||
func convertTimeSeriesWideToTimeSeriesLong(frames data.Frames) (data.Frames, error) { |
||||
// Wide should only be one frame
|
||||
if len(frames) != 1 { |
||||
return nil, fmt.Errorf("expected exactly one frame for wide format, but got %d", len(frames)) |
||||
} |
||||
inputFrame := frames[0] |
||||
longFrame, err := data.WideToLong(inputFrame) |
||||
if err != nil { |
||||
return nil, fmt.Errorf("failed to convert wide time series to long timeseries for sql expression: %w", err) |
||||
} |
||||
return data.Frames{longFrame}, nil |
||||
} |
||||
|
||||
func getToLongConversionFunc(inputType data.FrameType) func(data.Frames) (data.Frames, error) { |
||||
switch inputType { |
||||
case data.FrameTypeNumericMulti: |
||||
return convertNumericMultiToNumericLong |
||||
case data.FrameTypeNumericWide: |
||||
return convertNumericWideToNumericLong |
||||
case data.FrameTypeTimeSeriesMulti: |
||||
return convertTimeSeriesMultiToTimeSeriesLong |
||||
case data.FrameTypeTimeSeriesWide: |
||||
return convertTimeSeriesWideToTimeSeriesLong |
||||
default: |
||||
return convertErr |
||||
} |
||||
} |
||||
|
||||
func convertErr(_ data.Frames) (data.Frames, error) { |
||||
return nil, fmt.Errorf("unsupported input type for SQL expression") |
||||
} |
||||
|
||||
func supportedToLongConversion(inputType data.FrameType) bool { |
||||
switch inputType { |
||||
case data.FrameTypeNumericMulti, data.FrameTypeNumericWide: |
||||
return true |
||||
case data.FrameTypeTimeSeriesMulti, data.FrameTypeTimeSeriesWide: |
||||
return true |
||||
default: |
||||
return false |
||||
} |
||||
} |
||||
@ -1,48 +0,0 @@ |
||||
package expr |
||||
|
||||
import ( |
||||
"testing" |
||||
|
||||
"github.com/google/go-cmp/cmp" |
||||
"github.com/grafana/grafana-plugin-sdk-go/data" |
||||
"github.com/stretchr/testify/require" |
||||
) |
||||
|
||||
func TestConvertNumericMultiToLong(t *testing.T) { |
||||
input := data.Frames{ |
||||
data.NewFrame("test", |
||||
data.NewField("Value", data.Labels{"city": "MIA"}, []int64{5})), |
||||
data.NewFrame("test", |
||||
data.NewField("Value", data.Labels{"city": "LGA"}, []int64{7}), |
||||
), |
||||
} |
||||
expectedFrame := data.NewFrame("", |
||||
data.NewField("Value", nil, []int64{5, 7}), |
||||
data.NewField("city", nil, []string{"MIA", "LGA"}), |
||||
) |
||||
output, err := convertNumericMultiToNumericLong(input) |
||||
require.NoError(t, err) |
||||
|
||||
if diff := cmp.Diff(expectedFrame, output[0], data.FrameTestCompareOptions()...); diff != "" { |
||||
require.FailNowf(t, "Result mismatch (-want +got):%s\n", diff) |
||||
} |
||||
} |
||||
|
||||
func TestConvertNumericWideToLong(t *testing.T) { |
||||
input := data.Frames{ |
||||
data.NewFrame("test", |
||||
data.NewField("Value", data.Labels{"city": "MIA"}, []int64{5}), |
||||
data.NewField("Value", data.Labels{"city": "LGA"}, []int64{7}), |
||||
), |
||||
} |
||||
expectedFrame := data.NewFrame("", |
||||
data.NewField("Value", nil, []int64{5, 7}), |
||||
data.NewField("city", nil, []string{"MIA", "LGA"}), |
||||
) |
||||
output, err := convertNumericWideToNumericLong(input) |
||||
require.NoError(t, err) |
||||
|
||||
if diff := cmp.Diff(expectedFrame, output[0], data.FrameTestCompareOptions()...); diff != "" { |
||||
require.FailNowf(t, "Result mismatch (-want +got):%s\n", diff) |
||||
} |
||||
} |
||||
Loading…
Reference in new issue