Allow setting Read/Write/Idle timeouts

httputils/server.go

@@ -27,12 +27,14 @@ type Server struct {
 	Debug                 bool
 	Handler               RequestHandler
 
+	ReadTimeout  time.Duration
+	WriteTimeout time.Duration
+	IdleTimeout  time.Duration
+
 	extraHeadersMutex sync.RWMutex
 	extraHeaders      map[string]string
 }
 
-const ServeStreamChunked = -1
-
 type RequestContext interface {
 	GetPath() string
 	GetConnectionTime() time.Time
@@ -94,8 +96,9 @@ func (server *Server) Serve() {
 			}
 
 			s := &fasthttp.Server{
-				ReadTimeout: 5 * time.Second,
+				ReadTimeout:  server.ReadTimeout,
-				IdleTimeout: 15 * time.Second,
+				WriteTimeout: server.WriteTimeout,
+				IdleTimeout:  server.IdleTimeout,
 				Handler: func(ctx *fasthttp.RequestCtx) {
 					server.Handler(handler(ctx))
 				},
@@ -137,10 +140,11 @@ func (server *Server) Serve() {
 			}
 
 			s := &http.Server{
-				ReadTimeout: 5 * time.Second,
+				ReadTimeout:  server.ReadTimeout,
-				IdleTimeout: 15 * time.Second,
+				WriteTimeout: server.WriteTimeout,
-				Addr:        server.ListenAddress,
+				IdleTimeout:  server.IdleTimeout,
-				TLSConfig:   server.TLSConfig.Config,
+				Addr:         server.ListenAddress,
+				TLSConfig:    server.TLSConfig.Config,
 				Handler: http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
 					server.Handler(handler(w, r))
 				}),
@@ -175,8 +179,11 @@ func (server *Server) Serve() {
 
 			s := &http3.Server{
 				Server: &http.Server{
-					Addr:      server.ListenAddress,
+					ReadTimeout:  server.ReadTimeout,
-					TLSConfig: server.TLSConfig.QUICConfig,
+					WriteTimeout: server.WriteTimeout,
+					IdleTimeout:  server.IdleTimeout,
+					Addr:         server.ListenAddress,
+					TLSConfig:    server.TLSConfig.QUICConfig,
 					Handler: http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
 						if server.Debug {
 							log.Print("Received HTTP/3 request")