set read and write timeouts to 0 to disable

txn2 · Dec 18, 2020 · a5b5ed5 · a5b5ed5
1 parent 894b1d1
commit a5b5ed5
Showing 1 changed file with 10 additions and 4 deletions.
diff --git a/cmd/jwtpxy.go b/cmd/jwtpxy.go
@@ -210,10 +210,16 @@ func main() {
 
 	// Proxy Server
 	pxySrv := &http.Server{
-		Addr:         *ip + ":" + *port,
-		Handler:      pMux,
-		ReadTimeout:  time.Duration(*readTimeout) * time.Second,
-		WriteTimeout: time.Duration(*writeTimeout) * time.Second,
+		Addr:    *ip + ":" + *port,
+		Handler: pMux,
+	}
+
+	if *readTimeout > 0 {
+		pxySrv.ReadTimeout = time.Duration(*readTimeout) * time.Second
+	}
+
+	if *writeTimeout > 0 {
+		pxySrv.WriteTimeout = time.Duration(*writeTimeout) * time.Second
 	}
 
 	// util mux