Merge pull request #289 from go-dummy/feat/graceful-shutdown

feat: add graceful shutdown

cmd/dummy/main.go

@@ -2,9 +2,14 @@ package main
 
 import (
 	"context"
+	"errors"
 	"flag"
 	"fmt"
+	"net/http"
 	"os"
+	"os/signal"
+	"syscall"
+	"time"
 
 	"github.com/cristalhq/acmd"
 
@@ -49,7 +54,24 @@ func run() error {
 				h := server.NewHandlers(api, l)
 				s := server.NewServer(cfg.Server, l, h)
 
-				return s.Run()
+				go func() {
+					if err := s.Run(); !errors.Is(err, http.ErrServerClosed) {
+						l.Logger.Err(err).Msg("run server")
+					}
+				}()
+
+				interrupt := make(chan os.Signal, 1)
+				signal.Notify(interrupt, os.Interrupt, syscall.SIGTERM)
+
+				x := <-interrupt
+				l.Info().Msgf("received `%v`", x)
+
+				const timeout = 5 * time.Second
+
+				ctx, cancelFunc := context.WithTimeout(ctx, timeout)
+				defer cancelFunc()
+
+				return s.Stop(ctx)
 			},
 		},
 	}

internal/api/build_test.go

@@ -2,10 +2,10 @@ package api_test
 
 import (
 	"fmt"
-	"github.com/go-dummy/openapi"
 	"strconv"
 	"testing"
 
+	"github.com/go-dummy/openapi"
 	"github.com/stretchr/testify/require"
 
 	"github.com/go-dummy/dummy/internal/api"

internal/server/server.go

@@ -1,6 +1,7 @@
 package server
 
 import (
+	"context"
 	"net/http"
 
 	"github.com/go-dummy/dummy/internal/config"
@@ -47,3 +48,7 @@ func (s *Server) Run() error {
 
 	return nil
 }
+
+func (s *Server) Stop(ctx context.Context) error {
+	return s.Server.Shutdown(ctx)
+}