fix: Improves unit detection

Detects units from the first row as well as unit col meta

Author: NeedleInAJayStack

pkg/plugin/datasource.go

@@ -421,14 +421,8 @@ func dataFrameFromGrid(grid haystack.Grid) (*data.Frame, error) {
 		// Set Grafana field info from Haystack grid info
 		config := &data.FieldConfig{}
 		config.DisplayName = col.Name()
-		switch unit := col.Meta().Get("unit").(type) {
-		case haystack.Str:
-			config.Unit = unit.String()
-		default:
-			config.Unit = ""
-		}
+		config.Unit = unitFromGrid(grid, col)
 		field.Config = config
-
 		fields = append(fields, field)
 	}
 
@@ -443,6 +437,29 @@ func dataFrameFromGrid(grid haystack.Grid) (*data.Frame, error) {
 	return frame, nil
 }
 
+// unitFromGrid returns the unit of a column in a haystack grid
+// It is expected that the column is from the grid
+// The unit is determined in the following order:
+// 1. If the column has a unit meta, return it
+// 2. If the column has a unit in the first row, return it
+// 3. Return empty string
+func unitFromGrid(grid haystack.Grid, col haystack.Col) string {
+	switch unit := col.Meta().Get("unit").(type) {
+	case haystack.Str:
+		return unit.String()
+	default:
+		if grid.RowCount() >= 1 {
+			row := grid.RowAt(0)
+			val := row.Get(col.Name())
+			switch val := val.(type) {
+			case haystack.Number:
+				return val.Unit()
+			}
+		}
+		return ""
+	}
+}
+
 type colType int
 
 // colType represents the type of a column in a haystack grid

pkg/plugin/datasource_test.go

@@ -66,7 +66,7 @@ func TestQueryData_HisRead(t *testing.T) {
 	v0Val := 5.0
 	expected := data.NewFrame("",
 		data.NewField("ts", nil, []*time.Time{&tsVal}).SetConfig(&data.FieldConfig{DisplayName: "ts"}),
-		data.NewField("v0", nil, []*float64{&v0Val}).SetConfig(&data.FieldConfig{DisplayName: "v0"}),
+		data.NewField("v0", nil, []*float64{&v0Val}).SetConfig(&data.FieldConfig{DisplayName: "v0", Unit: "kWh"}),
 	)
 
 	if !cmp.Equal(actual, expected, data.FrameTestCompareOptions()...) {