feat(relay): gracefully shutdown

cmd/commands/help.go

@@ -1,5 +1,21 @@
 package commands
 
+import "fmt"
+
+const help = `Available commands:
+Commands:
+  run <path_to_config>      starts the application with the specified configuration file.
+  help                      displays this help information.
+  version                   displays the version of software.
+
+Usage:
+  <command> [options]
+
+Examples:
+  run config.yaml        run the application using config.yaml.
+  help                   display this help message.
+`
+
 func HandleHelp(_ []string) {
-	// TODO:::
+	fmt.Print(help) //nolint
 }

cmd/commands/run.go

@@ -2,6 +2,7 @@ package commands
 
 import (
 	"errors"
+	"fmt"
 	"os"
 	"os/signal"
 	"syscall"
@@ -25,15 +26,27 @@ func HandleRun(args []string) {
 		ExitOnError(err)
 	}
 
-	if err := r.Start(); err != nil {
-		ExitOnError(err)
-	}
-
+	errCh := make(chan error)
 	sigChan := make(chan os.Signal, 1)
 	signal.Notify(sigChan, syscall.SIGINT, syscall.SIGTERM, os.Interrupt)
-	<-sigChan
 
-	if err := r.Stop(); err != nil {
-		ExitOnError(err)
+	go func() {
+		if err := r.Start(); err != nil {
+			errCh <- err
+		}
+	}()
+
+	select {
+	case sig := <-sigChan:
+		fmt.Printf("Received signal: %s\nInitiating graceful shutdown...\n", sig.String()) //nolint
+		if err := r.Stop(); err != nil {
+			ExitOnError(err)
+		}
+
+	case err := <-errCh:
+		fmt.Printf("Unexpected error: %v\nInitiating shutdown due to the error...\n", err) //nolint
+		if err := r.Stop(); err != nil {
+			ExitOnError(err)
+		}
 	}
 }

cmd/main.go

@@ -11,7 +11,8 @@ import (
 
 func main() {
 	if len(os.Args) < 2 {
-		commands.ExitOnError(errors.New("at least 1 arguments expected.\nuse help command for more information"))
+		commands.HandleHelp(os.Args)
+		commands.ExitOnError(errors.New("at least 1 arguments expected"))
 	}
 
 	switch os.Args[1] {
@@ -20,6 +21,7 @@ func main() {
 
 	case "help":
 		commands.HandleHelp(os.Args)
+		os.Exit(0)
 
 	case "version":
 		fmt.Println(immortal.StringVersion()) //nolint

relay/relay.go

@@ -22,7 +22,7 @@ func New(cfg config.Config) (*Relay, error) {
 
 	return &Relay{
 		config:   cfg,
-		server:   server.NewServer(cfg.ServerConf),
+		server:   server.New(cfg.ServerConf),
 		database: db,
 	}, nil
 }

server/server.go

@@ -23,7 +23,7 @@ type Server struct {
 	connsLock sync.RWMutex
 }
 
-func NewServer(cfg Config) *Server {
+func New(cfg Config) *Server {
 	return &Server{
 		config:    cfg,
 		conns:     make(map[*websocket.Conn]map[string]filter.Filters),
@@ -77,7 +77,7 @@ func (s *Server) readLoop(ws *websocket.Conn) {
 			go s.handleEvent(ws, msg)
 
 		case "CLOSE":
-			s.handleClose(ws, msg)
+			go s.handleClose(ws, msg)
 		}
 	}
 }