add stat memory

johnstcn · johnstcn · commit 18f4942b1f34 · 2023-06-13T17:33:29.000+01:00
diff --git a/cli/stat.go b/cli/stat.go
@@ -50,6 +50,11 @@ func (*RootCmd) stat() *clibase.Cmd {
 			} else {
 				sr.HostCPU = cs
 			}
+			if ms, err := statMem(hi); err != nil {
+				return err
+			} else {
+				sr.HostMemory = ms
+			}
 			out, err := formatter.Format(inv.Context(), []statsRow{sr})
 			if err != nil {
 				return err
@@ -85,6 +90,19 @@ func statCPU(hi sysinfotypes.Host, interval time.Duration) (*stat, error) {
 	return s, nil
 }
 
+func statMem(hi sysinfotypes.Host) (*stat, error) {
+	s := &stat{
+		Unit: "GB",
+	}
+	hm, err := hi.Memory()
+	if err != nil {
+		return nil, err
+	}
+	s.Total = float64(hm.Total) / 1024 / 1024 / 1024
+	s.Used = float64(hm.Used) / 1024 / 1024 / 1024
+	return s, nil
+}
+
 type statsRow struct {
 	HostCPU         *stat `json:"host_cpu" table:"host_cpu,default_sort"`
 	HostMemory      *stat `json:"host_memory" table:"host_memory"`
