From 833398cca0c266a0fb89b8feffafaa4b2170d5b2 Mon Sep 17 00:00:00 2001
From: UUBulb <35923940+uubulb@users.noreply.github.com>
Date: Sun, 12 May 2024 16:49:21 +0800
Subject: [PATCH] Fix ret.CPU detection on various platforms (#16)

---
 pkg/monitor/monitor.go | 14 +++++++++-----
 1 file changed, 9 insertions(+), 5 deletions(-)

diff --git a/pkg/monitor/monitor.go b/pkg/monitor/monitor.go
index 219445a..a78c825 100644
--- a/pkg/monitor/monitor.go
+++ b/pkg/monitor/monitor.go
@@ -67,11 +67,15 @@ func GetHost(agentConfig *model.AgentConfig) *model.Host {
 	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))
+		if runtime.GOOS == "linux" || runtime.GOOS == "windows" {
+			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))
+			}
+		} else {
+			ret.CPU = append(ret.CPU, fmt.Sprintf("%s %d %s Core", ci[0].ModelName, ci[0].Cores, cpuType))
 		}
 	}