enhance: service: add http timeouts

grokify · Apr 17, 2023 · 7df27ca · 7df27ca
1 parent 607f0e2
commit 7df27ca
Showing 1 changed file with 15 additions and 1 deletion.
diff --git a/pkg/service/service.go b/pkg/service/service.go
@@ -7,6 +7,7 @@ import (
 	"net/http"
 	"strconv"
 	"strings"
+	"time"
 
 	"github.com/apex/gateway"
 	"github.com/aws/aws-lambda-go/events"
@@ -303,7 +304,20 @@ func ServeNetHTTP(svc Service) {
 	log.Info().
 		Int("port", svc.Config.Port).
 		Msg("STARTING_NET_HTTP")
-	clog.Fatal(http.ListenAndServe(portAddress(svc.Config.Port), getHTTPServeMux(svc)))
+
+	srv := &http.Server{
+		Addr:              portAddress(svc.Config.Port),
+		ReadTimeout:       3 * time.Minute,
+		WriteTimeout:      3 * time.Minute,
+		IdleTimeout:       30 * time.Second,
+		ReadHeaderTimeout: 2 * time.Second,
+		Handler:           getHTTPServeMux(svc),
+		// TLSConfig:         tlsConfig,
+	}
+
+	clog.Fatal(srv.ListenAndServe())
+
+	// clog.Fatal(http.ListenAndServe(portAddress(svc.Config.Port), getHTTPServeMux(svc)))
 }
 
 func ServeFastHTTP(svc Service) {