github/bettercap
1
0
Fork 0
mirror of https://github.com/bettercap/bettercap synced 2025-08-19 13:09:49 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix: logging error when read from websocket fails

Browse source
This commit is contained in:
Simone Margaritelli 2020-04-08 10:36:04 +02:00
commit 61d9316cad
1 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
modules/api_rest/api_rest_ws.go
View file

@ -95,7 +95,7 @@ func (mod *RestAPI) streamReader(ws *websocket.Conn) {
for { for {
_, _, err := ws.ReadMessage() _, _, err := ws.ReadMessage()
if err != nil { if err != nil {
mod.Debug("Closing websocket reader.") mod.Warning("error reading message from websocket: %v", err)
break break
} }
} }
@ -105,12 +105,12 @@ func (mod *RestAPI) startStreamingEvents(w http.ResponseWriter, r *http.Request)
ws, err := mod.upgrader.Upgrade(w, r, nil) ws, err := mod.upgrader.Upgrade(w, r, nil)
if err != nil { if err != nil {
if _, ok := err.(websocket.HandshakeError); !ok { if _, ok := err.(websocket.HandshakeError); !ok {
mod.Error("Error while updating api.rest connection to websocket: %s", err) mod.Error("error while updating api.rest connection to websocket: %s", err)
} }
return return
} }
mod.Debug("Websocket streaming started for %s", r.RemoteAddr) mod.Debug("websocket streaming started for %s", r.RemoteAddr)
go mod.streamWriter(ws, w, r) go mod.streamWriter(ws, w, r)
mod.streamReader(ws) mod.streamReader(ws)