Refactor PeerServiceResolver check

open-telemetry · May 13, 2022 · 62feceb · 62feceb
1 parent 5eec545
commit 62feceb
Showing 1 changed file with 9 additions and 9 deletions.
diff --git a/src/OpenTelemetry.Exporter.OpenTelemetryProtocol/Implementation/ActivityExtensions.cs b/src/OpenTelemetry.Exporter.OpenTelemetryProtocol/Implementation/ActivityExtensions.cs
@@ -359,16 +359,16 @@ public bool ForEach(KeyValuePair<string, object> activityTag)
                 if (attribute != null)
                 {
                     PooledList<OtlpCommon.KeyValue>.Add(ref this.Tags, attribute);
-                }
 
-                switch (activityTag.Value)
-                {
-                    case string s:
-                        PeerServiceResolver.InspectTag(ref this, key, s);
-                        break;
-                    case int i:
-                        PeerServiceResolver.InspectTag(ref this, key, i);
-                        break;
+                    switch (attribute.Value.ValueCase)
+                    {
+                        case OtlpCommon.AnyValue.ValueOneofCase.StringValue:
+                            PeerServiceResolver.InspectTag(ref this, key, attribute.Value.StringValue);
+                            break;
+                        case OtlpCommon.AnyValue.ValueOneofCase.IntValue:
+                            PeerServiceResolver.InspectTag(ref this, key, attribute.Value.IntValue);
+                            break;
+                    }
                 }
 
                 return true;