From 51b6f7a9ce719bf0e748f1626b5115666a782df8 Mon Sep 17 00:00:00 2001 From: Alex Date: Tue, 15 Mar 2022 19:29:20 +0100 Subject: [PATCH] better error handling for SendBinaryMessage --- modules/structs/WebClient.go | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/modules/structs/WebClient.go b/modules/structs/WebClient.go index e5effc4..3df3773 100644 --- a/modules/structs/WebClient.go +++ b/modules/structs/WebClient.go @@ -45,6 +45,12 @@ func (webClient *WebClient) SendBinaryMessage(conn *websocket.Conn, msg []byte) if conn == webClient.MobileConn { webClient.mobileConnMu.Lock() defer webClient.mobileConnMu.Unlock() + + if conn == nil { + logger.WebMobile.Warnln("err conn nil SendBinaryMessage") + return errors.New("conn nil") + } + err = conn.WriteMessage(websocket.BinaryMessage, msg) if err != nil { @@ -56,6 +62,11 @@ func (webClient *WebClient) SendBinaryMessage(conn *websocket.Conn, msg []byte) webClient.voiceConnMu.Lock() defer webClient.voiceConnMu.Unlock() + if conn == nil { + logger.WebVoice.Warnln("err conn nil SendBinaryMessage") + return errors.New("conn nil") + } + err = conn.WriteMessage(websocket.BinaryMessage, msg) if err != nil {