bq/corbel-composr

'use strict'
var hub = require('../lib/hub')
var transactions = require('./transactions')
var fProxy = require('../utils/proxify.js')
var yn = require('yn')
var newrelic
/* *********************************
Newrelic events
**********************************/

function createNRTransaction (item) {
  return new Promise(function (resolve, reject) {
    newrelic.createWebTransaction(item.url, function (transaction) {
      if (transaction.segments) {
        var promises = transaction.segments.map(createNRTransaction)
        newrelic.endTransaction()
        Promise.all(promises)
          .then(resolve)
      } else {
        Object.keys(transaction).forEach(function (attribute) {
          newrelic.addCustomParameter(attribute, transaction.hasOwnProperty(attribute) ? transaction[attribute] : 'undefined attribute')
        })

        var startTime = transaction.time
        var tTime = transaction.endDate - startTime
        newrelic.agent.tracer.segment.transaction.timer.start = startTime
        newrelic.agent.tracer.segment.transaction.timer.hrstart = [Math.floor(startTime / 1e3), startTime % 1e3 * 1e6]
        newrelic.agent.tracer.segment.transaction.webSegment.timer.overwriteDurationInMillis(tTime)
        newrelic.endTransaction()
        return resolve()
      }
    })(item)
  })
}

function notifyTransaction (transactionId) {
  var transaction = transactions.getTransactionById(transactionId)
  createNRTransaction(transaction)
    .then(function () {
      transactions.deleteTransactionById(transactionId)
    })
}

function initMetrics (config, logger) {
  if (yn(config.get('newrelic.enabled'))) {
    newrelic = require('newrelic')
    logger.info('[Metrics]', 'Initializing NewRelic events...')
    fProxy.proxifyFunction(newrelic, ['agent', '_transactionFinished'], function (cb, transaction) {
      if (transaction.partialName.indexOf('Custom') === -1) {
        transaction.forceIgnore = true
        transaction.ignore = true
      }
      cb.call(this, transaction)
    })
    newrelic.agent._events['transactionFinished'] = newrelic.agent._transactionFinished

    hub.on('server:start', function () {
      newrelic.recordCustomEvent('server:start', {
        date: Date.now()
      })
    })

    hub.on('rabbitmq:error', function (err) {
      newrelic.recordCustomEvent('rabbitmq:error', {
        date: Date.now(),
        err: err
      })
    })

    hub.on('http:status', function (status, url, method) {
      newrelic.recordCustomEvent('http:status', {
        url: url,
        status: status,
        method: method
      })
    })

    hub.on('http:end', function (req, res) {
      var domain = req.url.match(/((?!\/).+:\w+)/)
      // @TODO: check if req.domain is setted by some middleware.
      var evtData = {
        guid: req.getId(),
        url: req.url,
        time: req.time(),
        startDate: req.date(),
        endDate: Date.now(),
        status: res.statusCode,
        method: req.method,
        domain: (domain && domain[0] ? domain[0] : 'domain:not:found'),
        isError: (res.statusCode.toString().indexOf('4') === 0 || res.statusCode.toString().indexOf('5') === 0),
        type: 'ROOT'
      }

      transactions.addSegment(req.getId(), evtData)
      notifyTransaction(req.getId())
    })

    hub.on('metrics:add:segment', function (evt) {
      transactions.addSegment(evt.guid, evt)
    })

    hub.on('metrics', function (domain, options) {
      newrelic.recordCustomEvent('client:metrics', {
        domain: domain,
        data: options
      })
    })

    hub.on('http:start', function (path, method, reqId) {
      newrelic.setIgnoreTransaction(true)
      transactions.addTransaction(reqId)
    })
  }
}

module.exports = initMetrics