| | |
|---|
| | | "vamicro/extend/util" |
|---|
| | | |
|---|
| | | "basic.com/valib/gogpu.git" |
|---|
| | | |
|---|
| | | "basic.com/valib/logger.git" |
|---|
| | | "github.com/shirou/gopsutil/cpu" |
|---|
| | | "github.com/shirou/gopsutil/disk" |
|---|
| | | "github.com/shirou/gopsutil/host" |
|---|
| | |
|---|
| | | 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() |
|---|
| | | gpuInfo, err := gogpu.Info() |
|---|
| | | var gpuMemTotal int64 |
|---|
| | | if err == nil { |
|---|
| | | for _, gpu := range gpuInfo.Info { |
|---|
| | | gpuMemTotal = gpuMemTotal + gpu.GpuMemoryTotal |
|---|
| | | } |
|---|
| | | } else { |
|---|
| | | logger.Debug("gpu error " + err.Error()) |
|---|
| | | } |
|---|
| | | gpu := gpuMemTotal |
|---|
| | | mem, _ := mem.VirtualMemory() |
|---|
| | | host, _ := host.Info() |
|---|
| | | |
|---|
| | | return util.Struct2Map(map[string]interface{}{ |
|---|
| | | "cpu": cpu, |
|---|
| | | "gpu": gpu, |
|---|
| | | "mem": mem, |
|---|
| | | "host": host, |
|---|
| | | "disk": string(disks), |
|---|
