IntenseWebs/grafana
3
0
Fork 0
mirror of https://github.com/grafana/grafana.git synced 2025-02-11 16:15:42 -06:00
Code Issues Packages Projects Releases Wiki Activity

Pyroscope: Fix panic on query when symbol names are repeating (#72188)

Browse Source 
Use EnumItemIndex type in values map
This commit is contained in:
Andrej Ocenas 2023-07-24 13:06:38 +02:00 committed by GitHub
parent a2442ef620
commit 689d9ed430
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 9 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
pkg/tsdb/grafana-pyroscope-datasource/query.go
View File

@ -312,14 +312,14 @@ func treeToNestedSetDataFrame(tree *ProfileTree, unit string) *data.Frame {
type EnumField struct {
field *data.Field
valuesMap map[string]int64
counter int64
valuesMap map[string]data.EnumItemIndex
counter data.EnumItemIndex
}
func NewEnumField(name string, labels data.Labels) *EnumField {
return &EnumField{
field: data.NewField(name, labels, []data.EnumItemIndex{}),
valuesMap: make(map[string]int64),
valuesMap: make(map[string]data.EnumItemIndex),
}
}
@ -328,7 +328,7 @@ func (e *EnumField) Append(value string) {
e.field.Append(valueIndex)
} else {
e.valuesMap[value] = e.counter
e.field.Append(data.EnumItemIndex(e.counter))
e.field.Append(e.counter)
e.counter++
}
}

9
pkg/tsdb/grafana-pyroscope-datasource/query_test.go
View File

@ -196,6 +196,7 @@ func Test_treeToNestedDataFrame(t *testing.T) {
},
{Value: 30, Level: 1, Self: 3, Name: "func2", Nodes: []*ProfileTree{
{Value: 15, Level: 2, Self: 4, Name: "func1:func3"},
{Value: 10, Level: 2, Self: 4, Name: "func1"},
}},
},
}
@ -211,10 +212,10 @@ func Test_treeToNestedDataFrame(t *testing.T) {
}
require.Equal(t,
[]*data.Field{
data.NewField("level", nil, []int64{0, 1, 1, 2}),
data.NewField("value", nil, []int64{100, 40, 30, 15}).SetConfig(&data.FieldConfig{Unit: "short"}),
data.NewField("self", nil, []int64{1, 2, 3, 4}).SetConfig(&data.FieldConfig{Unit: "short"}),
data.NewField("label", nil, []data.EnumItemIndex{0, 1, 2, 3}).SetConfig(labelConfig),
data.NewField("level", nil, []int64{0, 1, 1, 2, 2}),
data.NewField("value", nil, []int64{100, 40, 30, 15, 10}).SetConfig(&data.FieldConfig{Unit: "short"}),
data.NewField("self", nil, []int64{1, 2, 3, 4, 4}).SetConfig(&data.FieldConfig{Unit: "short"}),
data.NewField("label", nil, []data.EnumItemIndex{0, 1, 2, 3, 1}).SetConfig(labelConfig),
}, frame.Fields)
})