Force exit on DYNDNS server failure

main.go

@@ -1,6 +1,7 @@
 package main
 
 import (
+	"context"
 	"github.com/cromefire/fritzbox-cloudflare-dyndns/pkg/avm"
 	"github.com/cromefire/fritzbox-cloudflare-dyndns/pkg/cloudflare"
 	"github.com/cromefire/fritzbox-cloudflare-dyndns/pkg/dyndns"
@@ -24,6 +25,8 @@ func main() {
 	updater := newUpdater()
 	updater.StartWorker()
 
+	ctx, cancel := context.WithCancel(context.Background())
+
 	ipv6LocalAddress := os.Getenv("DEVICE_LOCAL_ADDRESS_IPV6")
 
 	var localIp net.IP
@@ -37,14 +40,21 @@ func main() {
 	}
 
 	startPollServer(updater.In, &localIp)
-	startPushServer(updater.In, &localIp)
+	startPushServer(updater.In, &localIp, cancel)
 
+	// Create a OS signal shutdown channel
 	shutdown := make(chan os.Signal)
 
 	signal.Notify(shutdown, syscall.SIGTERM)
 	signal.Notify(shutdown, syscall.SIGINT)
 
-	<-shutdown
+	// Wait for either the context to finish or the shutdown signal
+	select {
+	case <-ctx.Done():
+		break
+	case <-shutdown:
+		break
+	}
 
 	slog.Info("Shutdown detected")
 }
@@ -133,7 +143,7 @@ func newUpdater() *cloudflare.Updater {
 	return u
 }
 
-func startPushServer(out chan<- *net.IP, localIp *net.IP) {
+func startPushServer(out chan<- *net.IP, localIp *net.IP, cancel context.CancelFunc) {
 	bind := os.Getenv("DYNDNS_SERVER_BIND")
 
 	if bind == "" {
@@ -155,6 +165,7 @@ func startPushServer(out chan<- *net.IP, localIp *net.IP) {
 	go func() {
 		err := s.ListenAndServe()
 		slog.Error("Server stopped", logging.ErrorAttr(err))
+		cancel()
 	}()
 }