perf: reduce timer in write_control

Signed-off-by: rfyiamcool <rfyiamcool@163.com>
gorilla · Jan 23, 2024 · d15aba1 · d15aba1
1 parent cf50a3e
commit d15aba1
Showing 1 changed file with 9 additions and 4 deletions.
diff --git a/conn.go b/conn.go
@@ -454,13 +454,18 @@ func (c *Conn) WriteControl(messageType int, data []byte, deadline time.Time) er
 		}
 	}
 
-	timer := time.NewTimer(d)
 	select {
 	case <-c.mu:
-		timer.Stop()
-	case <-timer.C:
-		return errWriteTimeout
+	default:
+		timer := time.NewTimer(d)
+		select {
+		case <-c.mu:
+			timer.Stop()
+		case <-timer.C:
+			return errWriteTimeout
+		}
 	}
+
 	defer func() { c.mu <- struct{}{} }()
 
 	c.writeErrMu.Lock()