优化代码结构,增加解析gpu直通,audio解析等逻辑

pkg-agent/agent_constant/kvm_client.go

@@ -5,15 +5,16 @@ import (
 	"github.com/pkg/errors"
 	"libvirt.org/go/libvirt"
 	libvirtxml "libvirt.org/go/libvirtxml"
+	"strconv"
 	"strings"
 )
 
 type QemuClient interface {
-	CreateVM() error
+	CreateVM(vmName string, memoryMB uint, vcpu uint, diskImagePath string, gpuPCIAddress string, audioPCIAddress string) error
 	StartVM() error
 	StopVM() error
 	DestroyVM() error
-	GetAllDomainStats()  ([]libvirt.DomainStats,error)
+	GetAllDomainStats() ([]libvirt.DomainStats, error)
 	ListAllDomains(vName string)
 }
 
@@ -40,7 +41,7 @@ func NewClient() (QemuClient, error) {
 }
 
 // 生成虚拟机 XML 配置（含 GPU 直通）
-func generateVMXML() string {
+func generateVMXML(vmName string, memoryMB uint, vcpu uint, diskImagePath string, gpuPCIAddress DomainAddressPCI, audioPCIAddress DomainAddressPCI) string {
 	domainCfg := &libvirtxml.Domain{
 		Type:        "kvm",
 		Name:        vmName,
@@ -81,22 +82,29 @@ func generateVMXML() string {
 			},
 			Hostdevs: []libvirtxml.DomainHostdev{
 				{
-
-					Mode:    "subsystem",
-					Type:    "pci",
 					Managed: "yes",
-					Source: &libvirtxml.DomainHostdevSource{
-						Address: &libvirtxml.DomainAddress{
-							PCI: &libvirtxml.DomainAddressPCI{
-								Domain:   parsePCIComponent(gpuPCIAddress, 0),
-								Bus:      parsePCIComponent(gpuPCIAddress, 1),
-								Slot:     parsePCIComponent(gpuPCIAddress, 2),
-								Function: parsePCIComponent(gpuPCIAddress, 3),
+					SubsysPCI: &libvirtxml.DomainHostdevSubsysPCI{
+						Source: &libvirtxml.DomainHostdevSubsysPCISource{
+							Address: &libvirtxml.DomainAddressPCI{
+								Domain:   gpuPCIAddress.domain,
+								Bus:      gpuPCIAddress.bus,
+								Slot:     gpuPCIAddress.slot,
+								Function: gpuPCIAddress.function,
 							},
 						},
 					},
+				},
+				{
+					Managed: "yes",
 					SubsysPCI: &libvirtxml.DomainHostdevSubsysPCI{
-						Add
+						Source: &libvirtxml.DomainHostdevSubsysPCISource{
+							Address: &libvirtxml.DomainAddressPCI{
+								Domain:   audioPCIAddress.domain,
+								Bus:      audioPCIAddress.bus,
+								Slot:     audioPCIAddress.slot,
+								Function: audioPCIAddress.function,
+							},
+						},
 					},
 				},
 			},
@@ -107,31 +115,78 @@ func generateVMXML() string {
 	return xml
 }
 
-// 解析PCI地址组件（例如 "0000:01:00.0" -> 各部分转换为十六进制）
-func parsePCIComponent(addr string, index int) string {
+type DomainAddressPCI struct {
+	domain, bus, slot, function *uint
+}
+
+// 解析完整PCI地址（支持0000:3b:00.0 和 3b:00.0两种格式）
+func parsePCIAddress(addr string) (DomainAddressPCI, error) {
 	components := strings.Split(addr, ":")
-	if len(components) < 3 {
-		return "0x00"
+	var (
+		domainStr   = "0000" // 默认domain
+		busStr      string
+		slotFuncStr string
+	)
+
+	// 处理带domain的格式 (0000:3b:00.0)
+	if len(components) == 3 {
+		domainStr = components[0]
+		busStr = components[1]
+		slotFuncStr = components[2]
+	} else if len(components) == 2 { // 无domain格式 (3b:00.0)
+		busStr = components[0]
+		slotFuncStr = components[1]
+	} else {
+		return DomainAddressPCI{}, fmt.Errorf("invalid PCI address format: %s", addr)
+	}
+
+	// 解析slot和function
+	slotFuncParts := strings.Split(slotFuncStr, ".")
+	if len(slotFuncParts) != 2 {
+		return DomainAddressPCI{}, fmt.Errorf("invalid slot.function format: %s", slotFuncStr)
 	}
-	slotFunc := strings.Split(components[2], ".")
-	if index == 3 && len(slotFunc) > 1 {
-		return "0x" + slotFunc[1]
+	slotStr, functionStr := slotFuncParts[0], slotFuncParts[1]
+
+	// 转换为uint (16进制)
+	parseHex := func(s string) (uint, error) {
+		val, err := strconv.ParseUint(s, 16, 16)
+		if err != nil {
+			return 0, fmt.Errorf("parse error: %s", s)
 		}
-	switch index {
-	case 0:
-		return "0x" + components[0]
-	case 1:
-		return "0x" + components[1]
-	case 2:
-		return "0x" + slotFunc[0]
-	default:
-		return "0x00"
+		return uint(val), nil
+	}
+	//var domain, bus, slot, function uint
+	domain, err := parseHex(domainStr)
+	if err != nil {
+		return DomainAddressPCI{}, err
 	}
+	bus, err := parseHex(busStr)
+	if err != nil {
+		return DomainAddressPCI{}, err
+	}
+	slot, err := parseHex(slotStr)
+	if err != nil {
+		return DomainAddressPCI{}, err
+	}
+	function, err := parseHex(functionStr)
+	if err != nil {
+		return DomainAddressPCI{}, err
+	}
+
+	return DomainAddressPCI{domain: &domain, bus: &bus, slot: &slot, function: &function}, nil
 }
 
 // 创建虚拟机
-func (c Client) CreateVM() error {
-	xml := generateVMXML()
+func (c Client) CreateVM(vmName string, memoryMB uint, vcpu uint, diskImagePath string, gpuPCIAddress string, audioPCIAddress string) error {
+	pciAddress, err := parsePCIAddress(gpuPCIAddress)
+	if err != nil {
+		return err
+	}
+	audioAddress, err := parsePCIAddress(audioPCIAddress)
+	if err != nil {
+		return err
+	}
+	xml := generateVMXML(vmName, memoryMB, vcpu, diskImagePath, pciAddress, audioAddress)
 	dom, err := c.connect.DomainDefineXML(xml)
 	if err != nil {
 		return errors.Wrap(err, "Failed to define domain")
@@ -197,18 +252,16 @@ func (c Client) DestroyVM() error {
 	return nil
 }
 
-
-
 // 通过虚拟机名称获取其全部状态
-func (c Client) GetAllDomainStats() ([]libvirt.DomainStats,error) {
+func (c Client) GetAllDomainStats() ([]libvirt.DomainStats, error) {
 	dom, err := c.connect.LookupDomainByName(vmName)
 	if err != nil {
-		return nil,errors.Wrap(err, "Failed to find domain")
+		return nil, errors.Wrap(err, "Failed to find domain")
 	}
 	defer func(dom *libvirt.Domain) {
 		err := dom.Free()
 		if err != nil {
-			fmt.Printf("err:%v \n",err)
+			fmt.Printf("err:%v \n", err)
 		}
 	}(dom)
 	dobs := make([]*libvirt.Domain, 0)