ca-apm-probe/lib/probes/apollo-server-express.js

CA APM Node.js Agent monitors real-time health and performance of Node.js applications

/**
 * Copyright (c) 2015 CA. All rights reserved.
 *
 * This software and all information contained therein is confidential and proprietary and
 * shall not be duplicated, used, disclosed or disseminated in any way except as authorized
 * by the applicable license agreement, without the express written permission of CA. All
 * authorized reproductions must be marked with this language.
 *
 * EXCEPT AS SET FORTH IN THE APPLICABLE LICENSE AGREEMENT, TO THE EXTENT
 * PERMITTED BY APPLICABLE LAW, CA PROVIDES THIS SOFTWARE WITHOUT WARRANTY
 * OF ANY KIND, INCLUDING WITHOUT LIMITATION, ANY IMPLIED WARRANTIES OF
 * MERCHANTABILITY OR FITNESS FOR A PARTICULAR PURPOSE. IN NO EVENT WILL CA BE
 * LIABLE TO THE END USER OR ANY THIRD PARTY FOR ANY LOSS OR DAMAGE, DIRECT OR
 * INDIRECT, FROM THE USE OF THIS SOFTWARE, INCLUDING WITHOUT LIMITATION, LOST
 * PROFITS, BUSINESS INTERRUPTION, GOODWILL, OR LOST DATA, EVEN IF CA IS
 * EXPRESSLY ADVISED OF SUCH LOSS OR DAMAGE.
 */

var path = require('path');
var agent = require('../agent');
var proxy = require('../proxy');

var util = require('util');
var logger = require("../logger.js");

var logGraphqlVersionOnce = false;
var targetModule = new Object;
module.exports = function (apollo) {

  targetModule.apolloserver = findExecution(apollo, require);
  proxy.before(targetModule.apolloserver.ApolloServer.prototype, 'getMiddleware', queryHook);
};

function queryHook(obj, args, storage) {
  if(!agent.graphQLUrl){
    agent.graphQLUrl = args[0].path;
  }
}

function findExecution(parent, require) {
  var cache = require.cache;
  for (var key in cache) {
    var candidate = cache[key];
    if (candidate.exports !== parent) {
      continue;
    }

    var dirname = path.dirname(candidate.filename);
    var filename = path.join(dirname, 'ApolloServer.js');
    try {
      return require(filename);
    } catch (e) {
      return null;
    }
  }
  return null;
}