private void OnStop(KeyValuePair <string, object> kv, PropertyFetcherCollection cachedProperties)
{
var activity = Activity.Current;
if (activity is null)
{
Logger.Trace()?.Log("Current activity is null - exiting");
return;
}
if (!_processingSegments.TryRemove(activity.Id, out var segment))
{
Logger.Trace()
?.Log(
"Could not find segment for activity {ActivityId} in tracked segments",
activity.Id);
return;
}
FillSpanLinks(cachedProperties, segment, kv);
var status = cachedProperties.Fetch(kv.Value, "Status") as TaskStatus?;
var outcome = status switch
{
TaskStatus.RanToCompletion => Outcome.Success,
TaskStatus.Canceled => Outcome.Failure,
TaskStatus.Faulted => Outcome.Failure,
_ => Outcome.Unknown
};
segment.Outcome = outcome;
segment.End();
}
private void OnSendStart(KeyValuePair <string, object> kv, string action, PropertyFetcherCollection cachedProperties)
{
var currentSegment = ApmAgent.GetCurrentExecutionSegment();
if (currentSegment is null)
{
Logger.Trace()?.Log("No current transaction or span - exiting");
return;
}
if (kv.Value is null)
{
Logger.Trace()?.Log("Value is null - exiting");
return;
}
var activity = Activity.Current;
var queueName = cachedProperties.Fetch(kv.Value, "Entity") as string;
var destinationAddress = cachedProperties.Fetch(kv.Value, "Endpoint") as Uri;
if (MatchesIgnoreMessageQueues(queueName))
{
return;
}
var spanName = queueName is null
? $"{ServiceBus.SegmentName} {action}"
: $"{ServiceBus.SegmentName} {action} to {queueName}";
var span = currentSegment.StartSpan(spanName, ApiConstants.TypeMessaging, ServiceBus.SubType, action.ToLowerInvariant());
span.Context.Destination = new Destination
{
Address = destinationAddress?.AbsoluteUri,
Service = new Destination.DestinationService
{
Resource = queueName is null ? ServiceBus.SubType : $"{ServiceBus.SubType}/{queueName}"
}
};
if (queueName != null)
{
span.Context.Message = new Message {
Queue = new Queue {
Name = queueName
}
}
}
;
if (!_processingSegments.TryAdd(activity.Id, span))
{
Logger.Trace()
?.Log(
"Could not add {Action} span {SpanId} for activity {ActivityId} to tracked segments",
action,
span.Id,
activity.Id);
}
}
private void OnProcessStart(KeyValuePair <string, object> kv, string action, PropertyFetcherCollection cachedProperties)
{
if (kv.Value is null)
{
Logger.Trace()?.Log("Value is null - exiting");
return;
}
var queueName = cachedProperties.Fetch(kv.Value, "Entity") as string;
if (MatchesIgnoreMessageQueues(queueName))
{
return;
}
var transactionName = queueName is null
? $"{ServiceBus.SegmentName} {action}"
: $"{ServiceBus.SegmentName} {action} from {queueName}";
var transaction = ApmAgent.Tracer.StartTransaction(transactionName, ApiConstants.TypeMessaging);
transaction.Context.Service = new Service(null, null)
{
Framework = _framework
};
if (queueName != null)
{
transaction.Context.Message = new Message {
Queue = new Queue {
Name = queueName
}
}
}
;
// transaction creation will create an activity, so use this as the key.
var activityId = Activity.Current.Id;
if (!_processingSegments.TryAdd(activityId, transaction))
{
Logger.Trace()
?.Log(
"Could not add {Action} transaction {TransactionId} for activity {ActivityId} to tracked segments",
action,
transaction.Id,
activityId);
}
}
private void OnReceiveStart(KeyValuePair <string, object> kv, string action, PropertyFetcherCollection cachedProperties)
{
if (kv.Value is null)
{
Logger.Trace()?.Log("Value is null - exiting");
return;
}
var queueName = cachedProperties.Fetch(kv.Value, "Entity") as string;
if (MatchesIgnoreMessageQueues(queueName))
{
return;
}
var transactionName = queueName is null
? $"{ServiceBus.SegmentName} {action}"
: $"{ServiceBus.SegmentName} {action} from {queueName}";
IExecutionSegment segment;
if (ApmAgent.Tracer.CurrentTransaction is null)
{
var transaction = ApmAgent.Tracer.StartTransaction(transactionName, ApiConstants.TypeMessaging);
transaction.Context.Service = new Service(null, null)
{
Framework = _framework
};
segment = transaction;
}
else
{
var span = ApmAgent.GetCurrentExecutionSegment().StartSpan(transactionName, ApiConstants.TypeMessaging, ServiceBus.SubType, action);
segment = span;
}
// transaction creation will create an activity, so use this as the key.
var activityId = Activity.Current.Id;
if (!_processingSegments.TryAdd(activityId, segment))
{
Logger.Trace()?.Log(
"Could not add {Action} {SegmentName} {TransactionId} for activity {ActivityId} to tracked segments",
action,
segment is ITransaction ? "transaction" : "span",
segment.Id,
activityId);
}
}
private void FillSpanLinks(PropertyFetcherCollection cachedProperties, IExecutionSegment segment, KeyValuePair <string, object> kv)
{
var messages = cachedProperties.Fetch(kv.Value, "Messages") as IEnumerable <object>;
var spanLinks = new List <SpanLink>();
if (messages != null)
{
foreach (var message in messages)
{
if (message.GetType().GetProperty("UserProperties")?.GetValue(message) is Dictionary <string, object> properties)
{
foreach (var property in properties)
{
if (property.Key.Equals("Diagnostic-Id", StringComparison.InvariantCultureIgnoreCase))
{
var parsedTraceParent = DistributedTracingData.TryDeserializeFromString(property.Value.ToString());
if (parsedTraceParent != null)
{
spanLinks.Add(new SpanLink(parsedTraceParent.ParentId, parsedTraceParent.TraceId));
}
}
}
}
}
}
switch (segment)
{
case Model.Transaction t: t.InsertSpanLinkInternal(spanLinks);
break;
case Model.Span s: s.InsertSpanLinkInternal(spanLinks);
break;
}
}
