mirror of
https://github.com/overleaf/overleaf.git
synced 2024-10-31 21:21:03 -04:00
37 lines
1.2 KiB
CoffeeScript
37 lines
1.2 KiB
CoffeeScript
|
Settings = require('settings-sharelatex')
|
||
|
|
logger = require('logger-sharelatex')
|
||
|
|
Metrics = require('./Metrics')
|
||
|
|
|
||
|
|
module.exports = class RateLimiter
|
||
|
|
|
||
|
|
constructor: (number = 10) ->
|
||
|
|
@ActiveWorkerCount = 0
|
||
|
|
@CurrentWorkerLimit = number
|
||
|
|
@BaseWorkerCount = number
|
||
|
|
|
||
|
|
_adjustLimitUp: () ->
|
||
|
|
@CurrentWorkerLimit += 0.1 # allow target worker limit to increase gradually
|
||
|
|
|
||
|
|
_adjustLimitDown: () ->
|
||
|
|
@CurrentWorkerLimit = Math.max @BaseWorkerCount, (@CurrentWorkerLimit * 0.9)
|
||
|
|
logger.log {currentLimit: Math.ceil(@CurrentWorkerLimit)}, "reducing rate limit"
|
||
|
|
|
||
|
|
_trackAndRun: (task, callback = () ->) ->
|
||
|
|
@ActiveWorkerCount++
|
||
|
|
Metrics.gauge "processingUpdates", "+1" # increments/decrements gauge with +/- sign
|
||
|
|
task (err) =>
|
||
|
|
@ActiveWorkerCount--
|
||
|
|
Metrics.gauge "processingUpdates", "-1"
|
||
|
|
callback(err)
|
||
|
|
|
||
|
|
run: (task, callback) ->
|
||
|
|
if @ActiveWorkerCount < @CurrentWorkerLimit
|
||
|
|
@_trackAndRun task # below the limit, just put the task in the background
|
||
|
|
callback() # return immediately
|
||
|
|
if @CurrentWorkerLimit > @BaseWorkerCount
|
||
|
|
@_adjustLimitDown()
|
||
|
|
else
|
||
|
|
logger.log {active: @ActiveWorkerCount, currentLimit: Math.ceil(@CurrentWorkerLimit)}, "hit rate limit"
|
||
|
|
@_trackAndRun task, callback # only return after task completes
|
||
|
|
@_adjustLimitUp()
|