Merge pull request #185 from jeangovil/fix-memory-leak

Remove schedulers and widgets from memory after any live-reloading

wtf.go

@@ -121,6 +121,8 @@ func watchForConfigChanges(app *tview.Application, configFlag string, grid *tvie
 			select {
 			case <-watch.Event:
 				loadConfig(configFlag)
+				// Disable all widgets to stop scheduler goroutines and rmeove widgets from memory.
+				disableAllWidgets()
 				makeWidgets(app, pages)
 				grid = buildGrid(Widgets)
 				pages.AddPage("grid", grid, true, true)
@@ -155,6 +157,12 @@ var (
 	version = "dev"
 )
 
+func disableAllWidgets() {
+	for _, widget := range Widgets {
+		widget.Disable()
+	}
+}
+
 func addWidget(app *tview.Application, pages *tview.Pages, widgetName string) {
 	// Always in alphabetical order
 	switch widgetName {

wtf/enabler.go

@@ -3,4 +3,5 @@ package wtf
 type Enabler interface {
 	Disabled() bool
 	Enabled() bool
+	Disable()
 }

wtf/text_widget.go

@@ -61,6 +61,10 @@ func (widget *TextWidget) Enabled() bool {
 	return widget.enabled
 }
 
+func (widget *TextWidget) Disable() {
+	widget.enabled = false
+}
+
 func (widget *TextWidget) Focusable() bool {
 	return widget.enabled && widget.focusable
 }