nodereactionagent/lib/modules/mongo.js

NodeReaction Agent is node application performance monitoring tool

const NodeReactionAgent = require("../Agent");
const mongo = require("mongodb-core");
const library = "MongoDB";

//mongo insert 'Server' callback is the last parameter
let insertOriginal = mongo.Server.prototype.insert;

mongo.Server.prototype.insert = function() {
  // console.log("Node Reaction Agent - MongoDB - Insert");
  if (arguments.length > 0) {
    let trace = NodeReactionAgent.createTrace(library, "Insert");

    const index = arguments.length - 1;
    const cb = arguments[index];

    if (typeof cb === "function") {
      arguments[index] = function() {
        if (trace) trace.end();
        return cb.apply(this, arguments);
      };
    }
  }

  return insertOriginal.apply(this, arguments);
};
//mongo _find 'Cursor' callback is the first parameter
let _findOriginal = mongo.Cursor.prototype._find;

mongo.Cursor.prototype._find = function() {
  // console.log("Node Reaction Agent - MongoDB - Find");
  if (arguments.length > 0) {
    let trace = NodeReactionAgent.createTrace(library, "Find");

    const cb = arguments[0];

    if (typeof cb === "function") {
      arguments[0] = function() {
        if (trace) trace.end();
        return cb.apply(this, arguments);
      };
    }
  }

  return _findOriginal.apply(this, arguments);
};
//mongo update 'Server' callback is the last parameter
let updateOriginal = mongo.Server.prototype.update;

mongo.Server.prototype.update = function() {
  // console.log("Node Reaction Agent - MongoDB - Update");
  if (arguments.length > 0) {
    let trace = NodeReactionAgent.createTrace(library, "Update");

    const index = arguments.length - 1;
    const cb = arguments[index];
    if (typeof cb === "function") {
      arguments[index] = function() {
        if (trace) trace.end();
        return cb.apply(this, arguments);
      };
    }
  }
  return updateOriginal.apply(this, arguments);
};
//mongo remove 'Server' callback is the last parameter
let removeOriginal = mongo.Server.prototype.remove;

mongo.Server.prototype.remove = function() {
  // console.log("Node Reaction Agent - MongoDB - Remove");
  if (arguments.length > 0) {
    let trace = NodeReactionAgent.createTrace(library, "Remove");

    const index = arguments.length - 1;
    const cb = arguments[index];
    if (typeof cb === "function") {
      arguments[index] = function() {
        if (trace) trace.end();
        return cb.apply(this, arguments);
      };
    }
  }
  return removeOriginal.apply(this, arguments);
};

//mongo auth 'Server' callback is the last parameter
let authOriginal = mongo.Server.prototype.auth;

mongo.Server.prototype.auth = function() {
  // console.log("Node Reaction Agent - MongoDB - Auth");
  if (arguments.length > 0) {
    let trace = NodeReactionAgent.createTrace(library, "Auth");

    const index = arguments.length - 1;
    const cb = arguments[index];
    if (typeof cb === "function") {
      arguments[index] = function() {
        if (trace) trace.end();
        return cb.apply(this, arguments);
      };
    }
  }
  return authOriginal.apply(this, arguments);
};
//mongo _getmore 'Cursor' callback is the first parameter
let getMoreOriginal = mongo.Cursor.prototype._getmore;

mongo.Cursor.prototype._getmore = function() {
  // console.log("Node Reaction Agent - MongoDB - Get More");
  if (arguments.length > 0) {
    let trace = NodeReactionAgent.createTrace(library, "GetMore");

    const cb = arguments[0];

    if (typeof cb === "function") {
      arguments[0] = function() {
        if (trace) trace.end();
        return cb.apply(this, arguments);
      };
    }
  }

  return getMoreOriginal.apply(this, arguments);
};

module.exports = mongo;