perf: reduce timer in write_control

Signed-off-by: rfyiamcool <[email protected]>
gorilla · Dec 8, 2023 · 5346582 · 5346582
1 parent 01b0aae
commit 5346582
Showing 1 changed file with 9 additions and 4 deletions.
diff --git a/conn.go b/conn.go
@@ -455,13 +455,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()