feat: 优化CPU名称获取

monitoring/unit/cpu.go

@@ -1,6 +1,9 @@
 package monitoring
 
 import (
+	"bufio"
+	"os"
+	"os/exec"
 	"runtime"
 	"strings"
 	"time"
@@ -16,49 +19,90 @@ type CpuInfo struct {
 }
 
 func Cpu() CpuInfo {
-	cpuinfo := CpuInfo{}
-	info, err := cpu.Info()
-	if err != nil {
-		cpuinfo.CPUName = "Unknown"
+	cpuinfo := CpuInfo{
+		CPUName:         "Unknown",
+		CPUArchitecture: runtime.GOARCH,
+		CPUCores:        1,
+		CPUUsage:        0.0,
 	}
-	/*
-		// multiple CPU
-		// 多个 CPU
-		if len(info) > 1 {
-			cpuCountMap := make(map[string]int)
-			for _, cpu := range info {
-				cpuCountMap[cpu.ModelName]++
-			}
-			for modelName, count := range cpuCountMap {
-				if count > 1 {
-					cpuinfo.CPUName += modelName + " x " + strconv.Itoa(count) + ", "
+
+	// 优先使用 lscpu 获取 CPU 信息
+	name, err := readCPUNameFromLscpu()
+	if err == nil && name != "" {
+		cpuinfo.CPUName = strings.TrimSpace(name)
 	} else {
-					cpuinfo.CPUName += modelName + ", "
+		// 如果 lscpu 无法获取 CPU 名称，尝试使用 gopsutil
+		info, err := cpu.Info()
+		if err == nil && len(info) > 0 {
+			cpuinfo.CPUName = strings.TrimSpace(info[0].ModelName)
+			if cpuinfo.CPUName == "" {
+				if info[0].VendorID != "" || info[0].Family != "" {
+					cpuinfo.CPUName = strings.TrimSpace(info[0].VendorID + " " + info[0].Family)
+				}
 			}
 		}
-			cpuinfo.CPUName = cpuinfo.CPUName[:len(cpuinfo.CPUName)-2] // Remove trailing comma and space
-		} else if len(info) == 1 {
-			cpuinfo.CPUName = info[0].ModelName
 	}
-	*/
-	cpuinfo.CPUName = info[0].ModelName
 
-	cpuinfo.CPUName = strings.TrimSpace(cpuinfo.CPUName)
-
-	cpuinfo.CPUArchitecture = runtime.GOARCH
+	if cpuinfo.CPUName == "Unknown" {
+		name, err := readCPUNameFromProc()
+		if err == nil && name != "" {
+			cpuinfo.CPUName = strings.TrimSpace(name)
+		}
+	}
 
 	cores, err := cpu.Counts(true)
-	if err != nil {
-		cpuinfo.CPUCores = 1 // Error case
-	}
+	if err == nil {
 		cpuinfo.CPUCores = cores
+	}
 
-	// Get CPU Usage
 	percentages, err := cpu.Percent(1*time.Second, false)
-	if err != nil {
-		cpuinfo.CPUUsage = 0.0 // Error case
-	} else {
+	if err == nil && len(percentages) > 0 {
 		cpuinfo.CPUUsage = percentages[0]
 	}
+
 	return cpuinfo
 }
+
+// readCPUNameFromLscpu 从 lscpu 命令读取 CPU 名称
+func readCPUNameFromLscpu() (string, error) {
+	cmd := exec.Command("lscpu")
+	output, err := cmd.Output()
+	if err != nil {
+		return "", err
+	}
+
+	scanner := bufio.NewScanner(strings.NewReader(string(output)))
+	for scanner.Scan() {
+		line := scanner.Text()
+		if strings.HasPrefix(line, "Model name:") {
+			parts := strings.SplitN(line, ":", 2)
+			if len(parts) == 2 {
+				return strings.TrimSpace(parts[1]), nil
+			}
+		}
+	}
+
+	return "", scanner.Err()
+}
+
+// readCPUNameFromProc 从 /proc/cpuinfo 读取 CPU 名称
+func readCPUNameFromProc() (string, error) {
+	file, err := os.Open("/proc/cpuinfo")
+	if err != nil {
+		return "", err
+	}
+	defer file.Close()
+
+	scanner := bufio.NewScanner(file)
+	for scanner.Scan() {
+		line := scanner.Text()
+		if strings.HasPrefix(line, "Model\t") || strings.HasPrefix(line, "Hardware\t") || strings.HasPrefix(line, "Processor\t") {
+			parts := strings.SplitN(line, ":", 2)
+			if len(parts) == 2 {
+				return strings.TrimSpace(parts[1]), nil
+			}
+		}
+	}
+
+	return "", scanner.Err()
+}