ConnectedUsersManager.coffee 2.5 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576
  1. _ = require("underscore")
  2. async = require("async")
  3. Settings = require('settings-sharelatex')
  4. logger = require("logger-sharelatex")
  5. redis = require('redis')
  6. rclient = redis.createClient(Settings.redis.web.port, Settings.redis.web.host)
  7. rclient.auth(Settings.redis.web.password)
  8. ONE_HOUR_IN_S = 60 * 60
  9. ONE_DAY_IN_S = ONE_HOUR_IN_S * 24
  10. FOUR_DAYS_IN_S = ONE_DAY_IN_S * 4
  11. USER_TIMEOUT_IN_S = ONE_HOUR_IN_S
  12. buildProjectSetKey = (project_id)-> return "users_in_project:#{project_id}"
  13. buildUserKey = (project_id, user_id)-> return "connected_user:#{project_id}:#{user_id}"
  14. module.exports =
  15. markUserAsConnected: (project_id, user_id, callback = (err)->)->
  16. logger.log project_id:project_id, user_id:user_id, "marking user as connected"
  17. multi = rclient.multi()
  18. multi.sadd buildProjectSetKey(project_id), user_id
  19. multi.expire buildProjectSetKey(project_id), FOUR_DAYS_IN_S
  20. multi.hset buildUserKey(project_id, user_id), "connected_at", new Date()
  21. multi.expire buildUserKey(project_id, user_id), USER_TIMEOUT_IN_S
  22. multi.exec (err)->
  23. if err?
  24. logger.err err:err, project_id:project_id, user_id:user_id, "problem marking user as connected"
  25. callback(err)
  26. markUserAsDisconnected: (project_id, user_id, callback)->
  27. logger.log project_id:project_id, user_id:user_id, "marking user as disconnected"
  28. multi = rclient.multi()
  29. multi.srem buildProjectSetKey(project_id), user_id
  30. multi.expire buildProjectSetKey(project_id), FOUR_DAYS_IN_S
  31. multi.del buildUserKey(project_id, user_id)
  32. multi.exec callback
  33. _getConnectedUser: (project_id, user_id, callback)->
  34. rclient.hgetall buildUserKey(project_id, user_id), (err, result)->
  35. if !result?
  36. result =
  37. connected : false
  38. user_id:user_id
  39. else
  40. result.connected = true
  41. result.user_id = user_id
  42. if result.cursorData?
  43. result.cursorData = JSON.parse(result.cursorData)
  44. result.email = result.cursorData.email
  45. result.name = result.cursorData.name
  46. callback err, result
  47. setUserCursorPosition: (project_id, user_id, cursorData, callback)->
  48. multi = rclient.multi()
  49. multi.hset buildUserKey(project_id, user_id), "cursorData", JSON.stringify(cursorData)
  50. multi.expire buildUserKey(project_id, user_id), USER_TIMEOUT_IN_S
  51. multi.exec callback
  52. getConnectedUsers: (project_id, callback)->
  53. self = @
  54. rclient.smembers buildProjectSetKey(project_id), (err, results)->
  55. jobs = results.map (user_id)->
  56. (cb)->
  57. self._getConnectedUser(project_id, user_id, cb)
  58. async.series jobs, (err, users)->
  59. users = _.filter users, (user)->
  60. user.connected
  61. callback err, users