adding timeout to Allocation API service (#106)

PlayFab · Dec 13, 2021 · b6f58fb · b6f58fb
1 parent 0764a0d
commit b6f58fb
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/operator/http/http.go b/operator/http/http.go
@@ -7,6 +7,7 @@ import (
 	"net"
 	"net/http"
 	"os"
+	"time"
 
 	"k8s.io/apimachinery/pkg/runtime"
 	"k8s.io/client-go/rest"
@@ -98,8 +99,11 @@ func (s *ApiServer) Start(ctx context.Context) error {
 	log.Info("serving allocation API service", "addr", addr, "port", listeningPort)
 
 	srv := &http.Server{
-		Addr:    addr,
-		Handler: mux,
+		Addr:           addr,
+		Handler:        mux,
+		ReadTimeout:    5 * time.Second,
+		WriteTimeout:   5 * time.Second,
+		MaxHeaderBytes: http.DefaultMaxHeaderBytes,
 	}
 
 	done := make(chan struct{})