public AfterWrappedMethodDelegate BeforeWrappedMethod(InstrumentedMethodCall instrumentedMethodCall,
IAgentWrapperApi agentWrapperApi)
{
var context = instrumentedMethodCall.MethodCall.MethodArguments
.ExtractNotNullAs <IIncomingLogicalMessageContext>(0);
var incomingLogicalMessage = context.Message;
if (incomingLogicalMessage == null)
{
throw new NullReferenceException("logicalMessage");
}
var headers = context.Headers;
if (headers == null)
{
throw new NullReferenceException("headers");
}
var queueName = TryGetQueueName(incomingLogicalMessage);
agentWrapperApi.CreateMessageBrokerTransaction(0, "NServiceBus", queueName);
var isegment = agentWrapperApi.StartMessageBrokerSegment(instrumentedMethodCall.MethodCall, 0,
(MessageBrokerAction)1, "NServiceBus", queueName);
agentWrapperApi.ProcessInboundRequest(headers);
return(Delegates.GetDelegateFor(() =>
{
agentWrapperApi.EndSegment(isegment);
agentWrapperApi.EndTransaction();
}, null, agentWrapperApi.NoticeError));
}
public AfterWrappedMethodDelegate BeforeWrappedMethod(InstrumentedMethodCall instrumentedMethodCall,
IAgentWrapperApi agentWrapperApi)
{
var context = instrumentedMethodCall.MethodCall.MethodArguments
.ExtractNotNullAs <IIncomingLogicalMessageContext>(0);
var incomingLogicalMessage = context.Message;
if (incomingLogicalMessage == null)
{
throw new NullReferenceException("logicalMessage");
}
var headers = context.Headers;
if (headers == null)
{
throw new NullReferenceException("headers");
}
var queueName = TryGetQueueName(incomingLogicalMessage);
agentWrapperApi.CreateMessageBrokerTransaction(MessageBrokerDestinationType.Queue, "NServiceBus", queueName);
var segment = agentWrapperApi.StartMessageBrokerSegment(instrumentedMethodCall.MethodCall, MessageBrokerDestinationType.Queue, MessageBrokerAction.Peek, "NServiceBus", queueName);
agentWrapperApi.ProcessInboundRequest(headers);
return(Delegates.GetDelegateFor <Task>(null, task =>
{
agentWrapperApi.RemoveSegmentFromCallStack(segment);
if (task == null)
{
return;
}
if (SynchronizationContext.Current != null)
{
task.ContinueWith(responseTask => agentWrapperApi.HandleExceptions(() =>
{
agentWrapperApi.EndSegment(segment);
agentWrapperApi.EndTransaction();
}), TaskScheduler.FromCurrentSynchronizationContext());
}
else
{
task.ContinueWith(responseTask => agentWrapperApi.HandleExceptions(() =>
{
agentWrapperApi.EndSegment(segment);
agentWrapperApi.EndTransaction();
}), TaskContinuationOptions.ExecuteSynchronously);
}
}, ex =>
{
if (ex != null)
{
agentWrapperApi.NoticeError(ex);
}
agentWrapperApi.EndSegment(segment);
}));
}
private static void AttachCatHeaders(IAgentWrapperApi agentWrapperApi, IOutgoingContext context)
{
var outboundRequestHeaders = agentWrapperApi.GetOutboundRequestHeaders();
foreach (var header in outboundRequestHeaders)
{
if (header.Value != null && header.Key != null)
{
context.Headers[header.Key] = header.Value;
}
}
}
public AfterWrappedMethodDelegate BeforeWrappedMethod(InstrumentedMethodCall instrumentedMethodCall, IAgentWrapperApi agentWrapperApi, ITransactionWrapperApi transactionWrapperApi)
{
/// Read and setup the configured headers from newrelic.config file
if (configuredHeaders == null)
{
IReadOnlyDictionary <string, string> appSettings = agentWrapperApi.Configuration.GetAppSettings();
string reqHeaders = null;
if (appSettings.TryGetValue("requestHeaders", out reqHeaders))
{
configuredHeaders = reqHeaders?.Split(',').Select(p => p.Trim()).ToDictionary(t => t, t => $"http.{t}");
}
else
{
configuredHeaders = new Dictionary <string, string>();
}
}
if (webOperationContextType == null)
{
Assembly webAssembly = Assembly.Load("System.ServiceModel.Web");
webOperationContextType = webAssembly?.GetType("System.ServiceModel.Web.WebOperationContext");
}
if (configuredHeaders != null)
{
BindingFlags flags = BindingFlags.Static | BindingFlags.Public;
object currentContext = webOperationContextType?.GetProperty("Current", flags)?.GetValue(null);
object request = currentContext?.GetType()?.GetProperty("IncomingRequest")?.GetValue(currentContext);
object headers = request?.GetType()?.GetProperty("Headers")?.GetValue(request);
if (headers != null)
{
WebHeaderCollection headerCollection = headers as System.Net.WebHeaderCollection;
if (headerCollection != null)
{
foreach (KeyValuePair <string, string> entry in configuredHeaders)
{
string headerValue = headerCollection.Get(entry.Key);
if (headerValue != null)
{
InternalApi.AddCustomParameter(entry.Value, headerValue);
}
}
}
}
}
object[] methodArgs = instrumentedMethodCall.MethodCall.MethodArguments;
if (methodArgs.Length > 1)
{
object requesturi = methodArgs[0];
string localpath = requesturi?.GetType().GetProperty("LocalPath")?.GetValue(requesturi)?.ToString()?.TrimStart('/');
if (!string.IsNullOrWhiteSpace(localpath))
{
transactionWrapperApi.SetWebTransactionName(WebTransactionType.WCF, localpath, TransactionNamePriority.CustomTransactionName);
}
}
/// does not work see-> https://referencesource.microsoft.com/#System.Data.Services/System/Data/Services/ProcessRequestArgs.cs,fe129b45e15d2da5
/// interface declared internal
/*
* if (configuredHeaders != null)
* {
* // Captures the method argument to be used in the local method later on.
* object[] methodArgs = instrumentedMethodCall.MethodCall.MethodArguments;
* if (methodArgs.Length > 1)
* {
* object requesturi = methodArgs[0];
* //Effectively calling something like this below: string header = controllerContext.HttpContext.Request.Headers.Get("HEADER_NAME");
* object dataserviceObject = methodArgs[1];
* if (dataserviceObject == null)
* throw new NullReferenceException(nameof(dataserviceObject));
* BindingFlags flags = BindingFlags.FlattenHierarchy | BindingFlags.NonPublic ;
* object operationContext = dataserviceObject?.GetType()?.GetProperty("OperationContext", flags)?.GetValue(dataserviceObject);
* Inspect.GetAllProperties(dataserviceObject);
* Inspect.GetAllFields(dataserviceObject);
* Inspect.GetAllMethods(dataserviceObject);
* Inspect.GetAllMembers(dataserviceObject);
* object headers = operationContext?.GetType()?.GetProperty("RequestHeaders")?.GetValue(operationContext);
*
* if (headers != null)
* {
* NameValueCollection headerCollection = headers as NameValueCollection;
* if (headerCollection != null)
* {
* foreach (var cHeader in configuredHeaders)
* {
* string headerValue = headerCollection.Get(cHeader);
* if (headerValue != null)
* {
* InternalApi.AddCustomParameter("http." + cHeader, headerValue);
* }
* }
* }
* }
* }
* }
*/
return(Delegates.GetDelegateFor(
onSuccess: () =>
{
}));
}
public AfterWrappedMethodDelegate BeforeWrappedMethod(InstrumentedMethodCall instrumentedMethodCall, IAgentWrapperApi agentWrapperApi)
{
var context = instrumentedMethodCall.MethodCall.MethodArguments.ExtractNotNullAs <IOutgoingSendContext>(0);
AttachCatHeaders(agentWrapperApi, context);
var queueName = TryGetQueueName(context.Message);
var segment = agentWrapperApi.StartMessageBrokerSegment(instrumentedMethodCall.MethodCall, MessageBrokerDestinationType.Queue, MessageBrokerAction.Produce, "NServiceBus", queueName);
return(Delegates.GetDelegateFor <Task>(null, task =>
{
agentWrapperApi.RemoveSegmentFromCallStack(segment);
if (task == null)
{
return;
}
if (SynchronizationContext.Current != null)
{
task.ContinueWith(responseTask => agentWrapperApi.HandleExceptions(() =>
{
agentWrapperApi.EndSegment(segment);
}), TaskScheduler.FromCurrentSynchronizationContext());
}
else
{
task.ContinueWith(responseTask => agentWrapperApi.HandleExceptions(() =>
{
agentWrapperApi.EndSegment(segment);
}), TaskContinuationOptions.ExecuteSynchronously);
}
}, ex =>
{
if (ex != null)
{
agentWrapperApi.NoticeError(ex);
}
agentWrapperApi.EndSegment(segment);
}));
}
public AfterWrappedMethodDelegate BeforeWrappedMethod(InstrumentedMethodCall instrumentedMethodCall, IAgentWrapperApi agentWrapperApi, ITransactionWrapperApi transactionWrapperApi)
{
/// Read and setup the configured headers from newrelic.config file
if (configuredHeaders == null)
{
IReadOnlyDictionary <string, string> appSettings = agentWrapperApi.Configuration.GetAppSettings();
string reqHeaders = null;
if (appSettings.TryGetValue("requestHeaders", out reqHeaders))
{
configuredHeaders = reqHeaders?.Split(',').Select(p => p.Trim()).ToDictionary(t => t, t => $"http.{t}");
}
else
{
configuredHeaders = new Dictionary <string, string>();
}
}
transactionWrapperApi = agentWrapperApi.CreateTransaction(true, "WCF", "Windows Communication Foundation", false);
var segment = transactionWrapperApi.StartTransactionSegment(instrumentedMethodCall.MethodCall, "InvokeBegin");
if (webOperationContextType == null)
{
Assembly webAssembly = Assembly.Load("System.ServiceModel.Web");
webOperationContextType = webAssembly?.GetType("System.ServiceModel.Web.WebOperationContext");
}
if (configuredHeaders != null)
{
BindingFlags flags = BindingFlags.Static | BindingFlags.Public;
object currentContext = webOperationContextType?.GetProperty("Current", flags)?.GetValue(null);
object request = currentContext?.GetType()?.GetProperty("IncomingRequest")?.GetValue(currentContext);
object headers = request?.GetType()?.GetProperty("Headers")?.GetValue(request);
if (headers != null)
{
WebHeaderCollection headerCollection = headers as System.Net.WebHeaderCollection;
if (headerCollection != null)
{
foreach (KeyValuePair <string, string> entry in configuredHeaders)
{
string headerValue = headerCollection.Get(entry.Key);
if (headerValue != null)
{
InternalApi.AddCustomParameter(entry.Value, headerValue);
}
}
}
}
}
return(Delegates.GetDelegateFor(
onFailure: transactionWrapperApi.NoticeError,
onComplete: () =>
{
segment.End();
transactionWrapperApi.End();
}));
}
public AfterWrappedMethodDelegate BeforeWrappedMethod(InstrumentedMethodCall instrumentedMethodCall, IAgentWrapperApi agentWrapperApi)
{
var context = instrumentedMethodCall.MethodCall.MethodArguments.ExtractNotNullAs <IOutgoingSendContext>(0);
AttachCatHeaders(agentWrapperApi, context);
var queueName = TryGetQueueName(context.Message);
var segment = agentWrapperApi.StartMessageBrokerSegment(instrumentedMethodCall.MethodCall, MessageBrokerDestinationType.Queue, MessageBrokerAction.Produce, "NServiceBus", queueName);
return(Delegates.GetDelegateFor(() => agentWrapperApi.EndSegment(segment)));
}
