qugalet/gotosocial
1
0
Fork 0
mirror of https://github.com/superseriousbusiness/gotosocial.git synced 2024-11-22 11:46:40 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

[chore] Fix loop issue in streaming 🤦 (#3457)

Browse source
This commit is contained in:
tobi 2024-10-18 16:57:50 +02:00 committed by GitHub
parent 602c858379
commit 0d0314b98d
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 16 additions and 13 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

29
internal/api/client/streaming/stream.go
View file

@ -399,7 +399,7 @@ func (m *Module) writeToWSConn(
// - receipt of msg // - receipt of msg
// - timeout of pingCtx // - timeout of pingCtx
// - stream closed. // - stream closed.
msg, gotMsg := stream.Recv(pingCtx) msg, haveMsg := stream.Recv(pingCtx)
// If ping context has timed // If ping context has timed
// out, we should send a ping. // out, we should send a ping.
@ -410,35 +410,38 @@ func (m *Module) writeToWSConn(
cncl() cncl()
switch { switch {
case !haveMsg && !shouldPing:
// We have no message and we shouldn't
// send a ping; this means the stream
// has been closed from the client's end,
// so there's nothing further to do here.
l.Trace("no message and we shouldn't ping, returning...")
return
// We have a message to stream. case haveMsg:
case gotMsg: // We have a message to stream.
l.Tracef("writing websocket message: %+v", msg) l.Tracef("writing websocket message: %+v", msg)
if err := wsConn.WriteJSON(msg); err != nil { if err := wsConn.WriteJSON(msg); err != nil {
// If there's an error writing then drop the // If there's an error writing then drop the
// connection, as client may have disappeared // connection, as client may have disappeared
// suddenly; they can reconnect if necessary. // suddenly; they can reconnect if necessary.
l.Debugf("error writing websocket message: %v", err) l.Debugf("error writing websocket message: %v", err)
break return
} }
// We have no message but we
// need to send a keep-alive ping.
case shouldPing: case shouldPing:
// We have no message but we do
// need to send a keep-alive ping.
l.Trace("writing websocket ping") l.Trace("writing websocket ping")
if err := wsConn.WriteControl(websocket.PingMessage, pingMsg, time.Time{}); err != nil { if err := wsConn.WriteControl(websocket.PingMessage, pingMsg, time.Time{}); err != nil {
// If there's an error writing then drop the // If there's an error writing then drop the
// connection, as client may have disappeared // connection, as client may have disappeared
// suddenly; they can reconnect if necessary. // suddenly; they can reconnect if necessary.
l.Debugf("error writing websocket ping: %v", err) l.Debugf("error writing websocket ping: %v", err)
break return
} }
// We have no message and we shouldn't
// send a ping; this means the stream
// has been closed from the client's end.
default:
return
} }
} }
} }