Add telemetry data for Credentials retrieval

Author: muhammad-othman

generator/.DevConfigs/a37aa538-27a8-4185-9c66-44d38de5cf57.json

@@ -0,0 +1,9 @@
+{
+  "core": {
+    "changeLogMessages": [
+      "Add telemetry data for Credentials retrieval"
+    ],
+    "type": "patch",
+    "updateMinimum": true
+  }
+}

sdk/src/Core/Amazon.Runtime/Pipeline/Handlers/Signer.cs

@@ -17,6 +17,7 @@
 using Amazon.Runtime.Internal.Util;
 using Amazon.Runtime.Telemetry;
 using Amazon.Runtime.Telemetry.Metrics;
+using Amazon.Runtime.Telemetry.Tracing;
 using Amazon.Util;
 using System;
 using System.IO;
@@ -110,7 +111,14 @@ public static void SignRequest(IRequestContext requestContext)
             using (requestContext.Metrics.StartEvent(Metric.RequestSigningTime))
             using (MetricsUtilities.MeasureDuration(requestContext, TelemetryConstants.AuthSigningDurationMetricName))
             {
-                var immutableCredentials = (requestContext.Identity as AWSCredentials)?.GetCredentials();
+                ImmutableCredentials immutableCredentials = null;
+
+                using (TracingUtilities.CreateSpan(requestContext, TelemetryConstants.CredentialsRetrievalSpanName))
+                using (MetricsUtilities.MeasureDuration(requestContext, TelemetryConstants.ResolveIdentityDurationMetricName))
+                using (requestContext.Metrics.StartEvent(Metric.CredentialsRequestTime))
+                {
+                    immutableCredentials = (requestContext.Identity as AWSCredentials)?.GetCredentials();
+                }
 
                 if (immutableCredentials?.UseToken == true && 
                     !(requestContext.Signer is NullSigner) && 
@@ -155,7 +163,17 @@ private static async Task SignRequestAsync(IRequestContext requestContext)
             using (MetricsUtilities.MeasureDuration(requestContext, TelemetryConstants.AuthSigningDurationMetricName))
             {
                 var awsCredentials = requestContext.Identity as AWSCredentials;
-                var immutableCredentials = awsCredentials != null ? await awsCredentials.GetCredentialsAsync().ConfigureAwait(false) : null;
+                ImmutableCredentials immutableCredentials = null;
+
+                if (awsCredentials != null)
+                {
+                    using (TracingUtilities.CreateSpan(requestContext, TelemetryConstants.CredentialsRetrievalSpanName))
+                    using (MetricsUtilities.MeasureDuration(requestContext, TelemetryConstants.ResolveIdentityDurationMetricName))
+                    using (requestContext.Metrics.StartEvent(Metric.CredentialsRequestTime))
+                    {
+                        immutableCredentials = await awsCredentials.GetCredentialsAsync().ConfigureAwait(false);
+                    }
+                }
 
                 if (immutableCredentials?.UseToken == true &&
                     !(requestContext.Signer is NullSigner) &&

sdk/src/Core/Amazon.Runtime/Telemetry/Metrics/MetricsUtilities.cs

@@ -46,7 +46,7 @@ public static void RecordError(IRequestContext requestContext, Exception excepti
         /// <param name="initialAttributes">The attributes associated with the metric.</param>
         public static void AddMonotonicCounterValue(IRequestContext requestContext, string metricName, string unit, long value = 1, Attributes initialAttributes = null)
         {
-            var serviceId = requestContext.ServiceMetaData.ServiceId;
+            var serviceId = requestContext.ClientConfig.ServiceId;
 
             if (initialAttributes == null)
                 initialAttributes = new Attributes();

sdk/src/Core/Amazon.Runtime/Telemetry/Tracing/TracingUtilities.cs

@@ -39,7 +39,7 @@ public static TraceSpan CreateSpan(
             SpanKind spanKind = SpanKind.INTERNAL,
             SpanContext parentContext = null)
         {
-            var serviceId = requestContext.ServiceMetaData.ServiceId;
+            var serviceId = requestContext.ClientConfig.ServiceId;
 
             if (initialAttributes == null)
                 initialAttributes = new Attributes();