66 lines
1.5 KiB
Go
66 lines
1.5 KiB
Go
package services
|
|
|
|
import (
|
|
"bufio"
|
|
"cmii-uav-watchdog-common/models"
|
|
"fmt"
|
|
"os"
|
|
"strings"
|
|
)
|
|
|
|
// NewCPUInfo 创建一个默认的 CPUInfo
|
|
func NewCPUInfo() models.CPUInfo {
|
|
return models.CPUInfo{
|
|
ModelName: "unknown",
|
|
Cores: 0,
|
|
Architecture: "unknown",
|
|
}
|
|
}
|
|
|
|
// parseLine 解析单行 CPU 信息
|
|
func parseLine(line string, cpuInfo *models.CPUInfo) {
|
|
if strings.HasPrefix(line, "model name") {
|
|
cpuInfo.ModelName = strings.TrimSpace(strings.Split(line, ":")[1])
|
|
} else if strings.HasPrefix(line, "cpu cores") {
|
|
if _, err := fmt.Sscanf(line, "cpu cores\t: %d", &cpuInfo.Cores); err != nil {
|
|
fmt.Println("Error parsing cpu cores:", err)
|
|
}
|
|
} else if strings.HasPrefix(line, "Architecture") {
|
|
cpuInfo.Architecture = strings.TrimSpace(strings.Split(line, ":")[1])
|
|
}
|
|
}
|
|
|
|
// getCPUInfo 获取 CPU 信息
|
|
func GetCPUInfo() models.CPUInfo {
|
|
cpuInfo := NewCPUInfo()
|
|
|
|
file, err := os.Open("/proc/cpuinfo")
|
|
if err != nil {
|
|
fmt.Println("Error opening /proc/cpuinfo:", err)
|
|
return cpuInfo // 返回默认值
|
|
}
|
|
defer func() {
|
|
if err := file.Close(); err != nil {
|
|
fmt.Println("Error closing /proc/cpuinfo:", err)
|
|
}
|
|
}()
|
|
|
|
scanner := bufio.NewScanner(file)
|
|
for scanner.Scan() {
|
|
parseLine(scanner.Text(), &cpuInfo)
|
|
}
|
|
if err := scanner.Err(); err != nil {
|
|
fmt.Println("Error reading /proc/cpuinfo:", err)
|
|
return cpuInfo // 返回默认值
|
|
}
|
|
|
|
return cpuInfo
|
|
}
|
|
|
|
/*
|
|
CPU模型名称: Intel(R) Core(TM) i7-8565U CPU @ 1.80GHz
|
|
核心数量: 4
|
|
架构信息: x86_64
|
|
|
|
*/
|