From e6447d6dcf3f5e02cd707be13adf2ece08e1cde2 Mon Sep 17 00:00:00 2001
From: Aofei Sheng <aofei@aofeisheng.com>
Date: Sun, 22 Nov 2020 19:19:02 +0800
Subject: [PATCH] Only render mem info when its total > 0

---
 widgets/mem.go | 12 ++++++++----
 1 file changed, 8 insertions(+), 4 deletions(-)

diff --git a/widgets/mem.go b/widgets/mem.go
index b994276..b6379a9 100644
--- a/widgets/mem.go
+++ b/widgets/mem.go
@@ -26,8 +26,10 @@ func NewMemWidget(updateInterval time.Duration, horizontalScale int) *MemWidget
 	mems := make(map[string]devices.MemoryInfo)
 	devices.UpdateMem(mems)
 	for name, mem := range mems {
-		widg.Data[name] = []float64{0}
-		widg.renderMemInfo(name, mem)
+		if mem.Total > 0 {
+			widg.Data[name] = []float64{0}
+			widg.renderMemInfo(name, mem)
+		}
 	}
 
 	go func() {
@@ -35,7 +37,9 @@ func NewMemWidget(updateInterval time.Duration, horizontalScale int) *MemWidget
 			widg.Lock()
 			devices.UpdateMem(mems)
 			for label, mi := range mems {
-				widg.renderMemInfo(label, mi)
+				if mi.Total > 0 {
+					widg.renderMemInfo(label, mi)
+				}
 			}
 			widg.Unlock()
 		}
@@ -47,7 +51,7 @@ func NewMemWidget(updateInterval time.Duration, horizontalScale int) *MemWidget
 func (mem *MemWidget) EnableMetric() {
 	mems := make(map[string]devices.MemoryInfo)
 	devices.UpdateMem(mems)
-	for l, _ := range mems {
+	for l := range mems {
 		lc := l
 		metrics.NewGauge(makeName("memory", l), func() float64 {
 			if ds, ok := mem.Data[lc]; ok {