106 lines
2.4 KiB
Go
106 lines
2.4 KiB
Go
package controller
|
|
|
|
import (
|
|
"net/http"
|
|
"time"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/shirou/gopsutil/v4/cpu"
|
|
"github.com/shirou/gopsutil/v4/disk"
|
|
"github.com/shirou/gopsutil/v4/host"
|
|
"github.com/shirou/gopsutil/v4/load"
|
|
"github.com/shirou/gopsutil/v4/mem"
|
|
)
|
|
|
|
type SystemInfo struct {
|
|
CPUUsage float64 `json:"cpu_usage"`
|
|
MemoryUsage float64 `json:"memory_usage"`
|
|
DiskUsage float64 `json:"disk_usage"`
|
|
LoadAvg float64 `json:"load_avg"`
|
|
CPUCores int `json:"cpu_cores"`
|
|
CPUMHz float64 `json:"cpu_mhz"`
|
|
MemoryTotal uint64 `json:"memory_total"`
|
|
MemoryFree uint64 `json:"memory_free"`
|
|
DiskPartitions []DiskPartition `json:"disk_partitions"`
|
|
Uptime uint64 `json:"uptime"`
|
|
}
|
|
|
|
type DiskPartition struct {
|
|
Mountpoint string `json:"mountpoint"`
|
|
Total uint64 `json:"total"`
|
|
Used uint64 `json:"used"`
|
|
UsedPercent float64 `json:"used_percent"`
|
|
}
|
|
|
|
func getSystemInfo() (SystemInfo, error) {
|
|
var info SystemInfo
|
|
|
|
// CPU 使用率
|
|
cpuPercent, err := cpu.Percent(time.Second, false)
|
|
if err == nil && len(cpuPercent) > 0 {
|
|
info.CPUUsage = cpuPercent[0]
|
|
}
|
|
|
|
// CPU 详细信息
|
|
cpuInfo, err := cpu.Info()
|
|
if err == nil && len(cpuInfo) > 0 {
|
|
info.CPUCores = int(cpuInfo[0].Cores)
|
|
info.CPUMHz = cpuInfo[0].Mhz
|
|
}
|
|
|
|
// 内存信息
|
|
vm, err := mem.VirtualMemory()
|
|
if err == nil {
|
|
info.MemoryUsage = vm.UsedPercent
|
|
info.MemoryTotal = vm.Total
|
|
info.MemoryFree = vm.Free
|
|
}
|
|
|
|
// 磁盘使用率（根目录）
|
|
diskUsage, err := disk.Usage("/")
|
|
if err == nil {
|
|
info.DiskUsage = diskUsage.UsedPercent
|
|
}
|
|
|
|
// 磁盘分区信息
|
|
partitions, err := disk.Partitions(false)
|
|
if err == nil {
|
|
for _, p := range partitions {
|
|
usage, err := disk.Usage(p.Mountpoint)
|
|
if err == nil {
|
|
info.DiskPartitions = append(info.DiskPartitions, DiskPartition{
|
|
Mountpoint: p.Mountpoint,
|
|
Total: usage.Total,
|
|
Used: usage.Used,
|
|
UsedPercent: usage.UsedPercent,
|
|
})
|
|
}
|
|
}
|
|
}
|
|
|
|
// 系统负载
|
|
loadAvg, err := load.Avg()
|
|
if err == nil {
|
|
info.LoadAvg = loadAvg.Load1
|
|
}
|
|
|
|
// 系统运行时间
|
|
uptime, err := host.Uptime()
|
|
if err == nil {
|
|
info.Uptime = uptime
|
|
}
|
|
|
|
return info, nil
|
|
}
|
|
|
|
func SystemInfoHandle(c *gin.Context) error {
|
|
info, err := getSystemInfo()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
c.JSON(http.StatusOK, info)
|
|
|
|
return nil
|
|
}
|