add SetReadHeaderTimeout

pires · Oct 8, 2024 · 21542e2 · 21542e2
1 parent b718e7c
commit 21542e2
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/protocol.go b/protocol.go
@@ -63,6 +63,15 @@ func ValidateHeader(v Validator) func(*Conn) {
 	}
 }
 
+// SetReadHeaderTimeout sets the readHeaderTimeout for a connection when passed as option to NewConn()
+func SetReadHeaderTimeout(t time.Duration) func(*Conn) {
+	return func(c *Conn) {
+		if t >= 0 {
+			c.readHeaderTimeout = t
+		}
+	}
+}
+
 // Accept waits for and returns the next connection to the listener.
 func (p *Listener) Accept() (net.Conn, error) {
 	// Get the underlying connection