ConnectedUsersManager.coffee 1.3 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849
  1. _ = require("underscore")
  2. async = require("async")
  3. Settings = require('settings-sharelatex')
  4. redis = require('redis')
  5. rclient = redis.createClient(Settings.redis.web.port, Settings.redis.web.host)
  6. rclient.auth(Settings.redis.web.password)
  7. ONE_HOUR_IN_S = 60 * 60
  8. buildUserKey = (project_id, user_id)-> return "connected_user:#{project_id}:#{user_id}"
  9. buildProjectSetKey = (project_id)-> return "connected_user:#{project_id}"
  10. module.exports =
  11. markUserAsConnected: (project_id, user_id, callback = (err)->)->
  12. async.series [
  13. (cb)->
  14. rclient.sadd buildProjectSetKey(project_id), user_id, cb
  15. (cb)->
  16. rclient.setex buildUserKey(project_id, user_id), new Date(), ONE_HOUR_IN_S * 6, cb
  17. ], callback
  18. _getConnectedUser: (project_id, user_id, callback)->
  19. rclient.get buildUserKey(project_id, user_id), (err, result)->
  20. if !result?
  21. connected = false
  22. else
  23. connected = true
  24. callback err, {connected:connected, user_id:user_id}
  25. getConnectedUsers: (project_id, callback)->
  26. self = @
  27. rclient.get "connected_users_list:#{project_id}", (err, results)->
  28. jobs = results.map (user_id)->
  29. (cb)->
  30. self._getConnectedUser(project_id, user_id, cb)
  31. async.series jobs, (err, users)->
  32. users = _.filter users, (user)->
  33. user.connected
  34. user_ids = _.map users, (user)->
  35. user.user_id
  36. callback err, user_ids