Added graceful shutdown

Server.go

@@ -1,48 +1,51 @@
 package server
 
 import (
+	"context"
+	"fmt"
 	"io"
 	"log"
+	"net"
 	"net/http"
+	"os"
+	"os/signal"
+	"syscall"
 
 	"git.akyoto.dev/go/router"
 )
 
 // Server represents a single web service.
 type Server struct {
-	router *router.Router[Handler]
+	Router *router.Router[Handler]
+	Config Configuration
 }
 
 // New creates a new server.
 func New() *Server {
 	return &Server{
-		router: router.New[Handler](),
+		Router: router.New[Handler](),
+		Config: defaultConfig(),
 	}
 }
 
 // Get registers your function to be called when the given GET path has been requested.
 func (server *Server) Get(path string, handler Handler) {
-	server.router.Add(http.MethodGet, path, handler)
+	server.Router.Add(http.MethodGet, path, handler)
 }
 
 // Post registers your function to be called when the given POST path has been requested.
 func (server *Server) Post(path string, handler Handler) {
-	server.router.Add(http.MethodPost, path, handler)
+	server.Router.Add(http.MethodPost, path, handler)
 }
 
 // Delete registers your function to be called when the given DELETE path has been requested.
 func (server *Server) Delete(path string, handler Handler) {
-	server.router.Add(http.MethodDelete, path, handler)
+	server.Router.Add(http.MethodDelete, path, handler)
 }
 
 // Put registers your function to be called when the given PUT path has been requested.
 func (server *Server) Put(path string, handler Handler) {
-	server.router.Add(http.MethodPut, path, handler)
-}
-
-// Router returns the router used by the server.
-func (server *Server) Router() *router.Router[Handler] {
-	return server.router
+	server.Router.Add(http.MethodPut, path, handler)
 }
 
 // ServeHTTP responds to the given request.
@@ -50,7 +53,7 @@ func (server *Server) ServeHTTP(response http.ResponseWriter, request *http.Requ
 	ctx := newContext(request, response)
 	defer contextPool.Put(ctx)
 
-	handler := server.router.LookupNoAlloc(request.Method, request.URL.Path, ctx.addParameter)
+	handler := server.Router.LookupNoAlloc(request.Method, request.URL.Path, ctx.addParameter)
 
 	if handler == nil {
 		response.WriteHeader(http.StatusNotFound)
@@ -65,3 +68,33 @@ func (server *Server) ServeHTTP(response http.ResponseWriter, request *http.Requ
 		log.Println(request.URL, err)
 	}
 }
+
+// Run starts the server on the given address.
+func (server *Server) Run(address string) {
+	srv := &http.Server{
+		Addr:              address,
+		Handler:           server,
+		ReadTimeout:       server.Config.Timeout.Read,
+		WriteTimeout:      server.Config.Timeout.Write,
+		IdleTimeout:       server.Config.Timeout.Idle,
+		ReadHeaderTimeout: server.Config.Timeout.ReadHeader,
+	}
+
+	listener, err := net.Listen("tcp", address)
+
+	if err != nil {
+		fmt.Println(err)
+		return
+	}
+
+	go srv.Serve(listener)
+
+	stop := make(chan os.Signal, 1)
+	signal.Notify(stop, os.Interrupt, syscall.SIGTERM)
+	<-stop
+
+	ctx, cancel := context.WithTimeout(context.Background(), server.Config.Timeout.Shutdown)
+	defer cancel()
+
+	srv.Shutdown(ctx)
+}