add go bench client

backend/app/utils/pci_info_parser.py

@@ -0,0 +1,79 @@
+"""
+PCI Information Parser
+Combines lspci -v and lspci -n outputs to get complete device information.
+"""
+import re
+import subprocess
+from typing import Dict, Any, Optional
+
+
+def get_pci_ids_from_lspci_n(lspci_n_output: str) -> Dict[str, str]:
+    """
+    Parse lspci -n output to extract vendor:device IDs for all slots.
+
+    Args:
+        lspci_n_output: Output from 'lspci -n' command
+
+    Returns:
+        Dictionary mapping slot -> vendor:device ID
+        Example: {"04:00.0": "10ec:8168", "08:00.0": "10de:2504"}
+    """
+    slot_to_id = {}
+    lines = lspci_n_output.strip().split('\n')
+
+    for line in lines:
+        # Format: "04:00.0 0200: 10ec:8168 (rev 16)"
+        # Format: "00:00.0 0600: 1022:1480"
+        match = re.match(r'^([0-9a-fA-F]{2}:[0-9a-fA-F]{2}\.[0-9a-fA-F])\s+[0-9a-fA-F]+:\s+([0-9a-fA-F]{4}):([0-9a-fA-F]{4})', line)
+        if match:
+            slot = match.group(1)
+            vendor_id = match.group(2).lower()
+            device_id = match.group(3).lower()
+            slot_to_id[slot] = f"{vendor_id}:{device_id}"
+
+    return slot_to_id
+
+
+def enrich_device_info_with_ids(device_info: Dict[str, Any], pci_ids: Dict[str, str]) -> Dict[str, Any]:
+    """
+    Enrich device info with vendor:device ID from lspci -n output.
+
+    Args:
+        device_info: Parsed device information from lspci -v
+        pci_ids: Mapping from slot to vendor:device ID
+
+    Returns:
+        Enriched device info with pci_device_id field
+    """
+    slot = device_info.get("slot")
+    if slot and slot in pci_ids:
+        device_info["pci_device_id"] = pci_ids[slot]
+        # Also split into vendor_id and device_id
+        parts = pci_ids[slot].split(':')
+        if len(parts) == 2:
+            device_info["vendor_id"] = f"0x{parts[0]}"
+            device_info["device_id"] = f"0x{parts[1]}"
+
+    return device_info
+
+
+def run_lspci_n() -> Optional[str]:
+    """
+    Run lspci -n command and return output.
+    This is a helper function that executes the command.
+
+    Returns:
+        Output from lspci -n or None if command fails
+    """
+    try:
+        result = subprocess.run(
+            ['lspci', '-n'],
+            capture_output=True,
+            text=True,
+            timeout=10
+        )
+        if result.returncode == 0:
+            return result.stdout
+        return None
+    except Exception:
+        return None