From 8f7aac3076947519e3dbf5651f5948fd013529e9 Mon Sep 17 00:00:00 2001 From: Eugen Rochko Date: Thu, 2 Feb 2017 15:20:31 +0100 Subject: [PATCH] Fix potential resource leaks in streaming server --- streaming/index.js | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) diff --git a/streaming/index.js b/streaming/index.js index 945e287f5e..99aa5b0402 100644 --- a/streaming/index.js +++ b/streaming/index.js @@ -45,7 +45,7 @@ const authenticationMiddleware = (req, res, next) => { return next(err) } - client.query('SELECT oauth_access_tokens.resource_owner_id, users.account_id FROM oauth_access_tokens INNER JOIN users ON oauth_access_tokens.resource_owner_id = users.id WHERE token = $1 LIMIT 1', [token], (err, result) => { + client.query('SELECT oauth_access_tokens.resource_owner_id, users.account_id FROM oauth_access_tokens INNER JOIN users ON oauth_access_tokens.resource_owner_id = users.id WHERE oauth_access_tokens.token = $1 LIMIT 1', [token], (err, result) => { done() if (err) { @@ -115,8 +115,13 @@ const streamFrom = (id, req, res, needsFiltering = false) => { } }) - // Heartbeat to keep connection alive - setInterval(() => res.write(':thump\n'), 15000) + const heartbeat = setInterval(() => res.write(':thump\n'), 15000) + + req.on('close', () => { + log.verbose(`Ending stream from ${id} for ${req.accountId}`) + clearInterval(heartbeat) + redisClient.quit() + }) redisClient.subscribe(id) }