Open Telemetry : Adds implementation for network level Metrics

Directory.Build.props

@@ -3,7 +3,7 @@
 		<ClientOfficialVersion>3.46.0</ClientOfficialVersion>
 		<ClientPreviewVersion>3.47.0</ClientPreviewVersion>
 		<ClientPreviewSuffixVersion>preview.0</ClientPreviewSuffixVersion>
-		<DirectVersion>3.37.1</DirectVersion>
+		<DirectVersion>3.37.2</DirectVersion>
 		<EncryptionOfficialVersion>2.0.4</EncryptionOfficialVersion>
 		<EncryptionPreviewVersion>2.1.0</EncryptionPreviewVersion>
 		<EncryptionPreviewSuffixVersion>preview4</EncryptionPreviewSuffixVersion>

Microsoft.Azure.Cosmos/src/DocumentClient.cs

@@ -959,7 +959,8 @@ internal virtual void Initialize(Uri serviceEndpoint,
             if (this.cosmosClientTelemetryOptions.IsClientMetricsEnabled)
             {
                 CosmosDbOperationMeter.Initialize();
-
+                CosmosDbNetworkMeter.Initialize();
+
                 CosmosDbOperationMeter.AddInstanceCount(this.ServiceEndpoint);
             }
 

Microsoft.Azure.Cosmos/src/Resource/ClientContextCore.cs

@@ -533,16 +533,23 @@ private async Task<TResult> RunWithDiagnosticsHelperAsync<TResult>(
                         && (!this.ClientOptions.CosmosClientTelemetryOptions.DisableDistributedTracing || this.ClientOptions.CosmosClientTelemetryOptions.IsClientMetricsEnabled))
                     {
                         // Extracts and records telemetry data from the result of the operation.
-                        OpenTelemetryAttributes response = openTelemetry?.Item2(result);
+                        OpenTelemetryAttributes otelAttributes = openTelemetry?.Item2(result);
 
                         // Records the telemetry attributes for Distributed Tracing (if enabled)
-                        recorder.Record(response);
+                        recorder.Record(otelAttributes);
 
                         // Records metrics such as request units, latency, and item count for the operation.
                         CosmosDbOperationMeter.RecordTelemetry(getOperationName: getOperationName,
                                                              accountName: this.client.Endpoint,
                                                              containerName: containerName,
                                                              databaseName: databaseName,
+                                                             attributes: otelAttributes);
+
+                        CosmosDbNetworkMeter.RecordTelemetry(getOperationName: getOperationName,
+                                                             accountName: this.client.Endpoint,
+                                                             containerName: containerName,
+                                                             databaseName: databaseName,
+                                                             attributes: otelAttributes);
                                                              attributes: response);
                     }
                     return result;
@@ -584,6 +591,12 @@ private async Task<TResult> RunWithDiagnosticsHelperAsync<TResult>(
                                                              containerName: containerName,
                                                              databaseName: databaseName,
                                                              ex: cosmosException);
+                        CosmosDbNetworkMeter.RecordTelemetry(getOperationName: getOperationName,
+                                                             accountName: this.client.Endpoint,
+                                                             containerName: containerName,
+                                                             databaseName: databaseName,
+                                                             ex: cosmosException);
+
                     }
 
                     throw;

Microsoft.Azure.Cosmos/src/Telemetry/OpenTelemetry/AppInsightClassicAttributeKeys.cs

@@ -7,6 +7,7 @@ namespace Microsoft.Azure.Cosmos.Telemetry
     using System;
     using System.Collections.Generic;
     using global::Azure.Core;
+    using Microsoft.Azure.Cosmos.Tracing.TraceData;
 
     internal sealed class AppInsightClassicAttributeKeys : IActivityAttributePopulator
     {
@@ -162,12 +163,29 @@ public void PopulateAttributes(DiagnosticScope scope, QueryTextMode? queryTextMo
             }
         }
 
