timeAsyncMethod.coffee 984 B

12345678910111213141516171819202122232425262728293031
  1. module.exports = (obj, methodName, prefix, logger) ->
  2. metrics = require('./metrics')
  3. if typeof obj[methodName] != 'function'
  4. throw new Error("[Metrics] expected object property '#{methodName}' to be a function")
  5. realMethod = obj[methodName]
  6. key = "#{prefix}.#{methodName}"
  7. obj[methodName] = (originalArgs...) ->
  8. [firstArgs..., callback] = originalArgs
  9. if !callback? || typeof callback != 'function'
  10. if logger?
  11. logger.log "[Metrics] expected wrapped method '#{methodName}' to be invoked with a callback"
  12. return realMethod.apply this, originalArgs
  13. timer = new metrics.Timer(key)
  14. realMethod.call this, firstArgs..., (callbackArgs...) ->
  15. elapsedTime = timer.done()
  16. if logger?
  17. loggableArgs = {}
  18. try
  19. for arg, idx in firstArgs
  20. if arg.toString().match(/^[0-9a-f]{24}$/)
  21. loggableArgs["#{idx}"] = arg
  22. logger.log {key, args: loggableArgs, elapsedTime}, "[Metrics] timed async method call"
  23. callback.apply this, callbackArgs