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)); } }