[Status] In case of panicking, a special cleanup should take place

laurybueno · Nov 12, 2020 · fa3282b · fa3282b
1 parent e0ae759
commit fa3282b
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/cmd/top.go b/cmd/top.go
@@ -19,6 +19,7 @@ import (
 	"context"
 	"fmt"
 	"log"
+	"runtime/debug"
 	"sort"
 	"time"
 
@@ -87,8 +88,15 @@ func runTop(cmd *cobra.Command, args []string) {
 	}
 }
 
+func panicCleanUp() {	
+	if r := recover(); r != nil {
+		ui.Close()
+		fmt.Printf("Something went wrong. Here's what we know: %#v", string(debug.Stack()))
+  }
+}
+
 func prepareDataTable() {
-	defer ui.Close()
+	defer panicCleanUp()
 
 	for {
 		pods := getPodsData()
@@ -152,7 +160,7 @@ func updateInterface() {
 }
 
 func updateGauge(current, total int) {
-	defer ui.Close()
+	defer panicCleanUp()
 
 	if gauge == nil {
 		gauge = widgets.NewGauge()