feat(http): implement graceful shutdown (#79)

cmd/seasonpackarr/main.go

@@ -4,6 +4,7 @@
 package main
 
 import (
+	"context"
 	"encoding/json"
 	"fmt"
 	netHTTP "net/http"
@@ -125,16 +126,31 @@ func main() {
 
 		errorChannel := make(chan error)
 		go func() {
-			errorChannel <- srv.Open()
+			err := srv.Open()
+			if err != nil {
+				if !errors.Is(err, http.ErrServerClosed) {
+					errorChannel <- err
+				}
+			}
 		}()
 
 		sigCh := make(chan os.Signal, 1)
 		signal.Notify(sigCh, syscall.SIGHUP, syscall.SIGINT, syscall.SIGQUIT, syscall.SIGTERM)
 
-		for sig := range sigCh {
-			log.Info().Msgf("received signal: %v, shutting down server.", sig)
+		select {
+		case sig := <-sigCh:
+			log.Info().Msgf("received signal: %q, shutting down server.", sig.String())
 			os.Exit(0)
+
+		case err := <-errorChannel:
+			log.Error().Err(err).Msg("unexpected error from server")
 		}
+		if err := srv.Shutdown(context.Background()); err != nil {
+			log.Error().Err(err).Msg("error during http shutdown")
+			os.Exit(1)
+		}
+
+		os.Exit(0)
 
 	default:
 		pflag.Usage()

internal/http/middleware.go

@@ -6,7 +6,7 @@ package http
 
 import "net/http"
 
-func (s Server) isAuthenticated(next http.Handler) http.Handler {
+func (s *Server) isAuthenticated(next http.Handler) http.Handler {
 	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
 		// allow access if apiToken value is set to an empty string
 		if s.cfg.Config.APIToken == "" {

internal/http/server.go

@@ -5,6 +5,7 @@
 package http
 
 import (
+	"context"
 	"fmt"
 	"net"
 	"net/http"
@@ -17,9 +18,13 @@ import (
 	"github.com/go-chi/chi/v5/middleware"
 )
 
+var ErrServerClosed = http.ErrServerClosed
+
 type Server struct {
 	log logger.Logger
 	cfg *config.AppConfig
+
+	httpServer http.Server
 }
 
 func NewServer(log logger.Logger, config *config.AppConfig) *Server {
@@ -29,7 +34,7 @@ func NewServer(log logger.Logger, config *config.AppConfig) *Server {
 	}
 }
 
-func (s Server) Open() error {
+func (s *Server) Open() error {
 	addr := fmt.Sprintf("%v:%v", s.cfg.Config.Host, s.cfg.Config.Port)
 
 	var err error
@@ -44,23 +49,28 @@ func (s Server) Open() error {
 	return err
 }
 
-func (s Server) tryToServe(addr, proto string) error {
+func (s *Server) tryToServe(addr, proto string) error {
 	listener, err := net.Listen(proto, addr)
 	if err != nil {
 		return err
 	}
 
 	s.log.Info().Msgf("Starting %s server. Listening on %s", proto, listener.Addr().String())
 
-	server := http.Server{
+	s.httpServer = http.Server{
 		Handler:           s.Handler(),
 		ReadHeaderTimeout: time.Second * 15,
 	}
 
-	return server.Serve(listener)
+	return s.httpServer.Serve(listener)
+}
+
+func (s *Server) Shutdown(ctx context.Context) error {
+	s.log.Info().Msgf("shutting down http server gracefully...")
+	return s.httpServer.Shutdown(ctx)
 }
 
-func (s Server) Handler() http.Handler {
+func (s *Server) Handler() http.Handler {
 	r := chi.NewRouter()
 
 	r.Use(middleware.RequestID)