package sys
|
|
import (
|
"errors"
|
"fmt"
|
"io/ioutil"
|
"os"
|
"os/exec"
|
"sort"
|
"strconv"
|
"strings"
|
"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 GetGpu(key string) (vGpu []string, err error) {
|
resultMsg, e := util.ExeCmd("dlsmi --query-gpu=" + key)
|
if e != nil {
|
return vGpu, errors.New("无可用GPU!请检查是否正常安显卡及显卡驱动，通过dlsmi查询状态")
|
}
|
vGpu = strings.Split(string(resultMsg), "\n")
|
if len(vGpu) == 0 {
|
return vGpu, errors.New("无可用GPU!请检查是否正常安显卡及显卡驱动，通过dlsmi查询状态")
|
}
|
return vGpu, nil
|
}
|
|
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()
|
|
var gpuStr string
|
|
//获取型号
|
gpuName, _ := GetGpu("name")
|
if len(gpuName) > 0 {
|
var gpuArr []string = []string{}
|
gpuTotal, _ := GetGpu("memory.total")
|
for i, name := range gpuName {
|
//i>0跳过header
|
if i > 0 && len(name) > 0 {
|
gpuArr = append(gpuArr, fmt.Sprintf("%s %s", name, gpuTotal[i]))
|
}
|
}
|
gpuStr = strings.Join(gpuArr, ",")
|
}
|
|
mem, _ := mem.VirtualMemory()
|
host, _ := host.Info()
|
|
return util.Struct2Map(map[string]interface{}{
|
"cpu": cpu,
|
"gpu": gpuStr,
|
"mem": mem,
|
"host": host,
|
"disk": string(disks),
|
})
|
}
|
|
func GetDeviceTemp() string {
|
data, err := ioutil.ReadFile("/sys/class/thermal/thermal_zone0/temp")
|
if err != nil {
|
fmt.Println("Error reading temperature:", err)
|
return ""
|
}
|
|
temperature, err := strconv.Atoi(strings.TrimSpace(string(data)))
|
if err != nil {
|
fmt.Println("Error parsing temperature:", err)
|
return ""
|
}
|
|
tempCelsius := float64(temperature) / 1000.0
|
|
tempStr := fmt.Sprintf("%.2f°C", tempCelsius)
|
|
return tempStr
|
}
|
