ca-apm-probe
Version:
CA APM Node.js Agent monitors real-time health and performance of Node.js applications
212 lines (186 loc) • 7.09 kB
JavaScript
/**
* 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('ca-apm-probe/lib/agent');
var proxy = require('ca-apm-probe/lib/proxy');
var util = require('util');
var logger = require("ca-apm-probe/lib/logger");
var traceUtil = require("ca-apm-probe/lib/utils/trace-util");
const { prototype } = require('stream');
var targetModule = new Object;
var eventNameFormatted = 'mssql.query';
module.exports = function(mssql) {
targetModule.mssql = mssql;
logger.info('Loading Mssql probe');
findRequest(targetModule.mssql, require);
targetModule.Request;
targetModule.Base;
targetModule.methodMap = getMethodsWithProbes();
var methodMap = targetModule.methodMap;
var Request = targetModule.Request;
var Base = targetModule.Base;
if(Request){
Request = targetModule.Request;
}
else if(Base){
Request = Base.Request;
}
proxy.before(Request.prototype, 'query', function(obj, args, storage) {
if (agent.paused) return;
// check if we need to instrument
if (methodMap[0] == "skip_instrument") { return; }
var command = args.length > 0 ? args[0] : undefined;
console.log(command);
var dburl;
var config;
if (obj.parent.config) {
config = obj.parent.config;
dburl = 'sqlserver://' + config.server + ':' + config.port + '/' + config.database;
}
else if(obj.parent._acquiredConfig){
config = obj.parent._acquiredConfig;
dburl = 'sqlserver://' + config.server + ':' + config.port + '/' + config.database;
}
var ctx = storage.get('ctx');
if (ctx != null) {
logger.debug('mssql[%d %d %d]: %s', ctx.txid, ctx.lane, ctx.evtid, command);
}
else {
logger.debug('mssql.query - no context: %s', command);
logger.debug((new Error('No context')).stack);
}
ctx = agent.asynchEventStart(ctx, eventNameFormatted, { url: dburl, query: command});
storage.set('ctx', ctx);
proxy.callback(args, -1, function(obj, args, storage) {
var errorObject = agent.checkAndSetErrorObject(args, 'mssqlError');
if (ctx != null) {
logger.debug('mssql[%d %d %d]: callback', ctx.txid, ctx.lane, ctx.evtid);
ctx = agent.asynchEventDone(ctx, eventNameFormatted, null, errorObject);
storage.set('ctx', ctx);
}
else {
logger.debug('mssql - no context: callback');
logger.debug((new Error('No context')).stack);
}
}, function(obj, args) {
if (ctx) {
logger.debug('mssql[%d %d %d]: finish', ctx.txid, ctx.lane, ctx.evtid);
agent.asynchEventFinish(ctx);
}
else {
logger.debug('mssql - no context: finish');
logger.debug((new Error('No context')).stack);
}
});
});
proxy.after(Request.prototype, 'query', function(obj, args, promise, storage) {
if (agent.paused) return;
var ctx = storage.get('ctx');
//For async calls with Promise have to be handled after function execution using Promise.then in after API
if (ctx) {
logger.debug('mssql[%d %d %d]', ctx.txid, ctx.lane, ctx.evtid);
//Two handlers are used for resolve and reject of Promise
promise.then(getPromiseSuccessHandler(storage, ctx, obj), getPromiseRejectionHandler(storage, ctx, obj));
}
else {
logger.debug('mssql - no context: callback');
logger.debug((new Error('No context')).stack);
}
});
}
function getPromiseSuccessHandler(storage, ctx, obj) {
return function mssqlPromiseSuccessHandler(val) {
logger.debug('event name: %s , current context: [%d %d %d]', eventNameFormatted, ctx.txid, ctx.lane, ctx.evtid);
var eventArgs = {};
if (obj) {
if (obj.parent.config && obj.parent.config.server) {
eventArgs.dbServer = obj.parent.config.server;
}
else if(obj.parent._acquiredConfig && obj.parent._acquiredConfig.server){
eventArgs.dbServer = obj.parent._acquiredConfig.server;
}
}
ctx = agent.asynchEventDone(ctx, eventNameFormatted, eventArgs, null);
storage.set('ctx', ctx);
agent.asynchEventFinish(ctx);
};
}
function getPromiseRejectionHandler(storage, ctx, obj) {
return function mssqlPromiseRejectionHandler(val) {
logger.debug('event name: %s , current context: [%d %d %d]', eventNameFormatted, ctx.txid, ctx.lane, ctx.evtid);
var eventArgs = {};
var errorObject = traceUtil.getFormattedErrorObject(val) || traceUtil.createErrorObject('mssqlError', val);
if (obj) {
if (obj.parent.config && obj.parent.config.server) {
eventArgs.dbServer = obj.parent.config.server;
}
else if(obj.parent._acquiredConfig && obj.parent._acquiredConfig.server){
eventArgs.dbServer = obj.parent._acquiredConfig.server;
}
}
ctx = agent.asynchEventDone(ctx, eventNameFormatted, eventArgs, errorObject);
storage.set('ctx', ctx);
agent.asynchEventFinish(ctx);
};
}
function findRequest(mssql, require) {
var cache = require.cache;
var json = require('mssql/package.json'); // relative path to package json
var version = json.version;
for (var key in cache) {
var candidate = cache[key];
if (candidate.exports !== mssql) {
continue;
}
var dirname = path.dirname(candidate.filename);
var filename;
if(version.charAt(0)>'5'){
filename = path.join(dirname, 'lib/base', 'request.js');
try {
targetModule.Request = require(filename);
break;
} catch (e) {
return null;
}
}
else{
filename = path.join(dirname, 'lib', 'base.js');
try {
targetModule.Base = require(filename);
break;
} catch (e) {
return null;
}
}
}
}
function getMethodsWithProbes() {
if ( !targetModule.methodMap)
{
var mt =new Object;
mt[0]='mssql#query';
targetModule.methodMap = mt;
}
return targetModule.methodMap;
}
function instrument(methodMap)
{
targetModule.methodMap = methodMap;
}
module.exports.getMethodsWithProbes = getMethodsWithProbes;
module.exports.instrument = instrument.bind(module);