Close sockets on server cmd

relay/cmd/main.go

@@ -2,20 +2,26 @@ package main
 
 import (
 	"os"
-
+	"os/signal"
-	"github.com/netbirdio/netbird/util"
+	"syscall"
 
 	log "github.com/sirupsen/logrus"
 
 	"github.com/netbirdio/netbird/relay/server"
+	"github.com/netbirdio/netbird/util"
 )
 
 func init() {
 	util.InitLog("trace", "console")
 }
 
-func main() {
+func waitForExitSignal() {
+	osSigs := make(chan os.Signal, 1)
+	signal.Notify(osSigs, syscall.SIGINT, syscall.SIGTERM)
+	_ = <-osSigs
+}
 
+func main() {
 	address := "10.145.236.1:1235"
 	srv := server.NewServer()
 	err := srv.Listen(address)
@@ -24,5 +30,11 @@ func main() {
 		os.Exit(1)
 	}
 
-	select {}
+	waitForExitSignal()
+
+	err = srv.Close()
+	if err != nil {
+		log.Errorf("failed to close server: %s", err)
+		os.Exit(1)
+	}
 }