timeAsyncMethod.coffee 1.3 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243
  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. keys = prefix.split(".")
  7. key = keys[0].toLowerCase()
  8. if keys[1]?
  9. methodName = "#{keys[1]}_#{methodName}"
  10. console.log "Async method", keys, methodName
  11. obj[methodName] = (originalArgs...) ->
  12. [firstArgs..., callback] = originalArgs
  13. if !callback? || typeof callback != 'function'
  14. if logger?
  15. logger.log "[Metrics] expected wrapped method '#{methodName}' to be invoked with a callback"
  16. return realMethod.apply this, originalArgs
  17. console.log("creating timer for async method")
  18. timer = new metrics.Timer(key, null, {method: methodName})
  19. realMethod.call this, firstArgs..., (callbackArgs...) ->
  20. elapsedTime = timer.done()
  21. possibleError = callbackArgs[0]
  22. if possibleError?
  23. metrics.inc "#{key}.failure"
  24. else
  25. metrics.inc "#{key}.success"
  26. if logger?
  27. loggableArgs = {}
  28. try
  29. for arg, idx in firstArgs
  30. if arg.toString().match(/^[0-9a-f]{24}$/)
  31. loggableArgs["#{idx}"] = arg
  32. logger.log {key, args: loggableArgs, elapsedTime}, "[Metrics] timed async method call"
  33. callback.apply this, callbackArgs