Skip to content

G
gpuhw
Commit 3dc1c565 authored by Your Name's avatar Your Name
Browse files
Options

filter disk device

parent 1f37349c
Show whitespace changes
Inline Side-by-side
Showing with 20 additions and 200 deletions
block-smi.go
View file @ 3dc1c565
...@@ -2,6 +2,7 @@ package main ...@@ -2,6 +2,7 @@ package main
import ( import (
"fmt" "fmt"
"strings"
"github.com/shirou/gopsutil/v3/disk" "github.com/shirou/gopsutil/v3/disk"
// "github.com/shirou/gopsutil/mem" // to use v2 // "github.com/shirou/gopsutil/mem" // to use v2
...@@ -17,6 +18,9 @@ func DiskSmi() ([]FileSystem, error) { ...@@ -17,6 +18,9 @@ func DiskSmi() ([]FileSystem, error) {
res := make([]FileSystem, len(parts)) res := make([]FileSystem, len(parts))
for _, p := range parts { for _, p := range parts {
if strings.Contains(p.Device, "overlay") || strings.Contains(p.Device, "/dev") {
usage, err := disk.Usage(p.Mountpoint) usage, err := disk.Usage(p.Mountpoint)
if err != nil { if err != nil {
...@@ -35,6 +39,7 @@ func DiskSmi() ([]FileSystem, error) { ...@@ -35,6 +39,7 @@ func DiskSmi() ([]FileSystem, error) {
res = append(res, newres) res = append(res, newres)
} }
}
return res, nil return res, nil
} }
gpuhw-smi
View file @ 3dc1c565
No preview for this file type
main-dcgm.go
View file @ 3dc1c565
...@@ -16,13 +16,7 @@ func DCGM() { ...@@ -16,13 +16,7 @@ func DCGM() {
promethAddr := flag.String("promethaddr", "http://127.0.0.1:9090", "a string parameter") promethAddr := flag.String("promethaddr", "http://127.0.0.1:9090", "a string parameter")
//http://47.94.59.74:9000/
//promethAddr := flag.String("promethaddr", "http://47.94.59.74:9000", "a string parameter")
flag.Parse() flag.Parse()
//cli, err := NewProCli("http://192.168.1.21:9090")
fmt.Println("connect to prometh addr-----------", *promethAddr) fmt.Println("connect to prometh addr-----------", *promethAddr)
//cli, err := NewProCli("http://192.168.1.21:9090") //cli, err := NewProCli("http://192.168.1.21:9090")
...@@ -117,185 +111,6 @@ func DCGM() { ...@@ -117,185 +111,6 @@ func DCGM() {
}) })
// app.Get("/hw/usage", func(c *fiber.Ctx) error {
// res := make([]DeviceInfo, 0, 10)
// // gpuUtils, err := cli.GpuUtil()
// // if err != nil {
// // return c.JSON(Response{
// // Success: false,
// // Error: err.Error(),
// // })
// // }
// // res = append(res, gpuUtils...)
// // gpuMemUtils, err := cli.GpuMemUtil()
// // if err != nil {
// // return c.JSON(Response{
// // Success: false,
// // Error: err.Error(),
// // })
// // }
// // res = append(res, gpuMemUtils...)
// diskUtil, err := cli.DiskUtil()
// if err != nil {
// return c.JSON(Response{
// Success: false,
// Error: err.Error(),
// })
// }
// res = append(res, diskUtil...)
// networkSpeed, err := cli.NetworkSpeed()
// if err != nil {
// return c.JSON(Response{
// Success: false,
// Error: err.Error(),
// })
// }
// res = append(res, networkSpeed...)
// cpuUtils, err := cli.CpuUtil()
// if err != nil {
// return c.JSON(Response{
// Success: false,
// Error: err.Error(),
// })
// }
// res = append(res, cpuUtils...)
// //MemUtil()
// memUtils, err := cli.MemUtil()
// if err != nil {
// return c.JSON(Response{
// Success: false,
// Error: err.Error(),
// })
// }
// res = append(res, memUtils...)
// netRece, err := cli.NetworkReceive()
// if err != nil {
// return c.JSON(Response{
// Success: false,
// Error: err.Error(),
// })
// }
// res = append(res, netRece...)
// netSend, err := cli.NetworkTransmit()
// if err != nil {
// return c.JSON(Response{
// Success: false,
// Error: err.Error(),
// })
// }
// res = append(res, netSend...)
// diskFreeSize, err := cli.DiskFreeSize()
// if err != nil {
// return c.JSON(Response{
// Success: false,
// Error: err.Error(),
// })
// }
// res = append(res, diskFreeSize...)
// // power, err := cli.GpuPowerUsage()
// // if err != nil {
// // return c.JSON(Response{
// // Success: false,
// // Error: err.Error(),
// // })
// // }
// // res = append(res, power...)
// memTemp, err := cli.GpuMemTemp()
// if err != nil {
// return c.JSON(Response{
// Success: false,
// Error: err.Error(),
// })
// }
// res = append(res, memTemp...)
// // gpuTemp, err := cli.GpuTemp()
// // if err != nil {
// // return c.JSON(Response{
// // Success: false,
// // Error: err.Error(),
// // })
// // }
// // res = append(res, gpuTemp...)
// return c.JSON(Response{
// Success: true,
// Devices: res,
// })
// })
// app.Get("/hw/info", func(c *fiber.Ctx) error {
// res := make([]DeviceInfo, 0, 10)
// diskTotalSize, err := cli.DiskTotalSize()
// if err != nil {
// return c.JSON(Response{
// Success: false,
// Error: err.Error(),
// })
// }
// res = append(res, diskTotalSize...)
// memSize, err := cli.MemInfo()
// if err != nil {
// return c.JSON(Response{
// Success: false,
// Error: err.Error(),
// })
// }
// res = append(res, memSize...)
// gpuInfo, err := cli.GpuInfo()
// if err != nil {
// return c.JSON(Response{
// Success: false,
// Error: err.Error(),
// })
// }
// res = append(res, gpuInfo...)
// return c.JSON(Response{
// Success: true,
// Devices: res,
// })
// })
// Start the server on port 3000
log.Fatal(app.Listen(":4000")) log.Fatal(app.Listen(":4000"))
} }
......
main.go
View file @ 3dc1c565
package main package main
func main() { func main() {
DCGM() //DCGM()
//Smi() Smi()
} }
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment