overleaf/libraries/metrics/timeAsyncMethod.coffee

37 lines
1.1 KiB
CoffeeScript
Raw Normal View History

2017-03-15 15:06:54 +00:00
2017-03-17 11:47:38 +00:00
module.exports = (obj, methodName, prefix, logger) ->
2017-03-15 15:06:54 +00:00
metrics = require('./metrics')
if typeof obj[methodName] != 'function'
2017-03-16 10:02:19 +00:00
throw new Error("[Metrics] expected object property '#{methodName}' to be a function")
2017-03-15 15:06:54 +00:00
realMethod = obj[methodName]
2017-03-17 11:47:38 +00:00
key = "#{prefix}.#{methodName}"
2017-03-15 15:06:54 +00:00
obj[methodName] = (originalArgs...) ->
[firstArgs..., callback] = originalArgs
2017-03-15 15:06:54 +00:00
if !callback? || typeof callback != 'function'
if logger?
logger.log "[Metrics] expected wrapped method '#{methodName}' to be invoked with a callback"
return realMethod.apply this, originalArgs
2017-03-15 15:06:54 +00:00
2017-03-15 16:07:36 +00:00
timer = new metrics.Timer(key)
2017-03-15 15:06:54 +00:00
2017-03-15 16:07:36 +00:00
realMethod.call this, firstArgs..., (callbackArgs...) ->
2017-03-16 15:07:25 +00:00
elapsedTime = timer.done()
possibleError = callbackArgs[0]
if possibleError?
metrics.inc "#{key}.failure"
else
metrics.inc "#{key}.success"
2017-03-15 15:06:54 +00:00
if logger?
loggableArgs = {}
try
for arg, idx in firstArgs
if arg.toString().match(/^[0-9a-f]{24}$/)
loggableArgs["#{idx}"] = arg
logger.log {key, args: loggableArgs, elapsedTime}, "[Metrics] timed async method call"
2017-03-15 15:06:54 +00:00
callback.apply this, callbackArgs