package sys
|
|
import (
|
"fmt"
|
"os"
|
"os/exec"
|
"sort"
|
"time"
|
"vamicro/extend/util"
|
|
"basic.com/valib/gogpu.git"
|
|
"github.com/shirou/gopsutil/cpu"
|
"github.com/shirou/gopsutil/disk"
|
"github.com/shirou/gopsutil/host"
|
"github.com/shirou/gopsutil/mem"
|
)
|
|
type SummaryInfo struct {
|
Cpu []float64 `json:"cpu" example:"cpu使用率"`
|
Gpu []gogpu.GpuUnitInfo `json:"gpu" example:"gpu使用率"`
|
Mem mem.VirtualMemoryStat `json:"mem" example:"内存使用率"`
|
Disk disk.UsageStat `json:"disk" example:"磁盘空间"`
|
}
|
|
const Interval time.Duration = time.Second * 60
|
|
var Stat = &SummaryInfo{}
|
|
func GatherStat() {
|
for range time.Tick(Interval) {
|
// Cpu
|
if cpuUsedPercent, err := cpu.Percent(time.Second, true); err == nil {
|
Stat.Cpu = Stat.Cpu[0:0]
|
for _, v := range cpuUsedPercent {
|
Stat.Cpu = append(Stat.Cpu, v)
|
}
|
}
|
|
// Memory
|
if meminfo, err := mem.VirtualMemory(); err == nil {
|
Stat.Mem = *meminfo
|
}
|
|
// Disk
|
if disk, err := disk.Usage("/"); err == nil {
|
Stat.Disk = *disk
|
}
|
|
// Gpu
|
agxGpuProcIface := "/sys/devices/gpu.0/load"
|
|
if util.Exists(agxGpuProcIface) {
|
fd, err1 := os.Open(agxGpuProcIface)
|
if err1 == nil {
|
var load int64
|
fmt.Fscanf(fd, "%d", &load)
|
// 模拟nvidia库的输出
|
Stat.Gpu = append(Stat.Gpu, gogpu.GpuUnitInfo{
|
GpuMemoryTotal: 1000,
|
GpuMemoryUsed: load,
|
})
|
fd.Close()
|
}
|
} else {
|
if gpuInfo, err := gogpu.Info(); err == nil {
|
sort.Sort(gogpu.GPURank(gpuInfo.Info))
|
Stat.Gpu = Stat.Gpu[0:0]
|
for _, v := range gpuInfo.Info {
|
Stat.Gpu = append(Stat.Gpu, v)
|
}
|
}
|
}
|
}
|
}
|
|
func GetSysInfo() map[string]interface{} {
|
return util.Struct2Map(Stat)
|
}
|
|
func GetDeviceInfo() map[string]interface{} {
|
cmd := exec.Command("/bin/sh", "-c", "lsblk -d | grep -v part | grep -v SWAP | grep -v M | grep disk | awk '{printf $4\" \"}'")
|
disks, _ := cmd.Output()
|
cpu, _ := cpu.Info()
|
mem, _ := mem.VirtualMemory()
|
host, _ := host.Info()
|
|
return util.Struct2Map(map[string]interface{}{
|
"cpu": cpu,
|
"mem": mem,
|
"host": host,
|
"disk": string(disks),
|
})
|
}
|
