meteor-elastic-apm.js

/* eslint-disable no-undef */
const shimmer = require('shimmer');
const Fibers = require('fibers');
const Agent = require('elastic-apm-node');

const { Session, Subscription, MongoCursor } = require('./meteorx');

const instrumentMethods = require('./instrumenting/methods');
const instrumentHttpIn = require('./instrumenting/http-in');
const instrumentHttpOut = require('./instrumenting/http-out');
const instrumentSession = require('./instrumenting/session');
const instrumentSubscription = require('./instrumenting/subscription');
const instrumentAsync = require('./instrumenting/async');
const instrumentDB = require('./instrumenting/db');
const startMetrics = require('./metrics');

const hackDB = require('./hacks');

const [framework, version] = Meteor.release.split('@');

Agent.setFramework({
  name: framework,
  version,
  override: true
});

shimmer.wrap(Agent, 'start', function(startAgent) {
  return function(...args) {
    const config = args[0] || {};

    if (config.active !== false) {
      try {
        // Must be called before any other route is registered on WebApp.
        instrumentHttpIn(Agent, WebApp);

        Meteor.startup(() => {
          try {
            hackDB();

            instrumentMethods(Agent, Meteor);
            instrumentHttpOut(Agent, WebApp);
            instrumentSession(Agent, Session);
            instrumentSubscription(Agent, Subscription);
            instrumentAsync(Agent, Fibers);
            instrumentDB(Agent, Meteor, MongoCursor);

            startAgent.apply(Agent, args);

            startMetrics(Agent);

            Agent.logger.info('meteor-elastic-apm completed instrumenting');
          } catch (e) {
            Agent.logger.error('Could not start meteor-elastic-apm');
            throw e;
          }
        });
      } catch (e) {
        Agent.logger.error('Could not start meteor-elastic-apm');
        throw e;
      }
    } else {
      Agent.logger.warn('meteor-elastic-apm is not active');
    }
  };
});

module.exports = Agent;