fix/header-timeout: adding ReadTimeout values

smartcontractkit · May 30, 2023 · 51d433a · 51d433a
1 parent 29bb12f
commit 51d433a
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/server.go b/server.go
@@ -10,6 +10,7 @@ import (
 	"net"
 	"net/http"
 	"sync"
+	"time"
 
 	"github.com/google/uuid"
 	"github.com/gorilla/websocket"
@@ -89,7 +90,8 @@ func (s *Server) Serve(lis net.Listener) {
 		})
 
 		hcsrv := &http.Server{
-			Handler: hchandler,
+			Handler:     hchandler,
+			ReadTimeout: 5 * time.Second,
 		}
 
 		//nolint:errcheck
@@ -101,8 +103,9 @@ func (s *Server) Serve(lis net.Listener) {
 	wshandler := http.NewServeMux()
 	wshandler.HandleFunc("/", s.wshandler)
 	wssrv := &http.Server{
-		TLSConfig: s.opts.creds.Config,
-		Handler:   wshandler,
+		TLSConfig:   s.opts.creds.Config,
+		Handler:     wshandler,
+		ReadTimeout: 10 * time.Second,
 	}
 
 	//nolint:errcheck