mirror of
https://github.com/overleaf/overleaf.git
synced 2024-11-21 20:47:08 -05:00
119 lines
5.1 KiB
CoffeeScript
Executable file
119 lines
5.1 KiB
CoffeeScript
Executable file
Settings = require('settings-sharelatex')
|
|
RedisWrapper = require("../../infrastructure/RedisWrapper")
|
|
rclient = RedisWrapper.client("clsi_recently_compiled")
|
|
ProjectGetter = require('../Project/ProjectGetter')
|
|
ProjectRootDocManager = require "../Project/ProjectRootDocManager"
|
|
UserGetter = require "../User/UserGetter"
|
|
ClsiManager = require "./ClsiManager"
|
|
Metrics = require('metrics-sharelatex')
|
|
logger = require("logger-sharelatex")
|
|
rateLimiter = require("../../infrastructure/RateLimiter")
|
|
|
|
module.exports = CompileManager =
|
|
|
|
|
|
compile: (project_id, user_id, options = {}, _callback = (error) ->) ->
|
|
timer = new Metrics.Timer("editor.compile")
|
|
callback = (args...) ->
|
|
timer.done()
|
|
_callback(args...)
|
|
|
|
@_checkIfAutoCompileLimitHasBeenHit options.isAutoCompile, "everyone", (err, canCompile)->
|
|
if !canCompile
|
|
return callback null, "autocompile-backoff", []
|
|
logger.log project_id: project_id, user_id: user_id, "compiling project"
|
|
CompileManager._checkIfRecentlyCompiled project_id, user_id, (error, recentlyCompiled) ->
|
|
return callback(error) if error?
|
|
if recentlyCompiled
|
|
logger.warn {project_id, user_id}, "project was recently compiled so not continuing"
|
|
return callback null, "too-recently-compiled", []
|
|
|
|
CompileManager._ensureRootDocumentIsSet project_id, (error) ->
|
|
return callback(error) if error?
|
|
CompileManager.getProjectCompileLimits project_id, (error, limits) ->
|
|
return callback(error) if error?
|
|
for key, value of limits
|
|
options[key] = value
|
|
# Put a lower limit on autocompiles for free users, based on compileGroup
|
|
CompileManager._checkCompileGroupAutoCompileLimit options.isAutoCompile, limits.compileGroup, (err, canCompile)->
|
|
if !canCompile
|
|
return callback null, "autocompile-backoff", []
|
|
# only pass user_id down to clsi if this is a per-user compile
|
|
compileAsUser = if Settings.disablePerUserCompiles then undefined else user_id
|
|
ClsiManager.sendRequest project_id, compileAsUser, options, (error, status, outputFiles, clsiServerId, validationProblems) ->
|
|
return callback(error) if error?
|
|
logger.log files: outputFiles, "output files"
|
|
callback(null, status, outputFiles, clsiServerId, limits, validationProblems)
|
|
|
|
|
|
stopCompile: (project_id, user_id, callback = (error) ->) ->
|
|
CompileManager.getProjectCompileLimits project_id, (error, limits) ->
|
|
return callback(error) if error?
|
|
ClsiManager.stopCompile project_id, user_id, limits, callback
|
|
|
|
deleteAuxFiles: (project_id, user_id, callback = (error) ->) ->
|
|
CompileManager.getProjectCompileLimits project_id, (error, limits) ->
|
|
return callback(error) if error?
|
|
ClsiManager.deleteAuxFiles project_id, user_id, limits, callback
|
|
|
|
getProjectCompileLimits: (project_id, callback = (error, limits) ->) ->
|
|
ProjectGetter.getProject project_id, owner_ref: 1, (error, project) ->
|
|
return callback(error) if error?
|
|
UserGetter.getUser project.owner_ref, {"features":1}, (err, owner)->
|
|
return callback(error) if error?
|
|
callback null, {
|
|
timeout: owner.features?.compileTimeout || Settings.defaultFeatures.compileTimeout
|
|
compileGroup: owner.features?.compileGroup || Settings.defaultFeatures.compileGroup
|
|
}
|
|
|
|
COMPILE_DELAY: 1 # seconds
|
|
_checkIfRecentlyCompiled: (project_id, user_id, callback = (error, recentlyCompiled) ->) ->
|
|
key = "compile:#{project_id}:#{user_id}"
|
|
rclient.set key, true, "EX", @COMPILE_DELAY, "NX", (error, ok) ->
|
|
return callback(error) if error?
|
|
if ok == "OK"
|
|
return callback null, false
|
|
else
|
|
return callback null, true
|
|
|
|
_checkCompileGroupAutoCompileLimit: (isAutoCompile, compileGroup, callback = (err, canCompile)->)->
|
|
if !isAutoCompile
|
|
return callback(null, true)
|
|
if compileGroup is "standard"
|
|
# apply extra limits to the standard compile group
|
|
CompileManager._checkIfAutoCompileLimitHasBeenHit isAutoCompile, compileGroup, callback
|
|
else
|
|
Metrics.inc "auto-compile-#{compileGroup}"
|
|
return callback(null, true) # always allow priority group users to compile
|
|
|
|
_checkIfAutoCompileLimitHasBeenHit: (isAutoCompile, compileGroup, callback = (err, canCompile)->)->
|
|
if !isAutoCompile
|
|
return callback(null, true)
|
|
Metrics.inc "auto-compile-#{compileGroup}"
|
|
opts =
|
|
endpointName:"auto_compile"
|
|
timeInterval:20
|
|
subjectName:compileGroup
|
|
throttle: Settings?.rateLimit?.autoCompile?[compileGroup] || 25
|
|
rateLimiter.addCount opts, (err, canCompile)->
|
|
if err?
|
|
canCompile = false
|
|
if !canCompile
|
|
Metrics.inc "auto-compile-#{compileGroup}-limited"
|
|
callback err, canCompile
|
|
|
|
_ensureRootDocumentIsSet: (project_id, callback = (error) ->) ->
|
|
ProjectGetter.getProject project_id, rootDoc_id: 1, (error, project) ->
|
|
return callback(error) if error?
|
|
if !project?
|
|
return callback new Error("project not found")
|
|
|
|
if project.rootDoc_id?
|
|
callback()
|
|
else
|
|
ProjectRootDocManager.setRootDocAutomatically project_id, callback
|
|
|
|
wordCount: (project_id, user_id, file, callback = (error) ->) ->
|
|
CompileManager.getProjectCompileLimits project_id, (error, limits) ->
|
|
return callback(error) if error?
|
|
ClsiManager.wordCount project_id, user_id, file, limits, callback
|