Skip to content

Commit f48f8be

Browse files
committed
Move EC2 Metadata Span Check
1 parent 638f18c commit f48f8be

File tree

1 file changed

+6
-10
lines changed

1 file changed

+6
-10
lines changed

src/AWS.Distro.OpenTelemetry.AutoInstrumentation/AwsSpanMetricsProcessor.cs

Lines changed: 6 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -77,11 +77,9 @@ public override void OnEnd(Activity activity)
7777
Dictionary<string, ActivityTagsCollection> attributeDictionary =
7878
this.generator.GenerateMetricAttributeMapFromSpan(activity, this.resource);
7979

80-
if (!IsEc2MetadataApiSpan(attributeDictionary)) {
81-
foreach (KeyValuePair<string, ActivityTagsCollection> attribute in attributeDictionary)
82-
{
83-
this.RecordMetrics(activity, attribute.Value);
84-
}
80+
foreach (KeyValuePair<string, ActivityTagsCollection> attribute in attributeDictionary)
81+
{
82+
this.RecordMetrics(activity, attribute.Value);
8583
}
8684
}
8785

@@ -167,18 +165,16 @@ private void RecordLatency(Activity span, ActivityTagsCollection attributes)
167165
private void RecordMetrics(Activity span, ActivityTagsCollection attributes)
168166
{
169167
// Only record metrics if non-empty attributes are returned.
170-
if (attributes.Count > 0)
168+
if (attributes.Count > 0 && !IsEc2MetadataApiSpan(attributes))
171169
{
172170
this.RecordErrorOrFault(span, attributes);
173171
this.RecordLatency(span, attributes);
174172
}
175173
}
176174

177-
private bool IsEc2MetadataApiSpan(Dictionary<string, ActivityTagsCollection> attributeDict)
175+
private bool IsEc2MetadataApiSpan(ActivityTagsCollection attributes)
178176
{
179-
if (attributeDict.TryGetValue(MetricAttributeGeneratorConstants.DependencyMetric, out ActivityTagsCollection? activityTagsCollection) &&
180-
activityTagsCollection != null &&
181-
activityTagsCollection.TryGetValue(AttributeAWSRemoteService, out object? value) &&
177+
if (attributes.TryGetValue(AttributeAWSRemoteService, out object? value) &&
182178
value is string ip &&
183179
ip == Ec2MetadataApiIp)
184180
{

0 commit comments

Comments
 (0)