add new ticker task convenience function

pkg/gui/containers_panel.go

@@ -123,23 +123,15 @@ func (gui *Gui) renderContainerStats(mainView *gocui.View, container *commands.C
 	mainView.Autoscroll = false
 	mainView.Title = "Stats"
 
-	return gui.T.NewTask(func(stop chan struct{}) {
+	return gui.T.NewTickerTask(time.Second, func() {
-		tickChan := time.NewTicker(time.Second)
+		width, _ := mainView.Size()
-		for {
-			select {
-			case <-stop:
-				return
-			case <-tickChan.C:
-				width, _ := mainView.Size()
 
-				contents, err := container.RenderStats(width)
+		contents, err := container.RenderStats(width)
-				if err != nil {
+		if err != nil {
-					gui.createErrorPanel(gui.g, err.Error())
+			gui.createErrorPanel(gui.g, err.Error())
-				}
-
-				gui.reRenderString(gui.g, "main", contents)
-			}
 		}
+
+		gui.reRenderString(gui.g, "main", contents)
 	})
 }
 

pkg/tasks/tasks.go

@@ -2,6 +2,7 @@ package tasks
 
 import (
 	"sync"
+	"time"
 
 	"github.com/sirupsen/logrus"
 )
@@ -53,3 +54,19 @@ func (t *Task) Stop() {
 	<-t.notifyStopped
 	return
 }
+
+// NewTickerTask is a convenience function for making a new task that repeats some action once per e.g. second
+func (t *TaskManager) NewTickerTask(duration time.Duration, f func()) error {
+	return t.NewTask(func(stop chan struct{}) {
+		tickChan := time.NewTicker(time.Second)
+		f() // calling f first so that we're not waiting for the first tick
+		for {
+			select {
+			case <-stop:
+				return
+			case <-tickChan.C:
+				f()
+			}
+		}
+	})
+}