overleaf/services/web/app/coffee/Features/DocumentUpdater/DocumentUpdaterHandler.coffee

161 lines
7.3 KiB
CoffeeScript
Raw Normal View History

2014-02-12 05:23:40 -05:00
request = require 'request'
request = request.defaults()
async = require 'async'
settings = require 'settings-sharelatex'
_ = require 'underscore'
async = require 'async'
logger = require('logger-sharelatex')
metrics = require('../../infrastructure/Metrics')
2014-09-26 09:52:00 -04:00
redis = require("redis-sharelatex")
rclient = redis.createClient(settings.redis.web)
2014-02-12 05:23:40 -05:00
Project = require("../../models/Project").Project
ProjectLocator = require('../../Features/Project/ProjectLocator')
module.exports = DocumentUpdaterHandler =
2014-02-12 05:23:40 -05:00
2014-10-15 09:11:02 -04:00
queueChange : (project_id, doc_id, change, callback = ()->)->
2014-02-12 05:23:40 -05:00
jsonChange = JSON.stringify change
doc_key = keys.combineProjectIdAndDocId(project_id, doc_id)
multi = rclient.multi()
multi.rpush keys.pendingUpdates(doc_id:doc_id), jsonChange
multi.sadd keys.docsWithPendingUpdates, doc_key
multi.rpush "pending-updates-list", doc_key
multi.exec (error) ->
2014-02-12 05:23:40 -05:00
return callback(error) if error?
callback()
2014-02-12 05:23:40 -05:00
2014-10-15 09:11:02 -04:00
flushProjectToMongo: (project_id, callback = (error) ->)->
logger.log project_id:project_id, "flushing project from document updater"
2014-02-12 05:23:40 -05:00
timer = new metrics.Timer("flushing.mongo.project")
url = "#{settings.apis.documentupdater.url}/project/#{project_id}/flush"
request.post url, (error, res, body)->
if error?
2014-10-15 09:11:02 -04:00
logger.error err: error, project_id: project_id, "error flushing project from document updater"
2014-02-12 05:23:40 -05:00
return callback(error)
else if res.statusCode >= 200 and res.statusCode < 300
2014-10-15 09:11:02 -04:00
logger.log project_id: project_id, "flushed project from document updater"
2014-02-12 05:23:40 -05:00
return callback(null)
else
error = new Error("document updater returned a failure status code: #{res.statusCode}")
2014-10-15 09:11:02 -04:00
logger.error err: error, project_id: project_id, "document updater returned failure status code: #{res.statusCode}"
2014-02-12 05:23:40 -05:00
return callback(error)
flushMultipleProjectsToMongo: (project_ids, callback = (error) ->) ->
jobs = []
for project_id in project_ids
do (project_id) ->
jobs.push (callback) ->
DocumentUpdaterHandler.flushProjectToMongo project_id, callback
async.series jobs, callback
2014-10-15 09:11:02 -04:00
flushProjectToMongoAndDelete: (project_id, callback = ()->) ->
logger.log project_id:project_id, "deleting project from document updater"
2014-02-12 05:23:40 -05:00
timer = new metrics.Timer("delete.mongo.project")
url = "#{settings.apis.documentupdater.url}/project/#{project_id}"
request.del url, (error, res, body)->
if error?
2014-10-15 09:11:02 -04:00
logger.error err: error, project_id: project_id, "error deleting project from document updater"
2014-02-12 05:23:40 -05:00
return callback(error)
else if res.statusCode >= 200 and res.statusCode < 300
2014-10-15 09:11:02 -04:00
logger.log project_id: project_id, "deleted project from document updater"
2014-02-12 05:23:40 -05:00
return callback(null)
else
error = new Error("document updater returned a failure status code: #{res.statusCode}")
2014-10-15 09:11:02 -04:00
logger.error err: error, project_id: project_id, "document updater returned failure status code: #{res.statusCode}"
2014-05-08 10:47:50 -04:00
return callback(error)
flushDocToMongo: (project_id, doc_id, callback = (error) ->) ->
logger.log project_id:project_id, doc_id: doc_id, "flushing doc from document updater"
timer = new metrics.Timer("flushing.mongo.doc")
url = "#{settings.apis.documentupdater.url}/project/#{project_id}/doc/#{doc_id}/flush"
request.post url, (error, res, body)->
if error?
logger.error err: error, project_id: project_id, doc_id: doc_id, "error flushing doc from document updater"
return callback(error)
else if res.statusCode >= 200 and res.statusCode < 300
logger.log project_id: project_id, doc_id: doc_id, "flushed doc from document updater"
return callback(null)
else
error = new Error("document updater returned a failure status code: #{res.statusCode}")
logger.error err: error, project_id: project_id, doc_id: doc_id, "document updater returned failure status code: #{res.statusCode}"
2014-02-12 05:23:40 -05:00
return callback(error)
2014-10-15 09:11:02 -04:00
deleteDoc : (project_id, doc_id, callback = ()->)->
logger.log project_id:project_id, doc_id: doc_id, "deleting doc from document updater"
2014-02-12 05:23:40 -05:00
timer = new metrics.Timer("delete.mongo.doc")
url = "#{settings.apis.documentupdater.url}/project/#{project_id}/doc/#{doc_id}"
request.del url, (error, res, body)->
if error?
2014-10-15 09:11:02 -04:00
logger.error err: error, project_id: project_id, doc_id: doc_id, "error deleting doc from document updater"
2014-02-12 05:23:40 -05:00
return callback(error)
else if res.statusCode >= 200 and res.statusCode < 300
2014-10-15 09:11:02 -04:00
logger.log project_id: project_id, doc_id: doc_id, "deleted doc from document updater"
2014-02-12 05:23:40 -05:00
return callback(null)
else
error = new Error("document updater returned a failure status code: #{res.statusCode}")
2014-10-15 09:11:02 -04:00
logger.error err: error, project_id: project_id, doc_id: doc_id, "document updater returned failure status code: #{res.statusCode}"
2014-02-12 05:23:40 -05:00
return callback(error)
2014-10-15 09:11:02 -04:00
getDocument: (project_id, doc_id, fromVersion, callback = (error, exists, doclines, version) ->) ->
2014-02-12 05:23:40 -05:00
timer = new metrics.Timer("get-document")
url = "#{settings.apis.documentupdater.url}/project/#{project_id}/doc/#{doc_id}?fromVersion=#{fromVersion}"
2014-10-15 09:11:02 -04:00
logger.log project_id:project_id, doc_id: doc_id, "getting doc from document updater"
2014-02-12 05:23:40 -05:00
request.get url, (error, res, body)->
timer.done()
if error?
logger.error err:error, url:url, project_id:project_id, doc_id:doc_id, "error getting doc from doc updater"
return callback(error)
if res.statusCode >= 200 and res.statusCode < 300
logger.log project_id:project_id, doc_id:doc_id, "got doc from document document updater"
try
body = JSON.parse(body)
catch error
return callback(error)
callback null, body.lines, body.version, body.ops
else
logger.error project_id:project_id, doc_id:doc_id, url: url, "doc updater returned a non-success status code: #{res.statusCode}"
callback new Error("doc updater returned a non-success status code: #{res.statusCode}")
setDocument : (project_id, doc_id, docLines, source, callback = (error) ->)->
2014-02-12 05:23:40 -05:00
timer = new metrics.Timer("set-document")
url = "#{settings.apis.documentupdater.url}/project/#{project_id}/doc/#{doc_id}"
body =
url: url
json:
lines: docLines
source: source
logger.log project_id:project_id, doc_id: doc_id, source: source, "setting doc in document updater"
2014-02-12 05:23:40 -05:00
request.post body, (error, res, body)->
timer.done()
if error?
logger.error err:error, url:url, project_id:project_id, doc_id:doc_id, "error setting doc in doc updater"
return callback(error)
if res.statusCode >= 200 and res.statusCode < 300
2014-10-15 09:11:02 -04:00
logger.log project_id: project_id, doc_id: doc_id, "set doc in document updater"
2014-02-12 05:23:40 -05:00
return callback(null)
else
logger.error project_id:project_id, doc_id:doc_id, url: url, "doc updater returned a non-success status code: #{res.statusCode}"
callback new Error("doc updater returned a non-success status code: #{res.statusCode}")
getNumberOfDocsInMemory : (callback)->
request.get "#{settings.apis.documentupdater.url}/total", (err, req, body)->
try
body = JSON.parse body
catch err
logger.err err:err, "error parsing response from doc updater about the total number of docs"
callback(err, body?.total)
PENDINGUPDATESKEY = "PendingUpdates"
DOCLINESKEY = "doclines"
DOCIDSWITHPENDINGUPDATES = "DocsWithPendingUpdates"
keys =
pendingUpdates : (op) -> "#{PENDINGUPDATESKEY}:#{op.doc_id}"
docsWithPendingUpdates: DOCIDSWITHPENDINGUPDATES
docLines : (op) -> "#{DOCLINESKEY}:#{op.doc_id}"
combineProjectIdAndDocId: (project_id, doc_id) -> "#{project_id}:#{doc_id}"