set read/write at same time

isayme · May 15, 2024 · 5a56fee · 5a56fee
1 parent 39882e8
commit 5a56fee
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/util/net.go b/util/net.go
@@ -21,7 +21,7 @@ func NewTimeoutConn(conn net.Conn, timeout time.Duration) *timeoutConn {
 
 func (c *timeoutConn) Read(p []byte) (n int, err error) {
 	if c.timeout > 0 {
-		if err := c.Conn.SetReadDeadline(time.Now().Add(c.timeout)); err != nil {
+		if err := c.Conn.SetDeadline(time.Now().Add(c.timeout)); err != nil {
 			return 0, err
 		}
 	}
@@ -31,7 +31,7 @@ func (c *timeoutConn) Read(p []byte) (n int, err error) {
 
 func (c *timeoutConn) Write(p []byte) (n int, err error) {
 	if c.timeout > 0 {
-		if err := c.Conn.SetWriteDeadline(time.Now().Add(c.timeout)); err != nil {
+		if err := c.Conn.SetDeadline(time.Now().Add(c.timeout)); err != nil {
 			return 0, err
 		}
 	}