timeAsyncMethod.coffee 933 B

123456789101112131415161718192021222324252627282930
  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. logger.log "[Metrics] expected wrapped method '#{methodName}' to be invoked with a callback"
  11. return
  12. timer = new metrics.Timer(key)
  13. realMethod.call this, firstArgs..., (callbackArgs...) ->
  14. elapsedTime = timer.done()
  15. if logger?
  16. loggableArgs = {}
  17. try
  18. for arg, idx in firstArgs
  19. if arg.toString().match(/^[0-9a-f]{24}$/)
  20. loggableArgs["#{idx}"] = arg
  21. logger.log {key, args: loggableArgs, elapsedTime}, "[Metrics] timed async method call"
  22. callback.apply this, callbackArgs