+        public KeyValuePair<string, object>[] PopulateNetworkMeterDimensions(string operationName, Uri accountName, string containerName, string databaseName, OpenTelemetryAttributes attributes, CosmosException ex, ClientSideRequestStatisticsTraceDatum.StoreResponseStatistics tcpStats = null, ClientSideRequestStatisticsTraceDatum.HttpResponseStatistics? httpStats = null)
+        {
+            return new KeyValuePair<string, object>[]
+            {
+                new KeyValuePair<string, object>(AppInsightClassicAttributeKeys.ContainerName, containerName),
+                new KeyValuePair<string, object>(AppInsightClassicAttributeKeys.DbName, databaseName),
+                new KeyValuePair<string, object>(AppInsightClassicAttributeKeys.ServerAddress, accountName?.Host),
+                new KeyValuePair<string, object>(AppInsightClassicAttributeKeys.DbOperation, operationName),
+                new KeyValuePair<string, object>(AppInsightClassicAttributeKeys.StatusCode, (int)(attributes?.StatusCode ?? ex?.StatusCode)),
+                new KeyValuePair<string, object>(AppInsightClassicAttributeKeys.SubStatusCode, attributes?.SubStatusCode ?? ex?.SubStatusCode)
+            };
+        }
+
         public KeyValuePair<string, object>[] PopulateOperationMeterDimensions(string operationName, string containerName, string databaseName, Uri accountName, OpenTelemetryAttributes attributes, CosmosException ex)
         {
             return new KeyValuePair<string, object>[]
             {
                 new KeyValuePair<string, object>(AppInsightClassicAttributeKeys.ContainerName, containerName),
                 new KeyValuePair<string, object>(AppInsightClassicAttributeKeys.DbName, databaseName),
+                new KeyValuePair<string, object>(AppInsightClassicAttributeKeys.ServerAddress, accountName?.Host),
+                new KeyValuePair<string, object>(AppInsightClassicAttributeKeys.DbOperation, operationName),
+                new KeyValuePair<string, object>(AppInsightClassicAttributeKeys.StatusCode, (int)(attributes?.StatusCode ?? ex?.StatusCode)),
+                new KeyValuePair<string, object>(AppInsightClassicAttributeKeys.SubStatusCode, attributes?.SubStatusCode ?? ex?.SubStatusCode)
                 new KeyValuePair<string, object>(AppInsightClassicAttributeKeys.ServerAddress, accountName.Host),
                 new KeyValuePair<string, object>(AppInsightClassicAttributeKeys.DbOperation, operationName),
                 new KeyValuePair<string, object>(AppInsightClassicAttributeKeys.StatusCode, (int)(attributes?.StatusCode ?? ex?.StatusCode)),

Microsoft.Azure.Cosmos/src/Telemetry/OpenTelemetry/CosmosDbClientMetrics.cs

@@ -12,7 +12,7 @@ namespace Microsoft.Azure.Cosmos
     public sealed class CosmosDbClientMetrics
     {
         /// <summary>
-        /// OperationMetrics
+        /// Operation Metrics
         /// </summary>
         public static class OperationMetrics
         {
@@ -101,6 +101,120 @@ public static class Description
             }
         }
 
+        /// <summary>
+        /// Network Metrics
+        /// </summary>
+        public static class NetworkMetrics
+        {
+            /// <summary>
+            /// the name of the operation meter
+            /// </summary>
+            public const string MeterName = "Azure.Cosmos.Client.Request";
+
+            /// <summary>
+            /// Version of the operation meter
+            /// </summary>
+            public const string Version = "1.0.0";
+
+            /// <summary>
+            /// Metric Names
+            /// </summary>
+            public static class Name
+            {
+                /// <summary>
+                /// Network Call Latency
+                /// </summary>
+                public const string Latency = "db.client.cosmosdb.request.duration";
+
+                /// <summary>
+                /// Request Payload Size
+                /// </summary>
+                public const string RequestBodySize = "db.client.cosmosdb.request.body.size";
+
+                /// <summary>
+                /// Request Payload Size
+                /// </summary>
+                public const string ResponseBodySize = "db.client.cosmosdb.response.body.size";
+
+                /// <summary>
+                /// Channel Aquisition Latency
+                /// </summary>
+                public const string ChannelAquisitionLatency = "db.client.cosmosdb.request.channel_aquisition.duration";
+
+                /// <summary>
+                /// Backend Server Latency
+                /// </summary>
+                public const string BackendLatency = "db.server.cosmosdb.request.duration";
+
+                /// <summary>
+                /// Transit Time Latency
+                /// </summary>
+                public const string TransitTimeLatency = "db.client.cosmosdb.request.transit.duration";
+
+                /// <summary>
+                /// Received Time Latency
+                /// </summary>
+                public const string ReceivedTimeLatency = "db.client.cosmosdb.request.received.duration";
+            }
+
+            /// <summary>
+            /// Unit for metrics
+            /// </summary>
+            public static class Unit
+            {
+                /// <summary>
+                /// Unit representing bytes
+                /// </summary>
+                public const string Bytes = "bytes";
+
+                /// <summary>
+                /// Unit representing time in seconds
+                /// </summary>
+                public const string Sec = "s";
+            }
+
+            /// <summary>
+            /// Provides descriptions for metrics.
+            /// </summary>
+            public static class Description
+            {
+                /// <summary>
+                /// Network Call Latency
+                /// </summary>
+                public const string Latency = "Duration of client requests.";
+
+                /// <summary>
+                /// Request Payload Size
+                /// </summary>
+                public const string RequestBodySize = "Size of client request body.";
+
+                /// <summary>
+                /// Request Payload Size
+                /// </summary>
+                public const string ResponseBodySize = "Size of client response body.";
+
+                /// <summary>
+                /// Channel Aquisition Latency
+                /// </summary>
+                public const string ChannelAquisitionLatency = "The duration of the successfully established outbound TCP connections. i.e. Channel Aquisition Time (for direct mode).";
+
+                /// <summary>
+                /// Backend Server Latency
+                /// </summary>
+                public const string BackendLatency = "Backend Latency (for direct mode).";
+
+                /// <summary>
+                /// Transit Time Latency
+                /// </summary>
+                public const string TransitTimeLatency = "Time spent on the wire (for direct mode).";
+
+                /// <summary>
+                /// Received Time Latency
+                /// </summary>
+                public const string ReceivedTimeLatency = "Time spent on 'Received' stage (for direct mode).";
+            }
+        }
+
         /// <summary>
         /// Buckets
         /// </summary>