ConnectedUsersManager.coffee 1.6 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758
  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. buildProjectSetKey = (project_id)-> return "users_in_project:#{project_id}"
  9. buildUserKey = (project_id, user_id)-> return "connected_user:#{project_id}:#{user_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. marksUserAsDisconnected: (project_id, user_id, callback)->
  19. async.series [
  20. (cb)->
  21. rclient.srem buildProjectSetKey(project_id), user_id, cb
  22. (cb)->
  23. rclient.del buildUserKey(project_id, user_id), cb
  24. ], callback
  25. _getConnectedUser: (project_id, user_id, callback)->
  26. rclient.get buildUserKey(project_id, user_id), (err, result)->
  27. if !result?
  28. connected = false
  29. else
  30. connected = true
  31. callback err, {connected:connected, user_id:user_id}
  32. getConnectedUsers: (project_id, callback)->
  33. self = @
  34. rclient.get "connected_users_list:#{project_id}", (err, results)->
  35. jobs = results.map (user_id)->
  36. (cb)->
  37. self._getConnectedUser(project_id, user_id, cb)
  38. async.series jobs, (err, users)->
  39. users = _.filter users, (user)->
  40. user.connected
  41. user_ids = _.map users, (user)->
  42. user.user_id
  43. callback err, user_ids