mirror of
https://github.com/overleaf/overleaf.git
synced 2024-11-07 20:31:06 -05:00
67 lines
2.1 KiB
CoffeeScript
67 lines
2.1 KiB
CoffeeScript
_ = require("underscore")
|
|
async = require("async")
|
|
Settings = require('settings-sharelatex')
|
|
logger = require("logger-sharelatex")
|
|
redis = require('redis')
|
|
rclient = redis.createClient(Settings.redis.web.port, Settings.redis.web.host)
|
|
rclient.auth(Settings.redis.web.password)
|
|
|
|
|
|
ONE_HOUR_IN_S = 60 * 60
|
|
ONE_DAY_IN_S = ONE_HOUR_IN_S * 24
|
|
FOUR_DAYS_IN_S = ONE_DAY_IN_S * 4
|
|
|
|
buildProjectSetKey = (project_id)-> return "users_in_project:#{project_id}"
|
|
buildUserKey = (project_id, user_id)-> return "connected_user:#{project_id}:#{user_id}"
|
|
|
|
|
|
module.exports =
|
|
|
|
markUserAsConnected: (project_id, user_id, callback = (err)->)->
|
|
logger.log project_id:project_id, user_id:user_id, "marking user as connected"
|
|
async.series [
|
|
(cb)->
|
|
rclient.sadd buildProjectSetKey(project_id), user_id, cb
|
|
(cb)->
|
|
rclient.expire buildProjectSetKey(project_id), FOUR_DAYS_IN_S, cb
|
|
(cb)->
|
|
rclient.setex buildUserKey(project_id, user_id), ONE_HOUR_IN_S, new Date(), cb
|
|
], (err)->
|
|
if err?
|
|
logger.err err:err, project_id:project_id, user_id:user_id, "problem marking user as connected"
|
|
callback(err)
|
|
|
|
markUserAsDisconnected: (project_id, user_id, callback)->
|
|
logger.log project_id:project_id, user_id:user_id, "marking user as disconnected"
|
|
async.series [
|
|
(cb)->
|
|
rclient.srem buildProjectSetKey(project_id), user_id, cb
|
|
(cb)->
|
|
rclient.expire buildProjectSetKey(project_id), FOUR_DAYS_IN_S, cb
|
|
(cb)->
|
|
rclient.del buildUserKey(project_id, user_id), cb
|
|
], callback
|
|
|
|
_getConnectedUser: (project_id, user_id, callback)->
|
|
rclient.get buildUserKey(project_id, user_id), (err, result)->
|
|
if !result?
|
|
connected = false
|
|
else
|
|
connected = true
|
|
|
|
callback err, {connected:connected, user_id:user_id}
|
|
|
|
getConnectedUsers: (project_id, callback)->
|
|
self = @
|
|
rclient.smembers buildProjectSetKey(project_id), (err, results)->
|
|
jobs = results.map (user_id)->
|
|
(cb)->
|
|
self._getConnectedUser(project_id, user_id, cb)
|
|
async.series jobs, (err, users)->
|
|
users = _.filter users, (user)->
|
|
user.connected
|
|
user_ids = _.map users, (user)->
|
|
user.user_id
|
|
callback err, user_ids
|
|
|
|
|