chore: watchgod, configuration update

Signed-off-by: Valery Piashchynski <[email protected]>

2 files changed, 24 insertions, 1 deletions

diff --git a/internal/cli/serve/command.go b/internal/cli/serve/command.go
index 21454933..a4caa265 100644
--- a/internal/cli/serve/command.go
+++ b/internal/cli/serve/command.go
@@ -114,9 +114,15 @@ func NewCommand(override *[]string, cfgFile *string, silent *bool) *cobra.Comman
 			if !*silent {
 				if notified {
 					fmt.Println("[INFO] sdnotify: notified")
+					stopCh := make(chan struct{}, 1)
+					if containerCfg.WatchdogSec > 0 {
+						fmt.Printf("[INFO] sdnotify: watchdog enabled, timeout: %d seconds\n", containerCfg.WatchdogSec)
+						sdnotify.StartWatchdog(containerCfg.WatchdogSec, stopCh)
+					}
+
 					// if notified -> notify about stop
 					defer func() {
-						_, _ = sdnotify.SdNotify(sdnotify.Stopping)
+						stopCh <- struct{}{}
 					}()
 				} else {
 					fmt.Println("[INFO] sdnotify: not notified")
diff --git a/internal/sdnotify/sdnotify.go b/internal/sdnotify/sdnotify.go
index aed38c92..d88c0e88 100644
--- a/internal/sdnotify/sdnotify.go
+++ b/internal/sdnotify/sdnotify.go
@@ -25,6 +25,7 @@ package sdnotify
 import (
 	"net"
 	"os"
+	"time"
 )
 
 type State string
@@ -84,3 +85,19 @@ func SdNotify(state State) (bool, error) {
 
 	return true, nil
 }
+
+func StartWatchdog(interval int, stopCh chan struct{}) {
+	go func() {
+		ticker := time.NewTicker(time.Duration(interval) * time.Second)
+		defer ticker.Stop()
+
+		for {
+			select {
+			case <-stopCh:
+				return
+			case <-ticker.C:
+				_, _ = SdNotify(Watchdog)
+			}
+		}
+	}()
+}