pdu telemetry

Signed-off-by: Harper, Jason M <[email protected]>

Author: harp-intel

internal/report/render_html.go

@@ -1542,11 +1542,31 @@ func gaudiTelemetryTableHTMLRenderer(tableValues TableValues, targetName string)
 
 func pduTelemetryTableHTMLRenderer(tableValues TableValues, targetName string) string {
 	data := [][]float64{}
+	for _, field := range tableValues.Fields[1:] {
+		points := []float64{}
+		for _, val := range field.Values {
+			if val == "" {
+				break
+			}
+			stat, err := strconv.ParseFloat(val, 64)
+			if err != nil {
+				slog.Error("error parsing stat", slog.String("error", err.Error()))
+				return ""
+			}
+			points = append(points, stat)
+		}
+		if len(points) > 0 {
+			data = append(data, points)
+		}
+	}
 	datasetNames := []string{}
+	for _, field := range tableValues.Fields[1:] {
+		datasetNames = append(datasetNames, field.Name)
+	}
 	chartConfig := chartTemplateStruct{
 		ID:            fmt.Sprintf("%s%d", tableValues.Name, util.RandUint(10000)),
 		XaxisText:     "Time",
-		YaxisText:     "Value",
+		YaxisText:     "Watts",
 		TitleText:     "",
 		DisplayTitle:  "false",
 		DisplayLegend: "true",

internal/report/table_defs.go

@@ -2768,7 +2768,37 @@ func gaudiTelemetryTableValues(outputs map[string]script.ScriptOutput) []Field {
 }
 
 func pduTelemetryTableValues(outputs map[string]script.ScriptOutput) []Field {
-	return []Field{}
+	// extract PDU fields and their values from PDU telemetry script output
+	// output is CSV formatted:
+	//   Timestamp,ActivePower(W)
+	//   18:32:38,123.45
+	//   18:32:40,124.10
+	//   ...
+	fields := []Field{}
+	reader := csv.NewReader(strings.NewReader(outputs[script.PDUTelemetryScriptName].Stdout))
+	records, err := reader.ReadAll()
+	if err != nil {
+		slog.Error("failed to read PDU telemetry CSV output", slog.String("error", err.Error()))
+		return []Field{}
+	}
+	if len(records) == 0 {
+		return []Field{}
+	}
+	// first row is the header
+	for _, header := range records[0] {
+		fields = append(fields, Field{Name: header, Values: []string{}})
+	}
+	// subsequent rows are data
+	for _, record := range records[1:] {
+		if len(record) != len(fields) {
+			slog.Error("unexpected number of fields in PDU telemetry output", slog.Int("expected", len(fields)), slog.Int("got", len(record)))
+			return []Field{}
+		}
+		for i, value := range record {
+			fields[i].Values = append(fields[i].Values, value)
+		}
+	}
+	return fields
 }
 
 func callStackFrequencyTableValues(outputs map[string]script.ScriptOutput) []Field {

internal/script/script_defs.go

@@ -1326,10 +1326,39 @@ fi
 		NeedsKill: true,
 	},
 	PDUTelemetryScriptName: {
-		Name:           PDUTelemetryScriptName,
-		ScriptTemplate: ``,
-		Superuser:      false,
-		NeedsKill:      false,
+		Name: PDUTelemetryScriptName,
+		ScriptTemplate: `
+duration={{.Duration}}       # total duration in seconds
+interval={{.Interval}}       # time between readings in seconds
+pdu="{{.PDUHost}}"           # PDU hostname or IP address (must not start with protocol like http://, may include port)
+pdu_ip=$(echo $pdu | awk -F/ '{print $NF}' | awk -F: '{print $1}') # remove http:// or https:// and port if present
+pdu_username="{{.PDUUser}}"
+pdu_password="{{.PDUPassword}}"
+outletgroup="{{.PDUOutlet}}"
+count=$((duration / interval))
+echo "Timestamp,ActivePower(W)"
+for ((i=0; i<count; i++)); do
+    timestamp=$(date +%H:%M:%S)
+    response=$(no_proxy=$pdu_ip curl --max-time $interval -sS -k -u "$pdu_username:$pdu_password" \
+        -d "{'jsonrpc':'2.0','method':'getReading'}" \
+        "https://${pdu}/model/outletgroup/${outletgroup}/activePower")
+    # Check if curl succeeded
+    if [ $? -ne 0 ] || [ -z "$response" ]; then
+        echo "ERROR: Failed to retrieve data from PDU" >&2
+        exit 1
+    fi
+    # Try to parse the value using jq
+    w=$(echo "$response" | jq -r '.result._ret_.value // empty')
+    if [ -z "$w" ]; then
+        echo "ERROR: Invalid response format or missing value" >&2
+        exit 1
+    fi
+    echo "$timestamp,$w"
+    sleep $interval
+done
+`,
+		Superuser: false,
+		NeedsKill: false,
 	},
 	// flamegraph scripts
 	CollapsedCallStacksScriptName: {