Set conn deadline to both reads and writes (antoniomika#281)

widhaprasa · Oct 9, 2023 · b89a463 · b89a463
1 parent 56816e6
commit b89a463
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/utils/conn.go b/utils/conn.go
@@ -184,7 +184,7 @@ type IdleTimeoutConn struct {
 
 // Read is needed to implement the reader part.
 func (i IdleTimeoutConn) Read(buf []byte) (int, error) {
-	err := i.Conn.SetReadDeadline(time.Now().Add(viper.GetDuration("idle-connection-timeout")))
+	err := i.Conn.SetDeadline(time.Now().Add(viper.GetDuration("idle-connection-timeout")))
 	if err != nil {
 		return 0, err
 	}
@@ -194,7 +194,7 @@ func (i IdleTimeoutConn) Read(buf []byte) (int, error) {
 
 // Write is needed to implement the writer part.
 func (i IdleTimeoutConn) Write(buf []byte) (int, error) {
-	err := i.Conn.SetWriteDeadline(time.Now().Add(viper.GetDuration("idle-connection-timeout")))
+	err := i.Conn.SetDeadline(time.Now().Add(viper.GetDuration("idle-connection-timeout")))
 	if err != nil {
 		return 0, err
 	}