only smi

docker-compose-smi.yml

@@ -2,7 +2,7 @@ version: "3.9"
 
 services:
   monitorSmi:
-    image: docker.ai.17xuexi.com/monitorapi/monitor-smi:v.0.0.5
+    image: docker.ai.17xuexi.com/monitorapi/monitor-smi:v.0.0.7
     network_mode: host
     volumes:
       - /proc:/host/proc:ro

main-smi.go

@@ -40,16 +40,21 @@ func Smi() {
 			fmt.Println("gpu error", err.Error())
 			res.Msg = err.Error()
 			res.Code = 500
+			return c.JSON(&res)
 		}
 
 		res.Data.Gpus = gpus
 
+		res.Code = 200
+		return c.JSON(&res)
+
 		cpus, err := CpuSmi()
 
 		if err != nil {
 			fmt.Println("cpu error", err.Error())
 			res.Msg = err.Error()
 			res.Code = 500
+			return c.JSON(&res)
 		}
 
 		res.Data.Cpus = cpus
@@ -60,6 +65,7 @@ func Smi() {
 			fmt.Println("network error", err.Error())
 			res.Msg = err.Error()
 			res.Code = 500
+			return c.JSON(&res)
 		}
 
 		res.Data.Networks = networks
@@ -70,6 +76,7 @@ func Smi() {
 			fmt.Println("disks error", err.Error())
 			res.Msg = err.Error()
 			res.Code = 500
+			return c.JSON(&res)
 		}
 
 		res.Data.Disk = filesystems
@@ -80,6 +87,7 @@ func Smi() {
 			fmt.Println("mem error", err.Error())
 			res.Msg = err.Error()
 			res.Code = 500
+			return c.JSON(&res)
 		}
 
 		res.Data.Mem = mem

main.go

 package main
 
 func main() {
-	DCGM()
-	//Smi()
+	//DCGM()
+	Smi()
 }