treehouse
/
mastodon
21
8
Fork 5
Code Issues 32 Pull Requests Projects 1 Releases Wiki Activity

Add heartbeat to websockets streaming API connections

lolsob-rspec
Eugen Rochko 2017-04-02 21:27:14 +02:00
parent 1ee4c6415c
commit 6966854f3d
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
streaming/index.js
View File

@ -215,8 +215,11 @@ const streamHttpEnd = req => (id, listener) => {
// Setup stream output to WebSockets // Setup stream output to WebSockets
const streamToWs = (req, ws) => { const streamToWs = (req, ws) => {
const heartbeat = setInterval(() => ws.ping(), 15000)
ws.on('close', () => { ws.on('close', () => {
log.verbose(req.requestId, `Ending stream for ${req.accountId}`) log.verbose(req.requestId, `Ending stream for ${req.accountId}`)
clearInterval(heartbeat)
}) })
return (event, payload) => { return (event, payload) => {
@ -234,6 +237,10 @@ const streamWsEnd = ws => (id, listener) => {
ws.on('close', () => { ws.on('close', () => {
unsubscribe(id, listener) unsubscribe(id, listener)
}) })
ws.on('error', e => {
unsubscribe(id, listener)
})
} }
app.use(setRequestId) app.use(setRequestId)