spdksetup: add status subcommand

Signed-off-by: Derek Su <[email protected]>

Author: derekbit

app/cmd/spdksetup/spdksetup.go

@@ -26,6 +26,7 @@ func Cmd() cli.Command {
 			BindCmd(),
 			UnbindCmd(),
 			GetDeviceDriverCmd(),
+			GetStatusCmd(),
 		},
 	}
 }
@@ -112,7 +113,36 @@ func getDeviceDriverCmd(c *cli.Context) error {
 		return err
 	}
 
-	logrus.Infof("Device driver of device %v is %v", deviceAddr, output)
+	fmt.Println(output)
+
+	return nil
+}
+
+func GetStatusCmd() cli.Command {
+	return cli.Command{
+		Name:  "status",
+		Flags: []cli.Flag{},
+		Usage: "Get status of all SPDK-compatible devices on the system",
+		Action: func(c *cli.Context) {
+			if err := getStatusCmd(c); err != nil {
+				logrus.WithError(err).Fatalf("Failed to get status of SPDK-compatible devices")
+			}
+		},
+	}
+}
+
+func getStatusCmd(c *cli.Context) error {
+	executor, err := util.NewExecutor(c.GlobalString("host-proc"))
+	if err != nil {
+		return err
+	}
+
+	output, err := spdksetup.GetStatus(executor)
+	if err != nil {
+		return err
+	}
+
+	fmt.Println(output)
 
 	return nil
 }

pkg/spdk/setup/setup.go

@@ -1,6 +1,7 @@
 package setup
 
 import (
+	"encoding/json"
 	"fmt"
 	"strings"
 
@@ -83,3 +84,126 @@ func extractJSON(outputStr string) (string, error) {
 	}
 	return "", fmt.Errorf("failed to extract JSON from output: %s", outputStr)
 }
+
+// DeviceInfo represents information about a device
+type DeviceInfo struct {
+	Type         string `json:"type"`
+	BDF          string `json:"bdf"`
+	Vendor       string `json:"vendor"`
+	Device       string `json:"device"`
+	NUMA         string `json:"numa"`
+	Driver       string `json:"driver"`
+	DeviceName   string `json:"device_name"`
+	BlockDevices string `json:"block_devices"`
+}
+
+// HugepagesInfo represents information about Hugepages
+type HugepagesInfo struct {
+	Node     string `json:"node"`
+	Hugesize string `json:"hugesize"`
+	Free     int    `json:"free"`
+	Total    int    `json:"total"`
+}
+
+// StatusInfo represents the complete status information
+type StatusInfo struct {
+	Hugepages   []HugepagesInfo `json:"hugepages"`
+	DeviceInfos []DeviceInfo    `json:"device_infos"`
+}
+
+func GetStatus(executor *commonNs.Executor) (string, error) {
+	cmdArgs := []string{
+		spdkSetupPath,
+		"status",
+	}
+
+	outputStr, err := executor.Execute(nil, "bash", cmdArgs, types.ExecuteTimeout)
+	if err != nil {
+		return "", err
+	}
+
+	status := parseStatus(outputStr)
+	jsonString, err := json.Marshal(status)
+	if err != nil {
+		return "", err
+	}
+
+	return string(jsonString), nil
+}
+
+func parseStatus(output string) StatusInfo {
+	var status StatusInfo
+	lines := strings.Split(output, "\n")
+
+	hugepagesIndex := -1
+	deviceInfoIndex := -1
+
+	for i, line := range lines {
+		if line == "Hugepages" {
+			hugepagesIndex = i
+		} else if line == "Type     BDF             Vendor Device NUMA    Driver           Device     Block devices" {
+			deviceInfoIndex = i
+			break
+		}
+	}
+
+	if hugepagesIndex != -1 {
+		status.Hugepages = parseHugepages(lines[hugepagesIndex+2:])
+	}
+
+	if deviceInfoIndex != -1 {
+		status.DeviceInfos = parseDeviceInfos(lines[deviceInfoIndex+1:])
+	}
+
+	return status
+}
+
+func parseHugepages(lines []string) []HugepagesInfo {
+	var hugepages []HugepagesInfo
+
+	for _, line := range lines {
+		fields := strings.Fields(line)
+		if len(fields) != 5 {
+			continue
+		}
+
+		hugepages = append(hugepages, HugepagesInfo{
+			Node:     fields[0],
+			Hugesize: fields[1],
+			Free:     parseInteger(fields[3]),
+			Total:    parseInteger(fields[5]),
+		})
+	}
+
+	return hugepages
+}
+
+func parseDeviceInfos(lines []string) []DeviceInfo {
+	var deviceInfos []DeviceInfo
+
+	for _, line := range lines {
+		fields := strings.Fields(line)
+		if len(fields) != 8 {
+			continue
+		}
+
+		deviceInfos = append(deviceInfos, DeviceInfo{
+			Type:         fields[0],
+			BDF:          fields[1],
+			Vendor:       fields[2],
+			Device:       fields[3],
+			NUMA:         fields[4],
+			Driver:       fields[5],
+			DeviceName:   fields[6],
+			BlockDevices: fields[7],
+		})
+	}
+
+	return deviceInfos
+}
+
+func parseInteger(s string) int {
+	var num int
+	fmt.Sscanf(s, "%d", &num)
+	return num
+}