mirror of
https://github.com/overleaf/overleaf.git
synced 2024-10-31 21:21:03 -04:00
82 lines
No EOL
3 KiB
CoffeeScript
82 lines
No EOL
3 KiB
CoffeeScript
express = require('express')
|
|
http = require("http")
|
|
Settings = require('settings-sharelatex')
|
|
logger = require('logger-sharelatex')
|
|
logger.initialize("documentupdater")
|
|
RedisManager = require('./app/js/RedisManager')
|
|
UpdateManager = require('./app/js/UpdateManager')
|
|
WorkersManager = require('./app/js/WorkersManager')
|
|
Keys = require('./app/js/RedisKeyBuilder')
|
|
redis = require('redis')
|
|
Errors = require "./app/js/Errors"
|
|
HttpController = require "./app/js/HttpController"
|
|
|
|
redisConf = Settings.redis.web
|
|
rclient = redis.createClient(redisConf.port, redisConf.host)
|
|
rclient.auth(redisConf.password)
|
|
|
|
Path = require "path"
|
|
Metrics = require "metrics-sharelatex"
|
|
Metrics.initialize("doc-updater")
|
|
Metrics.mongodb.monitor(Path.resolve(__dirname + "/node_modules/mongojs/node_modules/mongodb"), logger)
|
|
|
|
app = express()
|
|
app.configure ->
|
|
app.use(Metrics.http.monitor(logger));
|
|
app.use express.bodyParser()
|
|
app.use app.router
|
|
|
|
rclient.subscribe("pending-updates")
|
|
rclient.on "message", (channel, doc_key) ->
|
|
[project_id, doc_id] = Keys.splitProjectIdAndDocId(doc_key)
|
|
if !Settings.shuttingDown
|
|
UpdateManager.processOutstandingUpdatesWithLock project_id, doc_id, (error) ->
|
|
logger.error err: error, project_id: project_id, doc_id: doc_id, "error processing update" if error?
|
|
else
|
|
logger.log project_id: project_id, doc_id: doc_id, "ignoring incoming update"
|
|
|
|
WorkersManager.createAndStartWorkers(Settings.workerCount || 10)
|
|
|
|
UpdateManager.resumeProcessing()
|
|
|
|
app.get '/project/:project_id/doc/:doc_id', HttpController.getDoc
|
|
app.post '/project/:project_id/doc/:doc_id', HttpController.setDoc
|
|
app.post '/project/:project_id/doc/:doc_id/flush', HttpController.flushDocIfLoaded
|
|
app.delete '/project/:project_id/doc/:doc_id', HttpController.flushAndDeleteDoc
|
|
app.delete '/project/:project_id', HttpController.deleteProject
|
|
app.post '/project/:project_id/flush', HttpController.flushProject
|
|
|
|
app.get '/total', (req, res)->
|
|
timer = new Metrics.Timer("http.allDocList")
|
|
RedisManager.getCountOfDocsInMemory (err, count)->
|
|
timer.done()
|
|
res.send {total:count}
|
|
|
|
app.get '/status', (req, res)->
|
|
if Settings.shuttingDown
|
|
res.send 503 # Service unavailable
|
|
else
|
|
res.send('document updater is alive')
|
|
|
|
app.use (error, req, res, next) ->
|
|
logger.error err: error, "request errored"
|
|
if error instanceof Errors.NotFoundError
|
|
res.send 404
|
|
else
|
|
res.send(500, "Oops, something went wrong")
|
|
|
|
shutdownCleanly = (signal) ->
|
|
return () ->
|
|
logger.log signal: signal, "received interrupt, cleaning up"
|
|
Settings.shuttingDown = true
|
|
setTimeout () ->
|
|
logger.log signal: signal, "shutting down"
|
|
process.exit()
|
|
, 10000
|
|
|
|
port = Settings.internal?.documentupdater?.port or Settings.apis?.documentupdater?.port or 3003
|
|
app.listen port, "localhost", ->
|
|
logger.log("documentupdater-sharelatex server listening on port #{port}")
|
|
|
|
for signal in ['SIGINT', 'SIGHUP', 'SIGQUIT', 'SIGUSR1', 'SIGUSR2', 'SIGTERM', 'SIGABRT']
|
|
process.on signal, shutdownCleanly(signal) |