🚸 chore: refactor CPU model string

This commit is contained in:
naiba 2021-02-27 19:24:19 +08:00
parent 2133973579
commit dbf62991a5
5 changed files with 44 additions and 24 deletions

View File

@ -1,6 +1,6 @@
# 哪吒监控 # 哪吒监控
![GitHub Workflow Status](https://img.shields.io/github/workflow/status/naiba/nezha/Dashboard%20image?label=管理面板%20v0.4.9&logo=github&style=for-the-badge) ![Agent release](https://img.shields.io/github/v/release/naiba/nezha?color=brightgreen&label=Agent&style=for-the-badge&logo=github) ![GitHub Workflow Status](https://img.shields.io/github/workflow/status/naiba/nezha/Agent%20release?label=Agent%20CI&logo=github&style=for-the-badge) ![shell](https://img.shields.io/badge/安装脚本-v0.4.7-brightgreen?style=for-the-badge&logo=linux) ![GitHub Workflow Status](https://img.shields.io/github/workflow/status/naiba/nezha/Dashboard%20image?label=管理面板%20v0.4.10&logo=github&style=for-the-badge) ![Agent release](https://img.shields.io/github/v/release/naiba/nezha?color=brightgreen&label=Agent&style=for-the-badge&logo=github) ![GitHub Workflow Status](https://img.shields.io/github/workflow/status/naiba/nezha/Agent%20release?label=Agent%20CI&logo=github&style=for-the-badge) ![shell](https://img.shields.io/badge/安装脚本-v0.4.7-brightgreen?style=for-the-badge&logo=linux)
:trollface: 哪吒监控 一站式轻监控轻运维系统。支持系统状态、HTTP(SSL 证书变更、即将到期、到期)、TCP、Ping 监控报警,命令批量执行和计划任务。 :trollface: 哪吒监控 一站式轻监控轻运维系统。支持系统状态、HTTP(SSL 证书变更、即将到期、到期)、TCP、Ping 监控报警,命令批量执行和计划任务。

View File

@ -12,15 +12,17 @@ import (
"github.com/go-ping/ping" "github.com/go-ping/ping"
"github.com/naiba/nezha/pkg/utils" "github.com/naiba/nezha/pkg/utils"
"github.com/shirou/gopsutil/v3/cpu"
"github.com/shirou/gopsutil/v3/disk" "github.com/shirou/gopsutil/v3/disk"
"github.com/shirou/gopsutil/v3/host"
) )
func main() { func main() {
// icmp() // icmp()
// tcpping() // tcpping()
// httpWithSSLInfo() // httpWithSSLInfo()
// diskinfo() sysinfo()
cmdExec() // cmdExec()
} }
func tcpping() { func tcpping() {
@ -34,8 +36,26 @@ func tcpping() {
fmt.Println(time.Now().Sub(start).Microseconds(), float32(time.Now().Sub(start).Microseconds())/1000.0) fmt.Println(time.Now().Sub(start).Microseconds(), float32(time.Now().Sub(start).Microseconds())/1000.0)
} }
func diskinfo() { func sysinfo() {
// 硬盘信息 hi, _ := host.Info()
var cpuType string
if hi.VirtualizationSystem != "" {
cpuType = "Virtual"
} else {
cpuType = "Physical"
}
cpuModelCount := make(map[string]int)
ci, _ := cpu.Info()
for i := 0; i < len(ci); i++ {
cpuModelCount[ci[i].ModelName]++
}
var cpus []string
for model, count := range cpuModelCount {
cpus = append(cpus, fmt.Sprintf("%s %d %s Core", model, count, cpuType))
}
log.Println(cpus)
os.Exit(0)
// 硬盘信息,不使用的原因是会重复统计 Linux、Mac
dparts, _ := disk.Partitions(false) dparts, _ := disk.Partitions(false)
for _, part := range dparts { for _, part := range dparts {
u, _ := disk.Usage(part.Mountpoint) u, _ := disk.Usage(part.Mountpoint)

View File

@ -1,8 +1,6 @@
package model package model
import ( import (
"fmt"
pb "github.com/naiba/nezha/proto" pb "github.com/naiba/nezha/proto"
) )
@ -85,22 +83,10 @@ func (h *Host) PB() *pb.Host {
} }
func PB2Host(h *pb.Host) Host { func PB2Host(h *pb.Host) Host {
cpuCount := make(map[string]int, 0)
cpus := h.GetCpu()
for _, u := range cpus {
cpuCount[u]++
}
var distCpu []string
for u, num := range cpuCount {
distCpu = append(distCpu, fmt.Sprintf("%sx%d", u, num))
}
return Host{ return Host{
Platform: h.GetPlatform(), Platform: h.GetPlatform(),
PlatformVersion: h.GetPlatformVersion(), PlatformVersion: h.GetPlatformVersion(),
CPU: distCpu, CPU: h.GetCpu(),
MemTotal: h.GetMemTotal(), MemTotal: h.GetMemTotal(),
DiskTotal: h.GetDiskTotal(), DiskTotal: h.GetDiskTotal(),
SwapTotal: h.GetSwapTotal(), SwapTotal: h.GetSwapTotal(),

View File

@ -13,7 +13,7 @@ import (
pb "github.com/naiba/nezha/proto" pb "github.com/naiba/nezha/proto"
) )
var Version = "v0.4.9" // !!记得修改 README 重的 badge 版本!! var Version = "v0.4.10" // !!记得修改 README 重的 badge 版本!!
const ( const (
SnapshotDelay = 3 SnapshotDelay = 3

View File

@ -28,10 +28,20 @@ var netInSpeed, netOutSpeed, netInTransfer, netOutTransfer, lastUpdate uint64
func GetHost() *model.Host { func GetHost() *model.Host {
hi, _ := host.Info() hi, _ := host.Info()
var cpus []string var cpuType string
if hi.VirtualizationSystem != "" {
cpuType = "Virtual"
} else {
cpuType = "Physical"
}
cpuModelCount := make(map[string]int)
ci, _ := cpu.Info() ci, _ := cpu.Info()
for i := 0; i < len(ci); i++ { for i := 0; i < len(ci); i++ {
cpus = append(cpus, fmt.Sprintf("%v-%vC%vT", ci[i].ModelName, ci[i].Cores, ci[i].Stepping)) cpuModelCount[ci[i].ModelName]++
}
var cpus []string
for model, count := range cpuModelCount {
cpus = append(cpus, fmt.Sprintf("%s %d %s Core", model, count, cpuType))
} }
mv, _ := mem.VirtualMemory() mv, _ := mem.VirtualMemory()
ms, _ := mem.SwapMemory() ms, _ := mem.SwapMemory()
@ -47,7 +57,11 @@ func GetHost() *model.Host {
if err == nil { if err == nil {
defer resp.Body.Close() defer resp.Body.Close()
body, _ := ioutil.ReadAll(resp.Body) body, _ := ioutil.ReadAll(resp.Body)
ip.IP = fmt.Sprintf("ip(v4: %s, v6: %s)", ip.IP, body) if ip.IP == "" {
ip.IP = string(body)
} else {
ip.IP = fmt.Sprintf("ip(v4: %s, v6: %s)", ip.IP, body)
}
} }
return &model.Host{ return &model.Host{
Platform: hi.OS, Platform: hi.OS,