update monitor.go

xOS · Jan 5, 2024 · 0d7cfe7 · 0d7cfe7
1 parent 38c1605
commit 0d7cfe7
Showing 1 changed file with 6 additions and 8 deletions.
diff --git a/cmd/agent/monitor/monitor.go b/cmd/agent/monitor/monitor.go
@@ -58,17 +58,15 @@ func GetHost(agentConfig *model.AgentConfig) *model.Host {
 		ret.BootTime = hi.BootTime
 	}
 
-	cpuModelCount := make(map[string]int)
 	ci, err := cpu.Info()
 	if err != nil {
 		println("cpu.Info error:", err)
-	} else {
-		for i := 0; i < len(ci); i++ {
-			cpuModelCount[ci[i].ModelName]++
-		}
-		for model, count := range cpuModelCount {
-			ret.CPU = append(ret.CPU, fmt.Sprintf("%s %d %s Core", model, count, cpuType))
-		}
+	}
+	count, err := cpu.Counts(false)
+	if err != nil {
+		println("cpu.Counts error:", err)
+	} else if len(ci) > 0 {
+		ret.CPU = append(ret.CPU, fmt.Sprintf("%s %d %s Core", ci[0].ModelName, count, cpuType))
 	}
 
 	ret.DiskTotal, _ = getDiskTotalAndUsed(agentConfig)