public AfterWrappedMethodDelegate BeforeWrappedMethod(InstrumentedMethodCall instrumentedMethodCall, IAgent agent, ITransaction transaction)
{
//This only happens if we are in an async context. Regardless if we are adding the after delegate as a contination.
if (instrumentedMethodCall.IsAsync)
{
transaction.AttachToAsync();
}
var sqlCommand = (IDbCommand)instrumentedMethodCall.MethodCall.InvocationTarget;
if (sqlCommand == null)
{
return(Delegates.NoOp);
}
var sql = sqlCommand.CommandText ?? string.Empty;
var vendor = SqlWrapperHelper.GetVendorName(sqlCommand);
object GetConnectionInfo() => ConnectionInfoParser.FromConnectionString(vendor, sqlCommand.Connection.ConnectionString);
var connectionInfo = (ConnectionInfo)transaction.GetOrSetValueFromCache(sqlCommand.Connection.ConnectionString, GetConnectionInfo);
var parsedStatement = transaction.GetParsedDatabaseStatement(vendor, sqlCommand.CommandType, sql);
var queryParameters = SqlWrapperHelper.GetQueryParameters(sqlCommand, agent);
var segment = transaction.StartDatastoreSegment(instrumentedMethodCall.MethodCall, parsedStatement, connectionInfo, sql, queryParameters, isLeaf: true);
switch (vendor)
{
case DatastoreVendor.MSSQL:
agent.EnableExplainPlans(segment, () => SqlServerExplainPlanActions.AllocateResources(sqlCommand), SqlServerExplainPlanActions.GenerateExplainPlan, null);
break;
case DatastoreVendor.MySQL:
if (parsedStatement != null)
{
agent.EnableExplainPlans(segment, () => MySqlExplainPlanActions.AllocateResources(sqlCommand), MySqlExplainPlanActions.GenerateExplainPlan, () => MySqlExplainPlanActions.ShouldGenerateExplainPlan(sql, parsedStatement));
}
break;
}
return(ExecuteAsAsync
? Delegates.GetAsyncDelegateFor <Task>(agent, segment)
: Delegates.GetDelegateFor(segment));
}
public AfterWrappedMethodDelegate BeforeWrappedMethod(InstrumentedMethodCall instrumentedMethodCall, IAgent agent, ITransaction transaction)
{
{
var oleDbCommand = (OleDbCommand)instrumentedMethodCall.MethodCall.InvocationTarget;
if (oleDbCommand == null)
{
return(Delegates.NoOp);
}
var sql = oleDbCommand.CommandText ?? string.Empty;
var vendor = SqlWrapperHelper.GetVendorName(oleDbCommand);
var parsedStatement = transaction.GetParsedDatabaseStatement(vendor, oleDbCommand.CommandType, sql);
var queryParameters = SqlWrapperHelper.GetQueryParameters(oleDbCommand, agent);
var segment = transaction.StartDatastoreSegment(instrumentedMethodCall.MethodCall, parsedStatement, null, sql, queryParameters);
return(Delegates.GetDelegateFor(segment));
}
}