From dc3c7d2eeaa9743400d156758d69f2bab87858a8 Mon Sep 17 00:00:00 2001 From: kercylan98 Date: Tue, 12 Sep 2023 18:41:35 +0800 Subject: [PATCH] =?UTF-8?q?perf:=20server=20=E5=8C=85=20websocket=20SetRea?= =?UTF-8?q?dDeadline=20=E4=BC=98=E5=8C=96?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- server/server.go | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/server/server.go b/server/server.go index 639cbbd..7638ffe 100644 --- a/server/server.go +++ b/server/server.go @@ -310,9 +310,15 @@ func (slf *Server) Run(addr string) error { conn.Close(e) } }() + var deadline = times.Zero + if slf.websocketReadDeadline > 0 { + deadline = time.Now().Add(slf.websocketReadDeadline) + } for !conn.IsClosed() { - if err := ws.SetReadDeadline(super.If(slf.websocketReadDeadline <= 0, times.Zero, time.Now().Add(slf.websocketReadDeadline))); err != nil { - panic(err) + if slf.websocketReadDeadline > 0 { + if err := ws.SetReadDeadline(deadline); err != nil { + panic(err) + } } messageType, packet, readErr := ws.ReadMessage() if readErr != nil {