github/overleaf
1
0
Fork 0
mirror of https://github.com/overleaf/overleaf.git synced 2024-11-21 20:47:08 -05:00
Code Issues Projects Releases Packages Wiki Activity

add createMonitoredSubscriptionClient method

Browse source
This commit is contained in:
James Allen 2014-11-19 14:05:16 +00:00
parent 42cbaecbe0
commit 1ee287ed98
1 changed files with 31 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

31
libraries/redis-wrapper/index.coffee
View file

@ -20,6 +20,7 @@ module.exports = RedisSharelatex =
client = require("redis").createClient opts.port, opts.host, standardOpts
return client
# DO NOT USE. Not robust at all (ends up getting subscribed twice, etc)!
createRobustSubscriptionClient: (opts, heartbeatOpts = {}) ->
sub = RedisSharelatex.createClient(opts)
pub = RedisSharelatex.createClient(opts)
@ -65,6 +66,36 @@ module.exports = RedisSharelatex =
return sub
# Attaches an isAlive() method to the subsciption client which returns true
# so long as we have had a heartbeat recently.
createMonitoredSubscriptionClient: (opts, heartbeatOpts = {}) ->
sub = RedisSharelatex.createClient(opts)
pub = RedisSharelatex.createClient(opts)
heartbeatInterval = heartbeatOpts.heartbeat_interval or 2000 #ms
isAliveTimeout = heartbeatOpts.is_alive_timeout or 10000 #ms
id = require("crypto").pseudoRandomBytes(16).toString("hex")
heartbeatChannel = "heartbeat-#{id}"
lastHeartbeat = Date.now()
sub.subscribe heartbeatChannel, (error) ->
if error?
console.error "ERROR: failed to subscribe to #{heartbeatChannel} channel", error
sub.on "message", (channel, message) ->
if channel == heartbeatChannel
lastHeartbeat = Date.now()
sub.isAlive = () ->
timeSinceLastHeartbeat = Date.now() - lastHeartbeat
return (timeSinceLastHeartbeat < isAliveTimeout)
setInterval () ->
pub.publish heartbeatChannel, "PING"
, heartbeatInterval
return sub