mirror of
https://github.com/overleaf/overleaf.git
synced 2024-10-31 21:21:03 -04:00
fc3ecddd0e
1. the compileGroup is "standard" not default 2. was not excluding normal compiles from metrics
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")
|
|
Project = require("../../models/Project").Project
|
|
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) ->) ->
|
|
Project.findById 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) ->) ->
|
|
Project.findById project_id, 'rootDoc_id', (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
|