WIP

Signed-off-by: Mike Beaumont <[email protected]>
kumahq · Jul 11, 2023 · 76662ce · 76662ce
1 parent 044d838
commit 76662ce
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/pkg/util/watchdog/watchdog.go b/pkg/util/watchdog/watchdog.go
@@ -27,16 +27,16 @@ func (w *SimpleWatchdog) Start(stop <-chan struct{}) {
 		go func() {
 			<-stop
 			cancel()
-			if w.OnStop != nil {
-				w.OnStop()
-			}
 		}()
 		select {
 		case <-ticker.C:
 			if err := w.onTick(ctx); err != nil {
 				w.OnError(err)
 			}
 		case <-stop:
+			if w.OnStop != nil {
+				w.OnStop()
+			}
 			return
 		}
 	}